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)