def execute_get_confirm_filter(self, with_confirmation):
mock_context = mock.Mock()
mock_context.settings = mock.Mock()
mock_context.settings.command_annotate_confirm = with_confirmation
parser = annotatecommand.AnnotateCommandParser()
return parser.get_confirm_filter(mock_context)
def test_parse_unknown_attribute(self):
args = ['this', 'is', 'a', 'multi-word', 'message', 'invalid:wobbly']
mock_filter = mock.Mock()
mock_filter_factory = mock.Mock()
mock_filter_factory.parse = mock.MagicMock(return_value=mock_filter)
mock_context = mock.Mock()
mock_context.filter_factory = mock_filter_factory
mock_context.settings = mock.Mock()
parser = annotatecommand.AnnotateCommandParser()
with self.assertRaises(Exception):
parser.parse(mock_context, args)
def test_parse_duplicate_attribute(self):
args = ['this', 'is', 'a', 'multi-word', 'message', \
'created:2021-01-01', 'created:2021-01-02']
mock_filter = mock.Mock()
mock_filter_factory = mock.Mock()
mock_filter_factory.parse = mock.MagicMock(return_value=mock_filter)
mock_context = mock.Mock()
mock_context.filter_factory = mock_filter_factory
mock_context.settings = mock.Mock()
parser = annotatecommand.AnnotateCommandParser()
with self.assertRaises(Exception):
parser.parse(mock_context, args)
def test_parse_success(self):
args = ['this', 'is', 'a', 'multi-word', 'message']
mock_filter = mock.Mock()
mock_filter_factory = mock.Mock()
mock_filter_factory.parse = mock.MagicMock(return_value=mock_filter)
mock_context = mock.Mock()
mock_context.filter_factory = mock_filter_factory
mock_context.settings = mock.Mock()
parser = annotatecommand.AnnotateCommandParser()
command = parser.parse(mock_context, args)
self.assertIsInstance(command, annotatecommand.AnnotateCommand)
self.assertEqual(mock_context, command.context)
self.assertEqual(command.message, 'this is a multi-word message')
def test_parse_no_message(self):
mock_context = mock.Mock()
parser = annotatecommand.AnnotateCommandParser()
with self.assertRaises(Exception):
parser.parse(mock_context, [])