def test_register_ok(self):
     registry = CacheRegistry()
     registry.register(TestSerializer)
     expected_registry = {
         TestModel: [TestSerializer],
     }
     self.assertEqual(registry._registry, expected_registry)
示例#2
0
    def setUp(self):
        self.registry = CacheRegistry()
        self.registry.register(TestCachedSerializer)
        self.registry.register(TestCachedSerializer2)

        self.instance = instance = TestModel()
        instance.id = 1000
        instance.save()
        self.serializer = TestCachedSerializer(instance)
        self.key = self.registry.get_cache_key(self.instance, TestCachedSerializer)
        self.expected_data = {"id": 1000, "name": ""}
 def setUp(self):
     self.registry = CacheRegistry()
     self.registry.register(TestSerializer)
     self.registry.register(TestSerializer2)
 def test_alread_registered(self):
     registry = CacheRegistry()
     registry.register(TestSerializer)
     with self.assertRaises(AlreadyRegistered):
         registry.register(TestSerializer)
 def test_empty_registry(self):
     registry = CacheRegistry()
     self.assertEqual(registry._registry, {})