class MetadataSiteTests(TestCase): def setUp(self): self.site = MetadataSite() self.site.register(Article) self.site.register(Person) def tearDown(self): self.site.unregister_all() def test_registers_model(self): self.assertIn(Article, self.site._registry) def test_unregisters_model(self): article_metadata = self.site._registry[Article] self.site.unregister(Article) self.assertNotIn(Article, self.site._registry) self.assertFalse(article_metadata._enabled) @patch.object(MetadataSite, 'unregister') def test_unregisters_all_models(self, unregister): self.site.unregister_all() self.assertEqual(unregister.call_count, 2) unregister.assert_has_calls([call(Article), call(Person)], any_order=True)
def setUp(self): self.site = MetadataSite() self.site.register(Article) self.site.register(Person)