def setUp(self):
"""Setup test suite."""
self.parser = argparse.ArgumentParser(prog="yoda_test")
self.subparser = self.parser.add_subparsers(dest="jump_subcommand")
self.sandbox = Sandbox()
self.config = Config(self.sandbox.path + "/config")
self.jump = Jump()
self.jump.setup("jump", self.config, self.subparser)
def setUp(self):
"""Setup test suite."""
self.subcommand = Jump()
self.subcommand_str = "jump"
super(TestSubcommandJump, self).setUp()
class TestSubcommandJump(unittest.TestCase):
"""Jump subcommand test suite."""
config = None
sandbox = None
parser = None
subparser = None
jump = None
def setUp(self):
"""Setup test suite."""
self.parser = argparse.ArgumentParser(prog="yoda_test")
self.subparser = self.parser.add_subparsers(dest="jump_subcommand")
self.sandbox = Sandbox()
self.config = Config(self.sandbox.path + "/config")
self.jump = Jump()
self.jump.setup("jump", self.config, self.subparser)
def tearDown(self):
"""Tear down test suite."""
self.sandbox.destroy()
self.parser = None
self.jump = None
def test_parse_jump(self):
"""Test jump to workspace."""
self.jump.parse()
args = self.parser.parse_args(["jump", "yoda"])
self.assertEqual("jump", args.jump_subcommand)
self.assertEqual("yoda", args.name)
def test_exec_jump(self):
"""Test exec jump subcommand."""
args = Mock()
args.name = "yoda/baz"
self.jump._Jump__jump = Mock()
mock_path = {"yoda/baz": "/tmp/yoda/baz"}
with patch("yoda.subcommand.jump.find_path",
return_value=mock_path) as patch_fp:
self.jump.execute(args)
patch_fp.assert_called_once_with(
"yoda/baz", self.jump.config, True)
self.jump._Jump__jump.assert_called_once_with("/tmp/yoda/baz")
def test_exec_jump_no_matches(self):
"""Test exec jump subcommand when no matches."""
args = Mock()
args.name = "foo/bar"
self.jump.logger = Mock()
with patch("yoda.subcommand.jump.find_path", return_value={}):
self.assertFalse(self.jump.execute(args))
self.jump.logger.error.assert_called_once_with(
"No matches for `foo/bar`")
def test_exec_jump_method(self):
"""Test exec jump subcommand when no matches."""
args = Mock()
args.name = "foo/bar"
self.jump.logger = Mock()
os_system = Mock()
with patch("os.system", return_value=os_system):
mock_path = {"yoda/baz": "/tmp/yoda/baz"}
with patch("yoda.subcommand.jump.find_path",
return_value=mock_path):
self.jump.execute(args)
os_system.assert_called_once
calls = [
call("Spawn new shell on `/tmp/yoda/baz`"),
call("Use Ctrl-D to exit and go "
"back to the previous directory"),
call("Shell on `/tmp/yoda/baz` closed.")]
assert self.jump.logger.info.mock_calls == calls
class TestSubcommandJump(SubcommandTestHelper):
"""Jump subcommand test suite."""
def setUp(self):
"""Setup test suite."""
self.subcommand = Jump()
self.subcommand_str = "jump"
super(TestSubcommandJump, self).setUp()
def test_parse_jump(self):
"""Test jump to workspace."""
self.assert_subcommand_parsing(["jump", "yoda"], {
"subcommand": "jump",
"name": "yoda"})
def test_exec_jump(self):
"""Test exec jump subcommand."""
args = Mock()
args.name = "yoda/baz"
self.subcommand._Jump__jump = Mock()
mock_path = {"yoda/baz": "/tmp/yoda/baz"}
with patch("yoda.subcommand.jump.find_path",
return_value=mock_path) as patch_fp:
self.subcommand.execute(args)
patch_fp.assert_called_once_with(
"yoda/baz", self.subcommand.config, True)
self.subcommand._Jump__jump.assert_called_once_with(
"/tmp/yoda/baz")
def test_exec_jump_no_matches(self):
"""Test exec jump subcommand when no matches."""
args = Mock()
args.name = "foo/bar"
with patch("yoda.subcommand.jump.find_path", return_value={}):
with LogCapture() as lcap:
self.assertFalse(self.subcommand.execute(args))
lcap.check(("yoda.subcommand.jump", "ERROR",
"No matches for `foo/bar`"))
def test_exec_jump_method(self):
"""Test exec jump subcommand when no matches."""
args = Mock()
args.name = "foo/bar"
os_system = Mock()
with patch("os.system", return_value=os_system):
mock_path = {"yoda/baz": "/tmp/yoda/baz"}
with patch("yoda.subcommand.jump.find_path",
return_value=mock_path):
with LogCapture() as lcap:
self.subcommand.execute(args)
lcap.check(("yoda.subcommand.jump", "INFO",
"Spawn new shell on `/tmp/yoda/baz`"),
("yoda.subcommand.jump", "INFO",
"Use Ctrl-D to exit and go "
"back to the previous directory"),
("yoda.subcommand.jump", "INFO",
"Shell on `/tmp/yoda/baz` closed."))
os_system.assert_called_once
