63 lines
1.8 KiB
Python
63 lines
1.8 KiB
Python
import io
|
|
import pytest
|
|
from unittest import mock
|
|
|
|
from pathod import pathoc_cmdline as cmdline
|
|
|
|
from mitmproxy.test import tutils
|
|
|
|
|
|
@mock.patch("argparse.ArgumentParser.error")
|
|
def test_pathoc(perror):
|
|
assert cmdline.args_pathoc(["pathoc", "foo.com", "get:/"])
|
|
s = io.StringIO()
|
|
with pytest.raises(SystemExit):
|
|
cmdline.args_pathoc(["pathoc", "--show-uas"], s, s)
|
|
|
|
a = cmdline.args_pathoc(["pathoc", "foo.com:8888", "get:/"])
|
|
assert a.port == 8888
|
|
|
|
a = cmdline.args_pathoc(["pathoc", "foo.com:xxx", "get:/"])
|
|
assert perror.called
|
|
perror.reset_mock()
|
|
|
|
a = cmdline.args_pathoc(["pathoc", "-I", "10, 20", "foo.com:8888", "get:/"])
|
|
assert a.ignorecodes == [10, 20]
|
|
|
|
a = cmdline.args_pathoc(["pathoc", "-I", "xx, 20", "foo.com:8888", "get:/"])
|
|
assert perror.called
|
|
perror.reset_mock()
|
|
|
|
a = cmdline.args_pathoc(["pathoc", "-c", "foo:10", "foo.com:8888", "get:/"])
|
|
assert a.connect_to == ["foo", 10]
|
|
|
|
a = cmdline.args_pathoc(["pathoc", "foo.com", "get:/", "--http2"])
|
|
assert a.use_http2 is True
|
|
assert a.ssl is True
|
|
|
|
a = cmdline.args_pathoc(["pathoc", "foo.com", "get:/", "--http2-skip-connection-preface"])
|
|
assert a.use_http2 is True
|
|
assert a.ssl is True
|
|
assert a.http2_skip_connection_preface is True
|
|
|
|
a = cmdline.args_pathoc(["pathoc", "-c", "foo", "foo.com:8888", "get:/"])
|
|
assert perror.called
|
|
perror.reset_mock()
|
|
|
|
a = cmdline.args_pathoc(
|
|
["pathoc", "-c", "foo:bar", "foo.com:8888", "get:/"])
|
|
assert perror.called
|
|
perror.reset_mock()
|
|
|
|
a = cmdline.args_pathoc(
|
|
[
|
|
"pathoc",
|
|
"foo.com:8888",
|
|
tutils.test_data.path("pathod/data/request")
|
|
]
|
|
)
|
|
assert len(list(a.requests)) == 1
|
|
|
|
with pytest.raises(SystemExit):
|
|
cmdline.args_pathoc(["pathoc", "foo.com", "invalid"], s, s)
|