def test_process_all_files_processess_all_files_group(self):
groups = [[1, 2], [3, 4], [5, 6]]
self.group_n_elements.return_value = groups
process_all_files(self.files, self.remote_driver)
expected_calls = [
call(groups[0], self.remote_driver),
call(groups[1], self.remote_driver),
call(groups[2], self.remote_driver)
]
self.assertEqual(expected_calls, self.process_file_group.mock_calls)
def test_process_all_files_groups_n_elements(self):
process_all_files(self.files, self.remote_driver)
self.group_n_elements.assert_called_once_with(self.files, 20)
