mirror of
https://github.com/microsoft/debugpy.git
synced 2025-12-23 08:48:12 +00:00
150 lines
4 KiB
Python
150 lines
4 KiB
Python
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License. See LICENSE in the project root
|
|
# for license information.
|
|
|
|
import pickle
|
|
import pytest
|
|
import subprocess
|
|
import sys
|
|
|
|
from debugpy.common import log
|
|
from tests.patterns import some
|
|
|
|
|
|
@pytest.fixture
|
|
def cli(pyfile):
|
|
@pyfile
|
|
def cli_parser():
|
|
import os
|
|
import pickle
|
|
import sys
|
|
from debugpy.server import cli
|
|
|
|
try:
|
|
cli.parse_argv()
|
|
except Exception as exc:
|
|
os.write(1, pickle.dumps(exc))
|
|
sys.exit(1)
|
|
|
|
# Check that sys.argv has the correct type after parsing - there should be no bytes.
|
|
assert all(isinstance(s, str) for s in sys.argv)
|
|
|
|
# We only care about options that correspond to public switches.
|
|
options = {
|
|
name: getattr(cli.options, name)
|
|
for name in [
|
|
"address",
|
|
"config",
|
|
"log_to",
|
|
"log_to_stderr",
|
|
"mode",
|
|
"target",
|
|
"target_kind",
|
|
"wait_for_client",
|
|
]
|
|
}
|
|
os.write(1, pickle.dumps([sys.argv[1:], options]))
|
|
|
|
def parse(args):
|
|
log.debug("Parsing argv: {0!r}", args)
|
|
try:
|
|
output = subprocess.check_output(
|
|
[sys.executable, "-u", cli_parser.strpath] + args
|
|
)
|
|
argv, options = pickle.loads(output)
|
|
except subprocess.CalledProcessError as exc:
|
|
raise pickle.loads(exc.output)
|
|
|
|
log.debug("Adjusted sys.argv: {0!r}", argv)
|
|
log.debug("Parsed options: {0!r}", options)
|
|
return argv, options
|
|
|
|
return parse
|
|
|
|
|
|
@pytest.mark.parametrize("target_kind", ["file", "module", "code"])
|
|
@pytest.mark.parametrize("mode", ["listen", "connect"])
|
|
@pytest.mark.parametrize("address", ["8888", "localhost:8888"])
|
|
@pytest.mark.parametrize("wait_for_client", ["", "wait_for_client"])
|
|
@pytest.mark.parametrize("script_args", ["", "script_args"])
|
|
def test_targets(cli, target_kind, mode, address, wait_for_client, script_args):
|
|
expected_options = {
|
|
"mode": mode,
|
|
"target_kind": target_kind,
|
|
"wait_for_client": bool(wait_for_client),
|
|
}
|
|
|
|
args = ["--" + mode, address]
|
|
|
|
host, sep, port = address.partition(":")
|
|
if sep:
|
|
expected_options["address"] = (host, int(port))
|
|
else:
|
|
expected_options["address"] = ("127.0.0.1", int(address))
|
|
|
|
if wait_for_client:
|
|
args += ["--wait-for-client"]
|
|
|
|
if target_kind == "file":
|
|
target = "spam.py"
|
|
args += [target]
|
|
elif target_kind == "module":
|
|
target = "spam"
|
|
args += ["-m", target]
|
|
elif target_kind == "code":
|
|
target = "123"
|
|
args += ["-c", target]
|
|
else:
|
|
pytest.fail(target_kind)
|
|
expected_options["target"] = target
|
|
|
|
if script_args:
|
|
script_args = [
|
|
"ham",
|
|
"--listen",
|
|
"--wait-for-client",
|
|
"-y",
|
|
"spam",
|
|
"--",
|
|
"--connect",
|
|
"-c",
|
|
"--something",
|
|
"-m",
|
|
]
|
|
args += script_args
|
|
else:
|
|
script_args = []
|
|
|
|
argv, options = cli(args)
|
|
assert argv == script_args
|
|
assert options == some.dict.containing(expected_options)
|
|
|
|
|
|
@pytest.mark.parametrize("value", ["", True, False])
|
|
def test_configure_subProcess(cli, value):
|
|
args = ["--listen", "8888"]
|
|
|
|
if value == "":
|
|
value = True
|
|
else:
|
|
args += ["--configure-subProcess", str(value)]
|
|
|
|
args += ["spam.py"]
|
|
_, options = cli(args)
|
|
|
|
assert options["config"]["subProcess"] == value
|
|
|
|
|
|
def test_unsupported_switch(cli):
|
|
with pytest.raises(Exception):
|
|
cli(["--listen", "8888", "--xyz", "123", "spam.py"])
|
|
|
|
|
|
def test_unsupported_configure(cli):
|
|
with pytest.raises(Exception):
|
|
cli(["--connect", "127.0.0.1:8888", "--configure-xyz", "123", "spam.py"])
|
|
|
|
|
|
def test_address_required(cli):
|
|
with pytest.raises(Exception):
|
|
cli(["-m", "spam"])
|