Adolfo Reyna
32 lines
1.2 KiB
Python
32 lines
1.2 KiB
Python
import importlib.util
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
MODULE_PATH = Path(__file__).parents[1] / 'src' / 'server_catalog.py'
|
|
spec = importlib.util.spec_from_file_location('server_catalog', MODULE_PATH)
|
|
server_catalog = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(server_catalog)
|
|
|
|
|
|
class ServerCatalogTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.catalog = {
|
|
'units': [{
|
|
'filename': 'CKMath_G3U1.zip',
|
|
'materials': [{'id': 'guide-1', 'internal_path': 'unit/guide.pdf', 'filename': 'guide.pdf'}],
|
|
}]
|
|
}
|
|
|
|
def test_known_material_id_resolves_to_its_declared_zip_member(self):
|
|
index = server_catalog.material_index(self.catalog)
|
|
self.assertEqual(server_catalog.lookup_material(index, 'guide-1'), ('CKMath_G3U1.zip', 'unit/guide.pdf'))
|
|
|
|
def test_unknown_or_path_like_id_is_not_resolved(self):
|
|
index = server_catalog.material_index(self.catalog)
|
|
self.assertIsNone(server_catalog.lookup_material(index, '../../etc/passwd'))
|
|
self.assertIsNone(server_catalog.lookup_material(index, 'not-in-the-catalog'))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|