def testQueryLastEmittedNonMetricSequence(self, collectorsdbMock):
collectorsdbMock.engineFactory.return_value = Mock(
spec_set=sqlalchemy.engine.Engine)
metric_utils.queryLastEmittedNonMetricSequence("twitter-tweets-volume")
args, _ = (collectorsdbMock
.engineFactory
.return_value
.execute
.call_args_list[0])
self.assertTrue(args)
self.assertIsInstance(args[0], sqlalchemy.sql.Select)
def testEmittedNonMetricSequence(self):
key = "bogus-test-key"
metric_utils.updateLastEmittedNonMetricSequence(key, 1)
# Cleanup
self.addCleanup(
collectorsdb.engineFactory().execute,
schema.emittedNonMetricTracker.delete().where( # pylint: disable=E1120
(schema.emittedNonMetricTracker.c.key == key)
)
)
lastEmittedSample = metric_utils.queryLastEmittedNonMetricSequence(key)
self.assertEqual(1, lastEmittedSample)
