def test_handle_context(self):
with mock.patch('tddtags.core.CompileTags.iterate_child_list', spec=True):
gen = CompileTags(source_module_name='Sample.py')
gen.unit_test_module.append('sample')
gen.unit_test_class.append('SampleTests')
target = imp.load_source('sample', 'tddtags/sample.py')
child_classes = inspect.getmembers(target, inspect.isclass)
clazz = [clazz for name, clazz in child_classes if name == 'ChildSample'][0]
gen.handle_context(target=clazz, parent_context=target)
# --> Expecting 1 unit test items
self.assertEqual(gen.iterate_child_list.call_count, 1)
def test_handle_context_full(self):
gen = CompileTags(source_module_name='Sample.py')
gen.unit_test_module.append('sample')
gen.unit_test_class.append('SampleTests')
target = imp.load_source('sample', 'tddtags/sample.py')
child_classes = inspect.getmembers(target, inspect.isclass)
clazz = [clazz for name, clazz in child_classes if name == 'ChildSample'][0]
gen.handle_context(target=clazz, parent_context=target)
keys = _test_module_details.keys()
self.assertTrue(keys)
module_details = _test_module_details[keys[0]]
self.assertEqual(module_details.module_name, 'sample')
self.assertEqual(len(module_details.class_list), 1)
# print ".class_list: ", module_details.class_list
the_class = module_details.class_list['ChildSampleTests']
self.assertEqual(the_class.class_name, 'ChildSampleTests')
self.assertEqual(the_class.method_names[0], 'eat_more_chocolate')
