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)