/* * Copyright (c) 2023, Kemal Zebari . * * SPDX-License-Identifier: BSD-2-Clause */ #include #include TEST_CASE(determine_computed_mime_type_given_no_sniff_is_set) { auto mime_type = MUST(Web::MimeSniff::MimeType::create("text"_string, "html"_string)); auto computed_mime_type = MUST(Web::MimeSniff::Resource::sniff("\x00"sv.bytes(), Web::MimeSniff::SniffingConfiguration { .supplied_type = mime_type, .no_sniff = true })); EXPECT_EQ("text/html"sv, MUST(computed_mime_type.serialized())); // Cover the edge case in the context-specific sniffing algorithm. computed_mime_type = MUST(Web::MimeSniff::Resource::sniff("\x00"sv.bytes(), Web::MimeSniff::SniffingConfiguration { .sniffing_context = Web::MimeSniff::SniffingContext::Image, .supplied_type = mime_type, .no_sniff = true, })); EXPECT_EQ("text/html"sv, MUST(computed_mime_type.serialized())); } TEST_CASE(determine_computed_mime_type_given_no_sniff_is_unset) { auto supplied_type = MUST(Web::MimeSniff::MimeType::create("text"_string, "html"_string)); auto computed_mime_type = MUST(Web::MimeSniff::Resource::sniff("\x00"sv.bytes(), Web::MimeSniff::SniffingConfiguration { .supplied_type = supplied_type })); EXPECT_EQ("application/octet-stream"sv, MUST(computed_mime_type.serialized())); // Make sure we cover the XML code path in the mime type sniffing algorithm. auto xml_mime_type = "application/rss+xml"sv; supplied_type = MUST(Web::MimeSniff::MimeType::parse(xml_mime_type)).release_value(); computed_mime_type = MUST(Web::MimeSniff::Resource::sniff("\x00"sv.bytes(), Web::MimeSniff::SniffingConfiguration { .supplied_type = supplied_type })); EXPECT_EQ(xml_mime_type, MUST(computed_mime_type.serialized())); } static void set_image_type_mappings(HashMap>& mime_type_to_headers_map) { mime_type_to_headers_map.set("image/x-icon"sv, { "\x00\x00\x01\x00"sv, "\x00\x00\x02\x00"sv }); mime_type_to_headers_map.set("image/bmp"sv, { "BM"sv }); mime_type_to_headers_map.set("image/gif"sv, { "GIF87a"sv, "GIF89a"sv }); mime_type_to_headers_map.set("image/webp"sv, { "RIFF\x00\x00\x00\x00WEBPVP"sv }); mime_type_to_headers_map.set("image/png"sv, { "\x89PNG\x0D\x0A\x1A\x0A"sv }); mime_type_to_headers_map.set("image/jpeg"sv, { "\xFF\xD8\xFF"sv }); } static void set_audio_or_video_type_mappings(HashMap>& mime_type_to_headers_map) { mime_type_to_headers_map.set("audio/aiff"sv, { "FORM\x00\x00\x00\x00\x41IFF"sv }); mime_type_to_headers_map.set("audio/mpeg"sv, { "ID3"sv }); mime_type_to_headers_map.set("application/ogg"sv, { "OggS\x00"sv }); mime_type_to_headers_map.set("audio/midi"sv, { "MThd\x00\x00\x00\x06"sv }); mime_type_to_headers_map.set("video/avi"sv, { "RIFF\x00\x00\x00\x00\x41\x56\x49\x20"sv }); mime_type_to_headers_map.set("audio/wave"sv, { "RIFF\x00\x00\x00\x00WAVE"sv }); } TEST_CASE(determine_computed_mime_type_in_both_none_and_browsing_sniffing_context) { HashMap> mime_type_to_headers_map; mime_type_to_headers_map.set("application/octet-stream"sv, { "\x00"sv }); mime_type_to_headers_map.set("text/html"sv, { "\x09\x09"sv, "\x20