def test_consumer_is_executed_successfully(self, mock_get, mock_import): response_mock = Mock() response_mock.json.return_value = {} mock_get.return_value = response_mock post_list = consumer.consume() mock_get.assert_called_once_with( 'http://us-ignite.org/', params={'count': 1000, 'json': 'get_recent_posts'}) response_mock.json.assert_called_once_with() eq_(mock_import.call_count, 0) eq_(post_list, [])
def handle(self, *args, **kwargs): print "Importing blogposts." post_list = consumer.consume(count=1000) print "Imported %s posts" % len(post_list) print "Done at %s." % datetime.now()