test_zip_format.py

import unittest
from formats.zip_format import is_valid


class TestZipFormat(unittest.TestCase):
    def test_is_valid__valid(self):
        resource = {}
        with open("tests/data/bermuda.zip", "r") as file:
            self.assertTrue(is_valid(resource, file))
            self.assertIsNone(resource.get("error"))

    def test_is_valid__invalid(self):
        resource = {}
        with open("tests/data/correct.xml", "r") as file:
            self.assertFalse(is_valid(resource, file))
            self.assertIsNotNone(resource.get("error"))
            self.assertEqual("Not a ZIP file.", resource["error"])


if __name__ == "__main__":
    unittest.main()