def test_locked_put_with_overwite(self, djangoModel): get_or_create_mock = mock.Mock() fake_credentials = FakeCredentialsModelMock() get_or_create_mock.return_value = (fake_credentials, True) object_mock = mock.Mock() object_mock.get_or_create = get_or_create_mock FakeCredentialsModelMock.objects.get_or_create = get_or_create_mock storage = Storage(FakeCredentialsModelMock, self.key_name, self.key_value, self.property_name) storage.locked_put(self.credentials, True) self.assertTrue(fake_credentials.saved)
def test_locked_put(self, djangoModel): storage = Storage(FakeCredentialsModelMock, self.key_name, self.key_value, self.property_name) storage.locked_put(self.credentials)