def test_discover(self):
stub = Stub()
subparsers = StubSubparsers(stub)
parser = StubArgParser(stub)
subparsers.return_add_parser = parser
add_cli_subparser('discover', 'pytest', subparsers)
self.assertEqual(stub.calls, [
('subparsers.add_parser', None, {'name': 'pytest'}),
])
def test_unsupported_command(self):
subparsers = StubSubparsers(name=None)
subparsers.return_add_parser = None
with self.assertRaises(UnsupportedCommandError):
add_cli_subparser('run', 'pytest', subparsers)
with self.assertRaises(UnsupportedCommandError):
add_cli_subparser('debug', 'pytest', subparsers)
with self.assertRaises(UnsupportedCommandError):
add_cli_subparser('???', 'pytest', subparsers)
self.assertEqual(subparsers.calls, [
('add_parser', None, {'name': 'pytest'}),
('add_parser', None, {'name': 'pytest'}),
('add_parser', None, {'name': 'pytest'}),
])