def test_get_instance_fail(self):
model_action = ModelAction(
model=self.ma_model,
action=self.action,
content=self.ma_content,
instance=None,
timestamp=None,
access=None,
process=None,
)
with patch('audit_tools.tests.audit.models.test_model_action.Foo') as model_mock:
model_mock.objects.get.side_effect = Exception
self.assertIsNone(model_action.get_instance())
def test_get_instance_fail(self):
model_action = ModelAction(
model=self.ma_model,
action=self.action,
content=self.ma_content,
instance=None,
timestamp=None,
access=None,
process=None,
)
with patch('audit_tools.tests.audit.models.test_model_action.Foo'
) as model_mock:
model_mock.objects.get.side_effect = Exception
self.assertIsNone(model_action.get_instance())
def test_get_instance(self):
model_action = ModelAction(
model=self.ma_model,
action=self.action,
content=self.ma_content,
instance=self.ma_instance,
timestamp=None,
access=None,
process=None,
)
foo = Foo()
with patch('audit_tools.tests.audit.models.test_model_action.Foo') as model_mock:
model_mock.objects.get.return_value = foo
self.assertEqual(foo, model_action.get_instance())
def test_get_instance(self):
model_action = ModelAction(
model=self.ma_model,
action=self.action,
content=self.ma_content,
instance=self.ma_instance,
timestamp=None,
access=None,
process=None,
)
foo = Foo()
with patch('audit_tools.tests.audit.models.test_model_action.Foo'
) as model_mock:
model_mock.objects.get.return_value = foo
self.assertEqual(foo, model_action.get_instance())
