def testEmittedSampleDatetime(self): key = "bogus-test-key" # Establish initial sample datetime result = metric_utils.establishLastEmittedSampleDatetime(key, 300) # Cleanup self.addCleanup( collectorsdb.engineFactory().execute, schema.emittedSampleTracker.delete().where( # pylint: disable=E1120 (schema.emittedSampleTracker.c.key == key) ) ) self.assertIsInstance(result, datetime) # Update latest emitted sample datetime to now now = datetime.utcnow().replace(microsecond=0) metric_utils.updateLastEmittedSampleDatetime(key, now) # Verify that it was updated lastEmittedSample = metric_utils.queryLastEmittedSampleDatetime(key) self.assertEqual(now, lastEmittedSample) self.assertLess(result, lastEmittedSample)
def testEmittedSampleDatetime(self): key = "bogus-test-key" # Establish initial sample datetime result = metric_utils.establishLastEmittedSampleDatetime(key, 300) # Cleanup self.addCleanup( collectorsdb.engineFactory().execute, schema.emittedSampleTracker.delete().where( # pylint: disable=E1120 (schema.emittedSampleTracker.c.key == key))) self.assertIsInstance(result, datetime) # Update latest emitted sample datetime to now now = datetime.utcnow().replace(microsecond=0) metric_utils.updateLastEmittedSampleDatetime(key, now) # Verify that it was updated lastEmittedSample = metric_utils.queryLastEmittedSampleDatetime(key) self.assertEqual(now, lastEmittedSample) self.assertLess(result, lastEmittedSample)
def testUpdateLastEmittedSampleDatetime(self, collectorsdbMock): collectorsdbMock.engineFactory.return_value = Mock( spec_set=sqlalchemy.engine.Engine) metric_utils.updateLastEmittedSampleDatetime("twitter-tweets-volume", datetime.utcnow()) args, _ = (collectorsdbMock.engineFactory.return_value.execute. call_args_list[0]) self.assertTrue(args) self.assertIsInstance(args[0], sqlalchemy.sql.dml.Update)
def testUpdateLastEmittedSampleDatetime(self, collectorsdbMock): collectorsdbMock.engineFactory.return_value = Mock( spec_set=sqlalchemy.engine.Engine) metric_utils.updateLastEmittedSampleDatetime("twitter-tweets-volume", datetime.utcnow()) args, _ = (collectorsdbMock .engineFactory .return_value .execute .call_args_list[0]) self.assertTrue(args) self.assertIsInstance(args[0], sqlalchemy.sql.dml.Update)