def test_validate_options(self):
class DummyCommand(Command):
@classmethod
def supported_options(cls):
return set(['--foo', '--bar'])
env = StubEnvironment()
# No options should not raise any error
env.argument = ArgumentParser.parse(['googkit.py'])
cmd = DummyCommand(env)
cmd._validate_options()
# Supported option should not raise any error
env.argument = ArgumentParser.parse(['googkit.py', '--foo'])
cmd = DummyCommand(env)
cmd._validate_options()
# Unsupported option
env.argument = ArgumentParser.parse(['googkit.py', '--foo', '--blue-rose'])
cmd = DummyCommand(env)
with self.assertRaises(GoogkitError):
cmd._validate_options()
def test_run(self):
class DummyCommand(Command):
pass
env = StubEnvironment()
env.argument = Argument()
cmd = DummyCommand(env)
cmd._setup = mock.MagicMock()
cmd.run_internal = mock.MagicMock()
with mock.patch('sys.stdout', new_callable=StubStdout):
cmd.run()
self.assertTrue(cmd._setup.called)
self.assertTrue(cmd.run_internal.called)