def create_model(self):
reader = Reader()
reader.uuid = "uuid_of_device"
reader.device_token = "token_of_device"
reader.os = ANDROID
return reader
def tearDown(self):
Reader.drop_collection()
self.model = None
def test_save_not_call(self,getByUUIDMock, saveMock):
model = self.model
getByUUIDMock.return_value = model, False
Reader.getOrCreate(uuid=model.uuid, os=model.os, token=model.device_token)
saveMock.assert_called_once_with()
def test_save_to_database(self):
model = self.model
Reader.getOrCreate(uuid=model.uuid, os=model.os, token=model.device_token)
self.assertIsNotNone(Reader.objects.first())
def test_get_by_uuid(self, getByUUIDMock):
model = self.model
getByUUIDMock.return_value = model, False
reader = Reader.getOrCreate(uuid=model.uuid, os=model.os, token=model.device_token)
getByUUIDMock.assert_called_once_with(uuid=model.uuid)
self.assertEqual(model,reader)