def test_to_dict_returns_one_dictionary_when_cursor_has_one_result(self): source = Source() source.id = 1 source.name = 'foo_source' source.save() news_item = NewsItem() news_item.id = 1 news_item.title = 'foo' news_item.description = 'bar' news_item.url = 'https://www.google.com' news_item.added_at = '2018-11-23 01:00:00+00:00' news_item.source = source news_item.published = False news_item.score = 1 news_item.save() with connection.cursor() as cursor: cursor.execute('SELECT * FROM news_item') result = self.newsitem_metric.to_dict(cursor) self.assertEquals(1, len(result)) self.assertEquals([{ 'id': 1, 'title': 'foo', 'description': 'bar', 'url': 'https://www.google.com', 'added_at': datetime(2018, 11, 23, 1, 0), 'source_id': 1, 'published': False, 'score': 1 }], result)
def test_handle_publishes_when_newsitems_exist(self): news_item = NewsItem() news_item.id = 1 news_item.title = 'foo' news_item.save() # the method being tested self.command.handle() classify_requeue.pika.BlockingConnection.assert_called_once() classify_requeue.pika.ConnectionParameters.assert_called_once() self.connection.channel.assert_called_once() self.channel.queue_declare.assert_called_once_with( queue=settings.QUEUE_NAME_CLASSIFY, durable=True) self.channel.basic_publish.assert_called_once() self.logger.info.assert_any_call( 'Found %s news items that need to be classified.', 1) self.logger.info.assert_any_call('Successfully re-queued #%s "%s"!', 1, 'foo')