diff options
-rw-r--r-- | nova/tests/test_quota.py | 10 |
1 files changed, 3 insertions, 7 deletions
diff --git a/nova/tests/test_quota.py b/nova/tests/test_quota.py index 6fd206ae6..a5a2539cd 100644 --- a/nova/tests/test_quota.py +++ b/nova/tests/test_quota.py @@ -211,7 +211,7 @@ class QuotaIntegrationTestCase(test.TestCase): self._create_with_injected_files, files) def test_reservation_expire(self): - timeutils.set_time_override() + self.useFixture(test.TimeOverride()) def assertInstancesReserved(reserved): result = quota.QUOTAS.get_project_quotas(self.context, @@ -718,11 +718,7 @@ class DbQuotaDriverTestCase(test.TestCase): self.calls = [] - timeutils.set_time_override() - - def tearDown(self): - timeutils.clear_time_override() - super(DbQuotaDriverTestCase, self).tearDown() + self.useFixture(test.TimeOverride()) def test_get_defaults(self): # Use our pre-defined resources @@ -1422,7 +1418,7 @@ class QuotaReserveSqlAlchemyTestCase(test.TestCase): self.stubs.Set(sqa_api, '_quota_usage_create', fake_quota_usage_create) self.stubs.Set(sqa_api, 'reservation_create', fake_reservation_create) - timeutils.set_time_override() + self.useFixture(test.TimeOverride()) def _make_quota_usage(self, project_id, resource, in_use, reserved, until_refresh, created_at, updated_at): |