more cleanup of API tests regarding FLAGS

This commit is contained in:
Chris Behrens
2011-08-03 10:11:20 -07:00
parent 4cfd93cd28
commit 37a51f83e6

View File

@@ -100,6 +100,7 @@ class TestCase(unittest.TestCase):
self.injected = []
self._services = []
self._original_flags = FLAGS.FlagValuesDict()
self._original_flags_verbosity = FLAGS.verbose
def tearDown(self):
"""Runs after each test method to tear down test environment."""
@@ -157,6 +158,11 @@ class TestCase(unittest.TestCase):
FLAGS.Reset()
for k, v in self._original_flags.iteritems():
setattr(FLAGS, k, v)
FLAGS.verbose = self._original_flags_verbosity
def set_flags_verbosity(self, verbose=True):
"""Set the FLAGS verbosity for a test"""
FLAGS.verbose = verbose
def start_service(self, name, host=None, **kwargs):
host = host and host or uuid.uuid4().hex