def test_list_with_some_contents(self):
c1 = Colour(name="blue", episode=self.episode).save()
c2 = Colour(name="red", episode=self.episode).save()
mock_request = MagicMock(name='mock request')
mock_request.user = self.user
response = self.viewset().list(mock_request)
data = [{
u'consistency_token': u'',
u'created': None,
u'created_by_id': None,
u'episode_id': 1,
u'id': 1,
u'name': u'blue',
u'updated': None,
u'updated_by_id': None
}, {
u'consistency_token': u'',
u'created': None,
u'created_by_id': None,
u'episode_id': 1,
u'id': 2,
u'name': u'red',
u'updated': None,
u'updated_by_id': None
}]
self.assertEqual(data, json.loads(response.content.decode('UTF-8')))
def test_integration(self, subrecords):
subrecords.return_value = [Colour]
colour_1 = Colour(episode=self.episode)
colour_1.update_from_dict({
"name": "blue",
}, self.user)
consistency_token_1 = colour_1.consistency_token
colour_2 = Colour(episode=self.episode)
colour_2.update_from_dict({
"name": "orange",
}, self.user)
consistency_token_2 = colour_2.consistency_token
colour_dict_1 = colour_1.to_dict(self.user)
colour_dict_1["name"] = "Red"
colour_dict_1.pop("episode_id")
colour_dict_2 = colour_2.to_dict(self.user)
colour_dict_2.pop("episode_id")
provided_dict = dict(colour=[colour_dict_1, colour_dict_2])
dumped = json.loads(json.dumps(provided_dict, cls=OpalSerializer))
self.pathway_example.save(dumped, self.user, self.patient,
self.episode)
saved_colour_1 = self.episode.colour_set.get(id=colour_1.id)
self.assertNotEqual(saved_colour_1.consistency_token,
consistency_token_1)
saved_colour_2 = self.episode.colour_set.get(id=colour_2.id)
self.assertEqual(saved_colour_2.consistency_token, consistency_token_2)
def test_sender(self):
mock_receiver = MagicMock(name='mock receiver')
@receiver(worker.subrecord_post_save, sender=Colour)
def rec(*a, **k):
mock_receiver(*a, **k)
p, e = self.new_patient_and_episode_please()
blue = Colour(episode=e, name='Blue')
blue.save()
self.assertEqual(1, mock_receiver.call_count)