def test_should_get_the_assigned_value(self):
objectsMock = DjangoModelObjectsMock()
objectsMock.start()
objectsMock.when_called_with(name = 'value').return_value = 'assigned_value'
result = MyModel.objects.get(name = 'value')
self.assertEqual('assigned_value', result)
objectsMock.stop()
def test_should_get_a_mock_when_the_modelObject_mock_is_applied(self):
objectsMock = DjangoModelObjectsMock()
objectsMock.start()
result = MyModel.objects.get()
self.assertIsInstance(result, Mock)
objectsMock.stop()