def test_update_one_message(self): """ Test called with one message """ healthcheck = { 'messages': [{ 'status': 'status' }], 'category': 'category', 'id': 'id' } service = MagicMock() _update(service, healthcheck) service.state.set.assert_called_once_with('category', 'id', 'status')
def test_update_multiple_messages(self): """ Test called with one message """ healthcheck = { 'messages': [{ 'status': 'status', 'id': 1 }, { 'status': 'status', 'id': 2 }], 'category': 'category', 'id': 'id' } service = MagicMock() _update(service, healthcheck) service.state.set.assert_has_calls([ call('category', 'id-1', 'status'), call('category', 'id-2', 'status') ])