def test_instrument_dump(self): enable_test_value = False dump_path_value = "/tmp/test.dump" return_value = [1, 2, 3] class A(object): enable_test = enable_test_value dump_path = dump_path_value def __init__(self): self.state = 1 @test_ut.instrument_dump def meth_test(self): return return_value a = A() a.meth_test() self.should_raise("There was a dump file found in %s. " "It might be leftover from previous test." % dump_path_value, du.pick_load, dump_path_value) A.enable_test = True a.meth_test() dump = du.pick_load(dump_path_value) obs_value = dump[0][1] exp_return = return_value self.assertEqual(obs_value, exp_return, "Return values do not match: %s != %s" % (obs_value, exp_return)) fu.removefile(dump_path_value)
def test_instrument_dump(self): enable_test_value = False dump_path_value = "/tmp/test.dump" return_value = [1, 2, 3] class A(object): enable_test = enable_test_value dump_path = dump_path_value def __init__(self): self.state = 1 @test_ut.instrument_dump def meth_test(self): return return_value a = A() a.meth_test() self.should_raise( "There was a dump file found in %s. " "It might be leftover from previous test." % dump_path_value, du.pick_load, dump_path_value) A.enable_test = True a.meth_test() dump = du.pick_load(dump_path_value) obs_value = dump[0][1] exp_return = return_value self.assertEqual( obs_value, exp_return, "Return values do not match: %s != %s" % (obs_value, exp_return)) fu.removefile(dump_path_value)
def __load_wrapper(self, end): """Attempts to load a dump of an endpoint from a file.""" max_retries = 10 for _ in xrange(max_retries): try: return du.pick_load(const.DUMPS[end]) except: continue return []