def test_get_data_kerpow(self, mock_get):
mock_get.return_value.json.return_value = {'error': ':('}
pub = NotificationPublisher()
actual = pub.get_data('thingy', 'params')
self.assertEquals(actual, None)
self.assertEquals(mock_get.call_args_list, [
call(pub.api, 'thingy', 'params'),
])
def test_get_data_kaboom(self, mock_get):
mock_get.side_effect = ValueError
pub = NotificationPublisher()
actual = pub.get_data('thingy', 'params')
self.assertEquals(actual, None)
self.assertEquals(mock_get.call_args_list, [
call(pub.api, 'thingy', 'params'),
])
def test_get_data_kerpow(self, mock_get):
mock_get.return_value.json.return_value = {'error': ':('}
pub = NotificationPublisher()
actual = pub.get_data('thingy', 'params')
self.assertEquals(actual, None)
self.assertEquals(mock_get.call_args_list, [
call(pub.api, 'thingy', 'params'),
])
def test_get_data_kaboom(self, mock_get):
mock_get.side_effect = ValueError
pub = NotificationPublisher()
actual = pub.get_data('thingy', 'params')
self.assertEquals(actual, None)
self.assertEquals(mock_get.call_args_list, [
call(pub.api, 'thingy', 'params'),
])
def test_get_data(self, mock_get):
messages = [
{'id': 1, 'time': '2015-05-01T01:02:03+00:00'},
{'id': 2, 'time': '2015-05-01T01:02:04+00:00'},
{'id': 3, 'time': '2015-05-01T01:02:05+00:00'},
]
mock_get.return_value.json.return_value = {'objects': messages}
pub = NotificationPublisher()
actual = pub.get_data('thingy', 'params')
self.assertEquals(actual, messages)
self.assertEquals(mock_get.call_args_list, [
call(pub.api, 'thingy', 'params'),
])
def test_get_data(self, mock_get):
messages = [
{
'id': 1,
'time': '2015-05-01T01:02:03+00:00'
},
{
'id': 2,
'time': '2015-05-01T01:02:04+00:00'
},
{
'id': 3,
'time': '2015-05-01T01:02:05+00:00'
},
]
mock_get.return_value.json.return_value = {'objects': messages}
pub = NotificationPublisher()
actual = pub.get_data('thingy', 'params')
self.assertEquals(actual, messages)
self.assertEquals(mock_get.call_args_list, [
call(pub.api, 'thingy', 'params'),
])