Removes use of timeutils.set_time_override

The set_time_override function in timeutils was written as a
helper function to mock utcnow for unittests before 'mock' was
generally used. Now that we have mock and fixture, we no longer
need to use it.

Change-Id: I8106c27dd839e71fa0e5140f77b098a8872999b8
Closes-Bug:#1266962
This commit is contained in:
zhangyangyang 2017-09-17 22:15:00 +08:00
parent 481e6151af
commit b263fb17b7
1 changed files with 16 additions and 12 deletions

View File

@ -949,7 +949,7 @@ class BasePipelineTestCase(base.BaseTestCase):
] ]
self._set_pipeline_cfg('transformers', transformer_cfg) self._set_pipeline_cfg('transformers', transformer_cfg)
self._set_pipeline_cfg('meters', ['cpu']) self._set_pipeline_cfg('meters', ['cpu'])
now = timeutils.utcnow() now = datetime.datetime.utcnow()
later = now + datetime.timedelta(minutes=offset) later = now + datetime.timedelta(minutes=offset)
um = {'autoscaling_weight': weight} if weight else {} um = {'autoscaling_weight': weight} if weight else {}
counters = [ counters = [
@ -1073,7 +1073,7 @@ class BasePipelineTestCase(base.BaseTestCase):
] ]
self._set_pipeline_cfg('transformers', transformer_cfg) self._set_pipeline_cfg('transformers', transformer_cfg)
self._set_pipeline_cfg('meters', ['cpu']) self._set_pipeline_cfg('meters', ['cpu'])
now = timeutils.utcnow() now = datetime.datetime.utcnow()
counters = [ counters = [
sample.Sample( sample.Sample(
name='cpu', name='cpu',
@ -1119,7 +1119,7 @@ class BasePipelineTestCase(base.BaseTestCase):
self.cfg2file(self.pipeline_cfg), self.transformer_manager) self.cfg2file(self.pipeline_cfg), self.transformer_manager)
pipe = pipeline_manager.pipelines[0] pipe = pipeline_manager.pipelines[0]
now = timeutils.utcnow() now = datetime.datetime.utcnow()
now_time = monotonic.monotonic() now_time = monotonic.monotonic()
# Simulate a laggy poller # Simulate a laggy poller
later = now + datetime.timedelta(seconds=12345) later = now + datetime.timedelta(seconds=12345)
@ -1181,7 +1181,7 @@ class BasePipelineTestCase(base.BaseTestCase):
self.cfg2file(self.pipeline_cfg), self.transformer_manager) self.cfg2file(self.pipeline_cfg), self.transformer_manager)
pipe = pipeline_manager.pipelines[0] pipe = pipeline_manager.pipelines[0]
now = timeutils.utcnow() now = datetime.datetime.utcnow()
later = now + datetime.timedelta(seconds=10) later = now + datetime.timedelta(seconds=10)
rounding = 12345 rounding = 12345
@ -1239,7 +1239,7 @@ class BasePipelineTestCase(base.BaseTestCase):
self.cfg2file(self.pipeline_cfg), self.transformer_manager) self.cfg2file(self.pipeline_cfg), self.transformer_manager)
pipe = pipeline_manager.pipelines[0] pipe = pipeline_manager.pipelines[0]
now = timeutils.utcnow() now = datetime.datetime.utcnow()
earlier = now - datetime.timedelta(seconds=10) earlier = now - datetime.timedelta(seconds=10)
later = now + datetime.timedelta(seconds=10) later = now + datetime.timedelta(seconds=10)
@ -1293,7 +1293,7 @@ class BasePipelineTestCase(base.BaseTestCase):
) )
def _do_test_rate_of_change_mapping(self, pipe, meters, units): def _do_test_rate_of_change_mapping(self, pipe, meters, units):
now = timeutils.utcnow() now = datetime.datetime.utcnow()
base = 1000 base = 1000
offset = 7 offset = 7
rate = 42 rate = 42
@ -1689,8 +1689,10 @@ class BasePipelineTestCase(base.BaseTestCase):
publisher = pipe.publishers[0] publisher = pipe.publishers[0]
self.assertEqual(2, len(publisher.samples)) self.assertEqual(2, len(publisher.samples))
def test_aggregator_timed_flush(self): @mock.patch.object(timeutils, 'utcnow')
timeutils.set_time_override() def test_aggregator_timed_flush(self, mock_utcnow):
now = datetime.datetime.utcnow()
mock_utcnow.return_value = now
transformer_cfg = [ transformer_cfg = [
{ {
'name': 'aggregator', 'name': 'aggregator',
@ -1722,7 +1724,7 @@ class BasePipelineTestCase(base.BaseTestCase):
publisher = pipeline_manager.pipelines[0].publishers[0] publisher = pipeline_manager.pipelines[0].publishers[0]
self.assertEqual(0, len(publisher.samples)) self.assertEqual(0, len(publisher.samples))
timeutils.advance_time_seconds(120) mock_utcnow.return_value = now + datetime.timedelta(seconds=120)
pipe.flush() pipe.flush()
publisher = pipeline_manager.pipelines[0].publishers[0] publisher = pipeline_manager.pipelines[0].publishers[0]
self.assertEqual(1, len(publisher.samples)) self.assertEqual(1, len(publisher.samples))
@ -2048,8 +2050,10 @@ class BasePipelineTestCase(base.BaseTestCase):
self.assertEqual(2, len(publisher.samples)) self.assertEqual(2, len(publisher.samples))
self.assertEqual(2050.0, publisher.samples[1].volume) self.assertEqual(2050.0, publisher.samples[1].volume)
def test_aggregator_timed_flush_no_matching_samples(self): @mock.patch.object(timeutils, 'utcnow')
timeutils.set_time_override() def test_aggregator_timed_flush_no_matching_samples(self, mock_utcnow):
now = datetime.datetime.utcnow()
mock_utcnow.return_value = now
transformer_cfg = [ transformer_cfg = [
{ {
'name': 'aggregator', 'name': 'aggregator',
@ -2061,7 +2065,7 @@ class BasePipelineTestCase(base.BaseTestCase):
pipeline_manager = pipeline.PipelineManager( pipeline_manager = pipeline.PipelineManager(
self.CONF, self.CONF,
self.cfg2file(self.pipeline_cfg), self.transformer_manager) self.cfg2file(self.pipeline_cfg), self.transformer_manager)
timeutils.advance_time_seconds(200) mock_utcnow.return_value = now + datetime.timedelta(seconds=200)
pipe = pipeline_manager.pipelines[0] pipe = pipeline_manager.pipelines[0]
pipe.flush() pipe.flush()
publisher = pipeline_manager.pipelines[0].publishers[0] publisher = pipeline_manager.pipelines[0].publishers[0]