def test_add_bad_response(self):
client = Client(self.api_url)
contact_manager = ContactManager(client)
contact_value = '#channel'
with patch.object(client, 'put', return_value={}) as put_mock, \
patch.object(client, 'get', return_value={'list': []}) as get_mock:
with self.assertRaises(ResponseStructureError):
contact_manager.add(contact_value, CONTACT_SLACK)
self.assertTrue(put_mock.called)
self.assertTrue(get_mock.called)
expected_request_data = {'value': contact_value, 'type': CONTACT_SLACK}
put_mock.assert_called_with('contact/', json=expected_request_data)
def test_add(self):
client = Client(self.api_url)
contact_manager = ContactManager(client)
contact_value = '#channel'
contact_id = 1
with patch.object(client, 'put', return_value={'id': contact_id}) as put_mock, \
patch.object(client, 'get', return_value={'list': []}) as get_mock:
res_contact = contact_manager.add(contact_value, CONTACT_SLACK)
self.assertTrue(put_mock.called)
self.assertTrue(get_mock.called)
expected_request_data = {'value': contact_value, 'type': CONTACT_SLACK}
expected_contact = Contact(id=contact_id, **expected_request_data)
self.assertEqual(expected_contact, res_contact)
put_mock.assert_called_with('contact/', json=expected_request_data)
