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'}), ])