class GetAllCacheKeyTestCase(unittest.TestCase): def setUp(self): self.registry = CacheRegistry() self.registry.register(TestSerializer) self.registry.register(TestSerializer2) def test_ok(self): instance = TestModel(id=1000) keys = self.registry.get_all_cache_keys(instance) self.assertEqual(len(keys), 2) self.assertIn( CacheRegistry.get_cache_key(TestModel(id=1000), TestSerializer), keys) self.assertIn( CacheRegistry.get_cache_key(TestModel(id=1000), TestSerializer2), keys)