def test_after(self): def after_fn(*a): test_functions.test_list.append(2) with after('before_after.tests.test_functions.sample_fn', after_fn): test_functions.sample_fn(1) self.assertEqual(test_functions.test_list, [1, 2])
def test_before(self): def before_fn(*a): test_functions.test_list.append(1) with before('before_after.tests.test_functions.sample_fn', before_fn): test_functions.sample_fn(2) self.assertEqual(test_functions.test_list, [1, 2])
def test_after_once(self): def after_fn(*a): test_functions.test_list.append(2) with after('before_after.tests.test_functions.sample_fn', after_fn, once=True): test_functions.sample_fn(1) test_functions.sample_fn(3) self.assertEqual(test_functions.test_list, [1, 2, 3])
def test_before_once(self): def before_fn(*a): test_functions.test_list.append(1) with before('before_after.tests.test_functions.sample_fn', before_fn, once=True): test_functions.sample_fn(2) test_functions.sample_fn(3) self.assertEqual(test_functions.test_list, [1, 2, 3])
def test_before_and_after(self): def before_fn(*a): test_functions.test_list.append(1) def after_fn(*a): test_functions.test_list.append(3) with before_after('before_after.tests.test_functions.sample_fn', before_fn=before_fn, after_fn=after_fn): test_functions.sample_fn(2) self.assertEqual(test_functions.test_list, [1, 2, 3])