diff --git a/tests/data/image.jpeg b/tests/data/image.jpeg new file mode 100644 index 0000000000000000000000000000000000000000..3df13ea92d222614d298eb082b9ebae865f880d1 Binary files /dev/null and b/tests/data/image.jpeg differ diff --git a/tests/data/valid.pdf b/tests/data/valid.pdf new file mode 100644 index 0000000000000000000000000000000000000000..d30b784dfa4a4d7ace9109d100f460b92df8663b Binary files /dev/null and b/tests/data/valid.pdf differ diff --git a/tests/test_jpeg_format.py b/tests/test_jpeg_format.py new file mode 100644 index 0000000000000000000000000000000000000000..370409d9fd4131338053d2e27f2a7e74663b85f1 --- /dev/null +++ b/tests/test_jpeg_format.py @@ -0,0 +1,18 @@ +import unittest +from formats.jpeg_format import is_valid + + +class TestJpegFormat(unittest.TestCase): + def test_is_valid__valid(self): + resource = {} + with open("tests/data/image.jpeg", "r") as file: + self.assertTrue(is_valid(resource, file)) + + def test_is_valid__invalid(self): + resource = {} + with open("tests/data/image.png", "r") as file: + self.assertFalse(is_valid(resource, file)) + + +if __name__ == "__main__": + unittest.main() diff --git a/tests/test_pdf_format.py b/tests/test_pdf_format.py new file mode 100644 index 0000000000000000000000000000000000000000..1f6e25f87e2d3f056fbb5304e0bd6cb4bc1f552b --- /dev/null +++ b/tests/test_pdf_format.py @@ -0,0 +1,18 @@ +import unittest +from formats.pdf_format import is_valid + + +class TestPdfFormat(unittest.TestCase): + def test_is_valid__valid(self): + resource = {} + with open("tests/data/valid.pdf", "r") as file: + self.assertTrue(is_valid(resource, file)) + + def test_is_valid__invalid(self): + resource = {} + with open("tests/data/image.png", "r") as file: + self.assertFalse(is_valid(resource, file)) + + +if __name__ == "__main__": + unittest.main()