# -*- coding: utf-8 -*-

"""

Tests for :module:`pagure.lib.mimetype`.

"""

from __future__ import unicode_literals, absolute_import

import os

import unittest

import sys

from pagure.lib import mimetype

sys.path.insert(

    0, os.path.join(os.path.dirname(os.path.abspath(__file__)), "..")

)

class TestMIMEType(unittest.TestCase):

    def test_guess_type(self):

        dataset = [

            ("hello.html", None, "text/html", None),

            ("hello.html", b"#!", "text/html", "ascii"),

            ("hello", b"#!", "text/plain", "ascii"),

            ("hello.jpg", None, "image/jpeg", None),

            ("hello.jpg", b"#!", "image/jpeg", None),

            ("hello.jpg", b"\0", "image/jpeg", None),

            (None, "😋".encode("utf-8"), "text/plain", "utf-8"),

            ("hello", b"\0", "application/octet-stream", None),

            ("hello", None, None, None),

        ]

        for data in dataset:

            result = mimetype.guess_type(data[0], data[1])

            self.assertEqual(

                (data[2], data[3]),

                result,

                "Wrong mimetype for filename %r and content %r"

                % (data[0], data[1]),

            )

    def test_get_html_file_headers(self):

        result = mimetype.get_type_headers("hello.html", None)

        expected = {

            "Content-Type": "application/octet-stream",

            "Content-Disposition": "attachment",

            "X-Content-Type-Options": "nosniff",

        }

        self.assertEqual(result, expected)

    def test_get_normal_headers(self):

        dataset = [

            ("hello", b"#!", "text/plain; charset=ascii"),

            ("hello.jpg", None, "image/jpeg"),

            ("hello.jpg", b"#!", "image/jpeg"),

            ("hello.jpg", b"\0", "image/jpeg"),

            (None, "😋".encode("utf-8"), "text/plain; charset=utf-8"),

            ("hello", b"\0", "application/octet-stream"),

        ]

        for data in dataset:

            result = mimetype.get_type_headers(data[0], data[1])

            self.assertEqual(

                result["Content-Type"],

                data[2],

                "Wrong Content-Type for filename %r and content %r"

                % (data[0], data[1]),

            )

    def test_get_none_header(self):

        self.assertIsNone(mimetype.get_type_headers("hello", None))

if __name__ == "__main__":

    unittest.main(verbosity=2)