Merge pull request #4 from ilholmes/resource_encoder_tests

Resource encoder tests
yahoo · Oct 19, 2018 · 58fdc52 · 58fdc52
2 parents 90bb7a5 + 156f07d
commit 58fdc52
Showing 1 changed file with 19 additions and 1 deletion.
diff --git a/tests/test_framework.py b/tests/test_framework.py
@@ -21,7 +21,7 @@
 from yahoo_panoptes.framework.plugins.panoptes_base_plugin import PanoptesPluginInfo
 from yahoo_panoptes.framework.resources import PanoptesResource, PanoptesResourceSet, PanoptesResourceDSL, \
     PanoptesContext, ParseException, ParseResults, PanoptesResourcesKeyValueStore, PanoptesResourceStore, \
-    PanoptesResourceCache, PanoptesResourceError, PanoptesResourceCacheException
+    PanoptesResourceCache, PanoptesResourceError, PanoptesResourceCacheException, PanoptesResourceEncoder
 from yahoo_panoptes.framework.utilities.helpers import ordered
 from yahoo_panoptes.framework.utilities.key_value_store import PanoptesKeyValueStore
 
@@ -452,6 +452,24 @@ def test_resource_dsl_parsing(self):
             PanoptesResourceDSL('resources_site = local', panoptes_context)
 
 
+class TestPanoptesResourceEncoder(unittest.TestCase):
+    def setUp(self):
+        self.__panoptes_resource = PanoptesResource(resource_site='test', resource_class='test',
+                                                    resource_subclass='test',
+                                                    resource_type='test', resource_id='test', resource_endpoint='test',
+                                                    resource_plugin='test',
+                                                    resource_creation_timestamp=_TIMESTAMP,
+                                                    resource_ttl=RESOURCE_MANAGER_RESOURCE_EXPIRE)
+
+    def test_panoptes_resource_encoder(self):
+        resource_encoder = PanoptesResourceEncoder()
+        self.assertEqual(resource_encoder.default(set()), list(set()))
+        self.assertEqual(resource_encoder.default(self.__panoptes_resource),
+                         self.__panoptes_resource.__dict__['_PanoptesResource__data'])
+        with self.assertRaises(TypeError):
+            resource_encoder.default(dict())
+
+
 class TestPanoptesResourceCache(unittest.TestCase):
     def setUp(self):
         self.__panoptes_resource = PanoptesResource(resource_site='test', resource_class='test',