diff --git a/tests/test_engine_resources.py b/tests/test_engine_resources.py index f8ab558adb59593edf21a23029a9cc5620fd42ef..cf1c3d06365615b0b6ceee83c0e3c4d019d6933d 100644 --- a/tests/test_engine_resources.py +++ b/tests/test_engine_resources.py @@ -24,7 +24,12 @@ import validators from aura_engine.base.config import AuraConfig from aura_engine.core.channels import ChannelName, ChannelType -from aura_engine.resources import ResourceClass, ResourceMapping, ResourceType +from aura_engine.resources import ( + ResourceClass, + ResourceMapping, + ResourceType, + ResourceUtil, +) class TestEngineResources(unittest.TestCase): @@ -105,6 +110,36 @@ class TestEngineResources(unittest.TestCase): self.assertEqual(3, rc_playlist.value.get("numeric")) self.assertEqual(rc_playlist.types, rc_playlist.value.get("types")) + def test_resource_util_get_content_type(self): + print(self._testMethodName) + + rt = ResourceUtil.get_content_type("file://magic.resource") + self.assertEqual(ResourceType.FILE, rt) + rt = ResourceUtil.get_content_type("http://magic.resource") + self.assertEqual(ResourceType.STREAM_HTTP, rt) + rt = ResourceUtil.get_content_type("https://magic.resource") + self.assertEqual(ResourceType.STREAM_HTTP, rt) + rt = ResourceUtil.get_content_type("line://magic.resource") + self.assertEqual(ResourceType.LINE, rt) + rt = ResourceUtil.get_content_type("playlist://magic.resource") + self.assertEqual(ResourceType.PLAYLIST, rt) + rt = ResourceUtil.get_content_type("pool://magic.resource") + self.assertEqual(ResourceType.POOL, rt) + + def test_resource_util_get_content_class(self): + print(self._testMethodName) + + rc = ResourceUtil.get_content_class(ResourceType.FILE) + self.assertEqual(ResourceClass.FILE, rc) + rc = ResourceUtil.get_content_class(ResourceType.STREAM_HTTP) + self.assertEqual(ResourceClass.STREAM, rc) + rc = ResourceUtil.get_content_class(ResourceType.LINE) + self.assertEqual(ResourceClass.LIVE, rc) + rc = ResourceUtil.get_content_class(ResourceType.PLAYLIST) + self.assertEqual(ResourceClass.PLAYLIST, rc) + rc = ResourceUtil.get_content_class(ResourceType.POOL) + self.assertEqual(ResourceClass.PLAYLIST, rc) + if __name__ == "__main__": unittest.main()