test: content util type and class

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()