Merge branch 'restart'
This commit is contained in:
commit
bded01d3cc
|
@ -98,9 +98,13 @@ def start(cfg, signal_error, no_daemon, blur_usage, advertise_version,
|
||||||
"--signal-error", is_flag=True,
|
"--signal-error", is_flag=True,
|
||||||
help="force all clients to fail with a message",
|
help="force all clients to fail with a message",
|
||||||
)
|
)
|
||||||
|
@click.option(
|
||||||
|
"--disallow-list", is_flag=True,
|
||||||
|
help="never send list of allocated nameplates",
|
||||||
|
)
|
||||||
@click.pass_obj
|
@click.pass_obj
|
||||||
def restart(cfg, signal_error, no_daemon, blur_usage, advertise_version,
|
def restart(cfg, signal_error, no_daemon, blur_usage, advertise_version,
|
||||||
transit, rendezvous):
|
transit, rendezvous, disallow_list):
|
||||||
"""
|
"""
|
||||||
Re-start a relay server
|
Re-start a relay server
|
||||||
"""
|
"""
|
||||||
|
@ -111,6 +115,7 @@ def restart(cfg, signal_error, no_daemon, blur_usage, advertise_version,
|
||||||
cfg.transit = str(transit)
|
cfg.transit = str(transit)
|
||||||
cfg.rendezvous = str(rendezvous)
|
cfg.rendezvous = str(rendezvous)
|
||||||
cfg.signal_error = signal_error
|
cfg.signal_error = signal_error
|
||||||
|
cfg.allow_list = not disallow_list
|
||||||
|
|
||||||
restart_server(cfg)
|
restart_server(cfg)
|
||||||
|
|
||||||
|
|
|
@ -1195,3 +1195,18 @@ class Server(unittest.TestCase):
|
||||||
plugin = MyPlugin(cfg)
|
plugin = MyPlugin(cfg)
|
||||||
relay = plugin.makeService(None)
|
relay = plugin.makeService(None)
|
||||||
self.assertEqual(False, relay._allow_list)
|
self.assertEqual(False, relay._allow_list)
|
||||||
|
|
||||||
|
@mock.patch("wormhole.server.cmd_server.start_server")
|
||||||
|
def test_start_no_args(self, fake_start_server):
|
||||||
|
result = self.runner.invoke(server, ['start'])
|
||||||
|
self.assertEqual(0, result.exit_code)
|
||||||
|
cfg = fake_start_server.mock_calls[0][1][0]
|
||||||
|
MyPlugin(cfg).makeService(None)
|
||||||
|
|
||||||
|
@mock.patch("wormhole.server.cmd_server.restart_server")
|
||||||
|
def test_restart_no_args(self, fake_start_reserver):
|
||||||
|
result = self.runner.invoke(server, ['restart'])
|
||||||
|
self.assertEqual(0, result.exit_code)
|
||||||
|
cfg = fake_start_reserver.mock_calls[0][1][0]
|
||||||
|
MyPlugin(cfg).makeService(None)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user