1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
from io import StringIO
import pytest
from sshoot import main
from sshoot.manager import ManagerProfileError
@pytest.fixture
def stdout():
yield StringIO()
@pytest.fixture
def stderr():
yield StringIO()
@pytest.fixture
def manager(mocker):
yield mocker.MagicMock()
@pytest.fixture
def script(mocker, stdout, stderr, manager):
mocker.patch.object(main, "Manager").return_value = manager
script = main.Sshoot(stdout=stdout, stderr=stderr)
yield script
@pytest.fixture
def sys_exit(mocker):
return mocker.patch("sys.exit")
class TestSshoot:
def test_load_error(self, script, manager, stderr, sys_exit):
"""If config load fails, an error is returned."""
manager.load_config.side_effect = IOError("fail!")
script(["start", "profile1"])
sys_exit.assert_called_once_with(3)
assert stderr.getvalue() == "fail!\n"
def test_profile_error(self, script, manager, stderr, sys_exit):
"""If profile load fails, an error is returned."""
manager.get_profile.side_effect = ManagerProfileError("not found")
script(["show", "profile1"])
sys_exit.assert_called_once_with(2)
assert stderr.getvalue() == "not found\n"
def test_create(self, script, manager):
"""A profile can be added."""
script(
[
"create",
"profile1",
"-r",
"example.net",
"10.10.0.0/16",
"192.168.1.0/24",
]
)
manager.create_profile.assert_called_once_with(
"profile1",
{
"subnets": ["10.10.0.0/16", "192.168.1.0/24"],
"remote": "example.net",
"auto_hosts": False,
"auto_nets": False,
"dns": False,
"exclude_subnets": None,
"seed_hosts": None,
"extra_opts": None,
},
)
def test_show(self, script, manager, stdout):
"""Profile details can be viewed."""
script(["show", "profile1"])
manager.get_profile.assert_called_once_with("profile1")
assert "Name: profile1" in stdout.getvalue()
def test_list(self, script, stdout):
"""Profile list can be viewed."""
script(["list", "--format", "json"])
assert stdout.getvalue() == "{}"
def test_remove(self, script, manager):
"""A profile can be removed."""
script(["delete", "profile1"])
manager.remove_profile.assert_called_once_with("profile1")
def test_start(self, stdout, script, manager):
"""A profile can be started."""
script(
[
"start",
"--no-global-extra-options",
"profile1",
"--",
"--syslog",
]
)
manager.start_profile.assert_called_once_with(
"profile1",
extra_args=["--syslog"],
disable_global_extra_options=True,
)
assert stdout.getvalue() == "Profile started\n"
def test_stop(self, stdout, script, manager):
"""A profile can be stopped."""
script(["stop", "profile1"])
manager.stop_profile.assert_called_once_with("profile1")
assert stdout.getvalue() == "Profile stopped\n"
def test_restart(self, stdout, script, manager):
"""A profile can be restarted."""
script(["restart", "profile1", "--", "--syslog"])
manager.restart_profile.assert_called_once_with(
"profile1",
extra_args=["--syslog"],
disable_global_extra_options=False,
)
assert stdout.getvalue() == "Profile restarted\n"
@pytest.mark.parametrize("running,exit_value", [(True, 0), (False, 1)])
def test_is_running(
self, mocker, sys_exit, script, manager, running, exit_value
):
"""It's possible to check if a profile is running."""
manager.is_running.return_value = running
script(["is-running", "profile1"])
sys_exit.assert_called_once_with(exit_value)
def test_get_command(self, stdout, script, manager):
"""It's possible to get the sshuttle commandline."""
manager.get_cmdline.return_value = ["sshuttle", "-r", "example.net"]
script(["get-command", "profile1"])
manager.get_cmdline.assert_called_once_with(
"profile1", disable_global_extra_options=False
)
assert stdout.getvalue() == "sshuttle -r example.net\n"
|
