def test_is_plugin(self): pclass = get_plugin_types()[0] class Test1(pclass): pass class Test2(object): pass self.assertFalse(is_plugin(pclass)) self.assertTrue(is_plugin(Test1)) self.assertFalse(is_plugin(Test2))
def test_get_plugin_types(self): all_types = set(p.get_class() for p in PluginType.objects.all()) # Call two times to test caching self.assertEqual(all_types, set(get_plugin_types())) self.assertEqual(all_types, set(get_plugin_types()))