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'), ])