def test_queuereturns_none(self):
archive_topic = 'archive'
self.setup_archive_queue_tests(archive_topic)
with mock.patch('user.MQTTSubscribe.MQTTSubscriber') as mock_manager:
type(mock_manager.return_value).get_data = mock.Mock(return_value=self.generator([None]))
SUT = MQTTSubscribeDriver(**self.config_dict)
gen = SUT.genArchiveRecords(0)
data = next(gen, None)
self.assertIsNone(data)
def test_archive_topic_not_set(self):
config_dict = {}
config_dict['topics'] = {}
records = list()
with mock.patch('user.MQTTSubscribe.MQTTSubscriber'):
with self.assertRaises(NotImplementedError) as error:
SUT = MQTTSubscribeDriver(**config_dict)
for record in SUT.genArchiveRecords(int(time.time() + 10.5)):
records.append(record)
self.assertEqual(len(error.exception.args), 0)
def test_queue(self):
archive_topic = 'archive'
self.setup_archive_queue_tests(archive_topic)
queue_list = [self.queue_data, self.queue_data]
with mock.patch('user.MQTTSubscribe.MQTTSubscriber') as mock_manager:
type(mock_manager.return_value).get_data = mock.Mock(return_value=self.generator([self.queue_data, self.queue_data, None]))
records = list()
SUT = MQTTSubscribeDriver(**self.config_dict)
for record in SUT.genArchiveRecords(int(time.time() - 10.5)):
records.append(record)
self.assertListEqual(records, queue_list)