x
/
almanach
Code Issues Proposed changes

Restore entity controller test call 

Change-Id: Id32a0c7afab93eac4b96b301cccc71acfb8e5383
This commit is contained in:
Paul Millette 2017-01-06 15:10:03 -05:00
parent 3d175cdae4
commit 2f6e86049b
1 changed files with 12 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
almanach/tests/unit/core/controllers/test_entity_controller.py
View File

@ -115,11 +115,20 @@ class TestEntityController(base.BaseTestCase):
end = datetime(2016, 3, 3, 0, 0, 0, 0, pytz.utc)
flavor = 'a_new_flavor'
fake_instance1 = a(instance().with_start(2016, 3, 1, 0, 0, 0).with_end(2016, 3, 2, 0, 0, 0))
self.database_adapter.get_all_entities_by_id_and_date.return_value = fake_instance1
self.database_adapter.get_all_entities_by_id_and_date.side_effect = [[fake_instance1], [fake_instance1]]
calls = [
mock.call(fake_instance1.entity_id, start, end),
mock.call(fake_instance1.entity_id, start, end)
]
self.database_adapter.get_all_entities_by_id_and_date.expect_called_once_with(
fake_instance1.entity_id, start, end
self.controller.update_inactive_entity(
instance_id=fake_instance1.entity_id,
start=start,
end=end,
flavor=flavor,
)
self.database_adapter.get_all_entities_by_id_and_date.assert_has_calls(calls)
self.database_adapter.update_closed_entity.expect_called_once_with(
entity=fake_instance1, data={"flavor": flavor}
)