def test_generate_content_list(self):
content_pair_list = Helper.generate_content_list_from_file('D:\Download\installed')
for content_pair in content_pair_list:
assert len(content_pair) == 2
assert type(content_pair) == list
assert type(content_pair[0]) == str
assert type(content_pair[1]) == list
def test_generate_content_list(self):
content_pair_list = Helper.generate_content_list_from_file('D:\Download\installed')
for content_pair in content_pair_list:
self.assertEqual(len(content_pair), 2)
self.assertEqual(type(content_pair), list)
self.assertEqual(type(content_pair[0]), str)
self.assertEqual(type(content_pair[1]), list)
def help_generate_collection_with_mock(self, data_source: str):
if data_source == 'file':
Helper.generate_content_list_from_file = MagicMock(return_value=self.mock_content_list)
elif data_source == 'db':
Helper.generate_content_list_from_db = MagicMock(return_value=self.mock_content_list)
collection = Helper.generate_collection(data_source, 'mocked pass')
assert collection.get_container(-1) is None
container = collection.get_container(0)
assert container.file_name == 'mocked pass\calc.py'
assert container.num_of_class == 2
assert container.num_of_line == 26
assert container.num_of_method == 4
container = collection.get_container(1)
assert container.file_name == 'mocked pass\HandPy.py'
assert container.num_of_class == 0
assert container.num_of_line == 7
assert container.num_of_method == 0
container = collection.get_container(2)
assert container.file_name == 'mocked pass\py_calc_test.py'
assert container.num_of_class == 1
assert container.num_of_line == 10
assert container.num_of_method == 1
assert collection.get_container(3) is None
def help_generate_collection_with_mock(self, data_source: str):
if data_source == 'file':
Helper.generate_content_list_from_file = MagicMock(return_value=self.mock_content_list)
elif data_source == 'db':
Helper.generate_content_list_from_db = MagicMock(return_value=self.mock_content_list)
collection = Helper.generate_collection(data_source, 'mocked pass')
self.assertEqual(collection.get_container(-1), None)
container = collection.get_container(0)
self.assertEqual(container.file_name, 'mocked pass\calc.py')
self.assertEqual(container.num_of_class, 2)
self.assertEqual(container.num_of_line, 26)
self.assertEqual(container.num_of_method, 4)
container = collection.get_container(1)
self.assertEqual(container.file_name, 'mocked pass\HandPy.py')
self.assertEqual(container.num_of_class, 0)
self.assertEqual(container.num_of_line, 7)
self.assertEqual(container.num_of_method, 0)
container = collection.get_container(2)
self.assertEqual(container.file_name, 'mocked pass\py_calc_test.py')
self.assertEqual(container.num_of_class, 1)
self.assertEqual(container.num_of_line, 10)
self.assertEqual(container.num_of_method, 1)
self.assertEqual(collection.get_container(3), None)
def test_generate_container_list(self):
content_pair_list = Helper.generate_content_list_from_file('D:\Download\installed')
container_list = Helper.generate_container_list(content_pair_list)
assert len(container_list) == 3
container = container_list[0]
assert container.file_name == 'D:\Download\installed\calc.py'
assert container.num_of_class == 2
assert container.num_of_line == 26
assert container.num_of_method == 4
container = container_list[1]
assert container.file_name == 'D:\Download\installed\HandPy.py'
assert container.num_of_class == 0
assert container.num_of_line == 7
assert container.num_of_method == 0
container = container_list[2]
assert container.file_name == 'D:\Download\installed\py_calc_test.py'
assert container.num_of_class == 1
assert container.num_of_line == 10
assert container.num_of_method == 1
def test_generate_container_list(self):
content_pair_list = Helper.generate_content_list_from_file('D:\Download\installed')
container_list = Helper.generate_container_list(content_pair_list)
self.assertEqual(len(container_list), 3)
container = container_list[0]
self.assertEqual(container.file_name, 'D:\Download\installed\calc.py')
self.assertEqual(container.num_of_class, 2)
self.assertEqual(container.num_of_line, 26)
self.assertEqual(container.num_of_method, 4)
container = container_list[1]
self.assertEqual(container.file_name, 'D:\Download\installed\HandPy.py')
self.assertEqual(container.num_of_class, 0)
self.assertEqual(container.num_of_line, 7)
self.assertEqual(container.num_of_method, 0)
container = container_list[2]
self.assertEqual(container.file_name, 'D:\Download\installed\py_calc_test.py')
self.assertEqual(container.num_of_class, 1)
self.assertEqual(container.num_of_line, 10)
self.assertEqual(container.num_of_method, 1)
def help_generate_collection(self, data_source: str):
collection = Helper.generate_collection(data_source, 'D:\Download\installed')
assert collection.get_container(-1) is None
container = collection.get_container(0)
assert container.file_name == 'D:\Download\installed\calc.py'
assert container.num_of_class == 2
assert container.num_of_line == 26
assert container.num_of_method == 4
container = collection.get_container(1)
assert container.file_name == 'D:\Download\installed\HandPy.py'
assert container.num_of_class == 0
assert container.num_of_line == 7
assert container.num_of_method == 0
container = collection.get_container(2)
assert container.file_name == 'D:\Download\installed\py_calc_test.py'
assert container.num_of_class == 1
assert container.num_of_line == 10
assert container.num_of_method == 1
assert collection.get_container(3) is None
def help_generate_collection(self, data_source: str):
collection = Helper.generate_collection(data_source, 'D:\Download\installed')
self.assertEqual(collection.get_container(-1), None)
container = collection.get_container(0)
self.assertEqual(container.file_name, 'D:\Download\installed\calc.py')
self.assertEqual(container.num_of_class, 2)
self.assertEqual(container.num_of_line, 26)
self.assertEqual(container.num_of_method, 4)
container = collection.get_container(1)
self.assertEqual(container.file_name, 'D:\Download\installed\HandPy.py')
self.assertEqual(container.num_of_class, 0)
self.assertEqual(container.num_of_line, 7)
self.assertEqual(container.num_of_method, 0)
container = collection.get_container(2)
self.assertEqual(container.file_name, 'D:\Download\installed\py_calc_test.py')
self.assertEqual(container.num_of_class, 1)
self.assertEqual(container.num_of_line, 10)
self.assertEqual(container.num_of_method, 1)
self.assertEqual(collection.get_container(3), None)
def step_impl(context):
context.collection = Helper.generate_collection('file', '')
def step_impl(context):
container_list = Helper.generate_container_list(context.content_list)
context.collection = FileInfoCollection(container_list)
def execute(data_source: str, dir_path: str):
collection = Helper.generate_collection(data_source, dir_path)
collection.display_collection_info()