def _create_command(self, tasks=[], filter_tasks=[]): mock_context = self._create_mock_context(tasks) mock_filter = mock.Mock() mock_filter.filter_items = mock.MagicMock(return_value=filter_tasks) return listcommand.ListTaskCommand(mock_context, mock_filter)
def test_execute_calls_read_all_on_storage(self): tasks = [] tasks.append(mock.MagicMock()) tasks.append(mock.MagicMock()) mock_context = self._create_mock_context(tasks) mock_filter = mock.Mock() mock_filter.filter_items = mock.MagicMock(return_value=tasks) command = listcommand.ListTaskCommand(mock_context, mock_filter) command.execute() mock_context.storage.read_all.assert_called_once()
def test_execute_filters_tasks(self): tasks = [] tasks.append(mock.Mock()) tasks.append(mock.Mock()) tasks.append(mock.Mock()) tasks[1].index = 2 tasks[1].is_ended = False tasks[1].status = 'this status' tasks[1].name = 'some name' tasks[1].is_ended = False mock_context = self._create_mock_context(tasks) mock_filter = mock.Mock() mock_filter.filter_items = mock.MagicMock(return_value=[tasks[1]]) command = listcommand.ListTaskCommand(mock_context, mock_filter) command.execute() mock_filter.filter_items.assert_called() add_column_call1 = mock.call('ID') add_column_call2 = mock.call('Status') add_column_call3 = mock.call('Description') add_column_calls = [ add_column_call1, add_column_call2, add_column_call3 ] self.assertEqual(add_column_calls, mock_context.mock_table.add_column.mock_calls) add_row_call1 = mock.call(2, 'this status', 'some name') add_row_calls = [add_row_call1] self.assertEqual(add_row_calls, mock_context.mock_table.add_row.mock_calls) mock_context.console.print_lines.assert_called()
def test_constructor_sets_storage(self): mock_context = mock.Mock() mock_filter = mock.Mock() command = listcommand.ListTaskCommand(mock_context, mock_filter) self.assertEqual(mock_context, command.context)
def test_before_execute_calls_garbage_collect_on_storage(self): mock_context = self._create_mock_context([]) mock_filter = mock.Mock() command = listcommand.ListTaskCommand(mock_context, mock_filter) command.before_execute() mock_context.storage.garbage_collect.assert_called_once()