args: fix --no-listen, --tor

Also add tests to check that argv is being parsed properly, and to check
the defaults.

fixes #59
This commit is contained in:
Brian Warner 2016-06-26 18:14:07 -07:00
parent 0f8b98dec5
commit 4978be6b90
2 changed files with 144 additions and 7 deletions

View File

@ -91,7 +91,7 @@ class AliasedGroup(click.Group):
help="(debug) don't open a listening socket for Transit",
)
@click.option(
"--tor", is_flag=True, default=True,
"--tor", is_flag=True, default=False,
help="use Tor when connecting",
)
@click.version_option(
@ -99,7 +99,7 @@ class AliasedGroup(click.Group):
version=__version__,
)
@click.pass_context
def wormhole(ctx, tor, no_listen, dump_timing, hide_progress,
def wormhole(context, tor, no_listen, dump_timing, hide_progress,
verify, code_length, transit_helper, relay_url):
"""
Create a Magic Wormhole and communicate through it.
@ -108,10 +108,9 @@ def wormhole(ctx, tor, no_listen, dump_timing, hide_progress,
different places at the same time. Wormholes are secure against
anyone who doesn't use the same code.
"""
ctx.obj = cfg = Config()
ctx.tor = tor
if no_listen:
cfg.listen = False
context.obj = cfg = Config()
cfg.tor = tor
cfg.listen = not no_listen
cfg.relay_url = relay_url
cfg.transit_helper = transit_helper
cfg.code_length = code_length
@ -167,7 +166,7 @@ def _dispatch_command(reactor, cfg, command):
"--text", default=None, metavar="MESSAGE",
help="text message to send, instead of a file. Use '-' to read from stdin.",
)
@click.argument("what", default=u'')
@click.argument("what", required=False)
@click.pass_obj
def send(cfg, what, text, code, zeromode):
"""Send a text message, file, or directory"""

View File

@ -0,0 +1,138 @@
import mock
from twisted.trial import unittest
from ..cli.cli import wormhole
from ..cli.public_relay import RENDEZVOUS_RELAY, TRANSIT_RELAY
from click.testing import CliRunner
#from pprint import pprint
def run(argv):
r = CliRunner()
with mock.patch("wormhole.cli.cli.react") as react:
r.invoke(wormhole, argv)
cfg = react.call_args[0][1][0]
return cfg
class Send(unittest.TestCase):
def test_baseline(self):
cfg = run(["send", "--text", "hi"])
#pprint(cfg.__dict__)
self.assertEqual(cfg.what, None)
self.assertEqual(cfg.code, None)
self.assertEqual(cfg.code_length, 2)
self.assertEqual(cfg.dump_timing, None)
self.assertEqual(cfg.hide_progress, False)
self.assertEqual(cfg.listen, True)
self.assertEqual(cfg.output_file, None)
self.assertEqual(cfg.relay_url, RENDEZVOUS_RELAY)
self.assertEqual(cfg.transit_helper, TRANSIT_RELAY)
self.assertEqual(cfg.text, "hi")
self.assertEqual(cfg.tor, False)
self.assertEqual(cfg.verify, False)
self.assertEqual(cfg.zeromode, False)
def test_file(self):
cfg = run(["send", "fn"])
#pprint(cfg.__dict__)
self.assertEqual(cfg.what, u"fn")
self.assertEqual(cfg.text, None)
def test_text(self):
cfg = run(["send", "--text", "hi"])
self.assertEqual(cfg.what, None)
self.assertEqual(cfg.text, u"hi")
def test_nolisten(self):
cfg = run(["--no-listen", "send", "fn"])
self.assertEqual(cfg.listen, False)
def test_code(self):
cfg = run(["send", "--code", "1-abc", "fn"])
self.assertEqual(cfg.code, u"1-abc")
def test_code_length(self):
cfg = run(["-c", "3", "send", "fn"])
self.assertEqual(cfg.code_length, 3)
def test_dump_timing(self):
cfg = run(["--dump-timing", "tx.json", "send", "fn"])
self.assertEqual(cfg.dump_timing, "tx.json")
def test_hide_progress(self):
cfg = run(["--hide-progress", "send", "fn"])
self.assertEqual(cfg.hide_progress, True)
def test_tor(self):
cfg = run(["--tor", "send", "fn"])
self.assertEqual(cfg.tor, True)
def test_verify(self):
cfg = run(["--verify", "send", "fn"])
self.assertEqual(cfg.verify, True)
def test_zeromode(self):
cfg = run(["send", "-0", "fn"])
self.assertEqual(cfg.zeromode, True)
class Receive(unittest.TestCase):
def test_baseline(self):
cfg = run(["receive"])
#pprint(cfg.__dict__)
self.assertEqual(cfg.accept_file, False)
self.assertEqual(cfg.what, None)
self.assertEqual(cfg.code, None)
self.assertEqual(cfg.code_length, 2)
self.assertEqual(cfg.dump_timing, None)
self.assertEqual(cfg.hide_progress, False)
self.assertEqual(cfg.listen, True)
self.assertEqual(cfg.only_text, False)
self.assertEqual(cfg.output_file, None)
self.assertEqual(cfg.relay_url, RENDEZVOUS_RELAY)
self.assertEqual(cfg.transit_helper, TRANSIT_RELAY)
self.assertEqual(cfg.text, None)
self.assertEqual(cfg.tor, False)
self.assertEqual(cfg.verify, False)
self.assertEqual(cfg.zeromode, False)
def test_nolisten(self):
cfg = run(["--no-listen", "receive"])
self.assertEqual(cfg.listen, False)
def test_code(self):
cfg = run(["receive", "1-abc"])
self.assertEqual(cfg.code, u"1-abc")
def test_code_length(self):
cfg = run(["-c", "3", "receive"])
self.assertEqual(cfg.code_length, 3)
def test_dump_timing(self):
cfg = run(["--dump-timing", "tx.json", "receive"])
self.assertEqual(cfg.dump_timing, "tx.json")
def test_hide_progress(self):
cfg = run(["--hide-progress", "receive"])
self.assertEqual(cfg.hide_progress, True)
def test_tor(self):
cfg = run(["--tor", "receive"])
self.assertEqual(cfg.tor, True)
def test_verify(self):
cfg = run(["--verify", "receive"])
self.assertEqual(cfg.verify, True)
def test_zeromode(self):
cfg = run(["receive", "-0"])
self.assertEqual(cfg.zeromode, True)
def test_only_text(self):
cfg = run(["receive", "-t"])
self.assertEqual(cfg.only_text, True)
def test_accept_file(self):
cfg = run(["receive", "--accept-file"])
self.assertEqual(cfg.accept_file, True)
def test_output_file(self):
cfg = run(["receive", "--output-file", "fn"])
self.assertEqual(cfg.output_file, u"fn")