def test_register_fail(self, settings, pre_delete, post_save, pre_save, logger, dynamic_import):
settings.LOGGED_MODELS = ('audit_tools.audit.tests.TestClass', )
pre_save.connect.side_effect = Exception('TestException')
signals.register_models()
self.assertEqual(logger.error.call_count, 1)
def test_register_ok(self, settings, pre_delete, post_save, pre_save, logger, dynamic_import):
settings.LOGGED_MODELS = ('audit_tools.audit.fail.FailClass', )
signals.register_models()
self.assertEqual(pre_save.connect.call_count, 1)
self.assertEqual(post_save.connect.call_count, 1)
self.assertEqual(pre_delete.connect.call_count, 1)
self.assertEqual(logger.error.call_count, 0)
def ready(self):
# Register all models listed in LOGGED_MODELS
register_models()