class TestCase(object):
def __init__(self, max):
self.max = max
@DataProvider(n_3=call(n=3),
n_7=call(n=7),
n_12=call(n=12),
n_20=call(n=20))
def test_test(self, n):
assert n < self.max
def test_data_provider_function(self):
mock = Mock()
function_mock = FunctionMock(mock)
tests = {'call1': call(1, arg=2), 'call2': call(2, arg=3), 'call3':
call(3, arg=4)}
DataProvider(**tests)(function_mock)
loader = TestLoader()
test_suite = list(loader.load_object(function_mock, FakeModule({})))
self.assertEquals(len(test_suite), len(tests))
for test_number, test in enumerate(test_suite):
test()
self.assertTrue(test.id.endswith(tests.keys()[test_number]))
self.assertEquals(mock.call_args_list, tests.values())
def test_data_provider_class(self):
fake_test_case = MockTest()
fake_module = FakeModule({})
tests = {'call1': call(1, arg=2), 'call2': call(2, arg=3), 'call3':
call(3, arg=4)}
DataProvider(**tests)(fake_test_case)
loader = TestLoader()
test_suite = list(loader.load_object(fake_test_case, fake_module))
self.assertEquals(len(test_suite), len(tests))
for test_number, test in enumerate(test_suite):
test()
self.assertIn(tests.keys()[test_number], test.id)
self.assertEqual(fake_test_case.init_mock.call_args_list,
tests.values())
def test_data_provider_method(self):
fake_test_case = MockTest()
fake_module = FakeModule({})
tests = {'call1': call(1, arg=2), 'call2': call(2, arg=3), 'call3':
call(3, arg=4)}
DataProvider(**tests)(fake_test_case.__dict__['test_test'])
loader = TestLoader()
test_suite = list(loader.load_object(fake_test_case, fake_module))
self.assertEquals(len(test_suite), len(tests))
for test_number, test in enumerate(test_suite):
test()
self.assertTrue(test.id.endswith(tests.keys()[test_number]))
self.assertEquals(fake_test_case.set_up_mock.call_count, len(tests))
self.assertEquals(fake_test_case.test_mock.call_args_list, tests.values())
self.assertEquals(fake_test_case.tear_down_mock.call_count, len(tests))
def test_data_provider_function_messages(self):
mock = Mock()
function_mock = FunctionMock(mock)
tests = {'call1': call(1, arg=2), 'call2': call(2, arg=3), 'call3':
call(3, arg=4)}
DataProvider(**tests)(function_mock)
loader = TestLoader()
test_suite = list(loader.load_object(function_mock, FakeModule({})))
self.assertEquals(len(test_suite), len(tests))
# TODO check order
for test_number, test in enumerate(test_suite):
test_call = tests.values()[test_number]
message = test.messages[0][1]
self.assertIn(str(test_call), message)
self.assertIn('Function', message)
test()
self.assertEquals(mock.call_args_list, tests.values())
def test_data_provider_method_messages(self):
fake_test_case = MockTest()
fake_module = FakeModule({})
tests = {'call1': call(1, arg=2), 'call2': call(2, arg=3), 'call3':
call(3, arg=4)}
DataProvider(**tests)(fake_test_case.__dict__['test_test'])
loader = TestLoader()
test_suite = list(loader.load_object(fake_test_case, fake_module))
self.assertEquals(len(test_suite), len(tests))
# TODO check order
for test_number, test in enumerate(test_suite):
test_call = tests.values()[test_number]
message = test.messages[0][1]
self.assertIn(str(test_call), message)
self.assertIn('Method', message)
test()
self.assertEquals(fake_test_case.set_up_mock.call_count, len(tests))
self.assertEquals(fake_test_case.test_mock.call_args_list, tests.values())
self.assertEquals(fake_test_case.tear_down_mock.call_count, len(tests))
def test_call_generator_function(self):
mock = Mock()
function_mock = FunctionMock(mock)
arg_name = 'arg'
test_args = {'call1': 1, 'call2': 2, 'call3': 3}
CallGenerator(arg_name, **test_args)(function_mock)
loader = TestLoader()
test_suite = list(loader.load_object(function_mock, FakeModule({})))
self.assertEquals(len(test_args), len(test_args))
for test in test_suite:
test()
expected = [call(**{arg_name: x}) for x in test_args.values()]
self.assertEquals(mock.call_args_list, expected)
def test_data_provider_class(self):
fake_test_case = MockTest()
fake_module = FakeModule({})
arg_name = 'arg'
test_args = {'call1': 1, 'call2': 2, 'call3': 3}
CallGenerator(arg_name, **test_args)(fake_test_case)
loader = TestLoader()
test_suite = list(loader.load_object(fake_test_case, fake_module))
self.assertEquals(len(test_suite), len(test_args))
for test_number, test in enumerate(test_suite):
test()
self.assertIn(test_args.keys()[test_number], test.id)
expected = [call(**{arg_name: x}) for x in test_args.values()]
self.assertEqual(fake_test_case.init_mock.call_args_list, expected)
def test_call_generator_method(self):
fake_test_case = MockTest()
fake_module = FakeModule({})
arg_name = 'arg'
test_args = {'call1': 1, 'call2': 2, 'call3': 3}
CallGenerator(arg_name, **test_args)(fake_test_case.__dict__['test_test'])
loader = TestLoader()
test_suite = list(loader.load_object(fake_test_case, fake_module))
self.assertEquals(len(test_suite), len(test_args))
for test in test_suite:
test()
self.assertEquals(fake_test_case.set_up_mock.call_count, len(test_args))
expected = [call(**{arg_name: x}) for x in test_args.values()]
self.assertEquals(fake_test_case.test_mock.call_args_list, expected)
self.assertEquals(fake_test_case.tear_down_mock.call_count, len(test_args))
def test_data_provider_class_and_method(self):
fake_test_case = MockTest()
fake_module = FakeModule({})
class_tests = {'call1': call(1, arg=2), 'call2': call(2, arg=3), 'call3':
call(3, arg=4)}
DataProvider(**class_tests)(fake_test_case)
method_tests = {'call4': call(4, arg=5), 'call5': call(5, arg=6),
'call6': call(6, arg=7)}
DataProvider(**method_tests)(fake_test_case.__dict__['test_test'])
loader = TestLoader()
test_suite = list(loader.load_object(fake_test_case, fake_module))
self.assertEquals(len(test_suite), len(class_tests) * len(method_tests))
for test_number, test in enumerate(test_suite):
test()
self.assertIn(class_tests.keys()[test_number/3], test.id)
self.assertTrue(test.id.endswith(method_tests.keys()[(test_number%3)]))
self.assertEqual(fake_test_case.init_mock.call_args_list,
class_tests.values())
from pytf.dataprovider import DataProvider, call
@DataProvider(max_5=call(max=5), max_10=call(max=10), max_15=call(max=15))
class TestCase(object):
def __init__(self, max):
self.max = max
@DataProvider(n_3=call(n=3),
n_7=call(n=7),
n_12=call(n=12),
n_20=call(n=20))
def test_test(self, n):
assert n < self.max
