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()