This is an automated email from the ASF dual-hosted git repository.
manikumar pushed a commit to branch 2.7
in repository https://gitbox.apache.org/repos/asf/kafka.git
The following commit(s) were added to refs/heads/2.7 by this push:
new 5d8c364 KAFKA-13041: Enable connecting VS Code remote debugger
(#10915)
5d8c364 is described below
commit 5d8c3647126d5914efb7adc47e4a7a9e4d2ab1aa
Author: Stanislav Vodetskyi <[email protected]>
AuthorDate: Thu Jul 8 08:05:14 2021 -0700
KAFKA-13041: Enable connecting VS Code remote debugger (#10915)
The changes in this PR enable connecting VS Code's remote debugger to a
system test running locally with ducker-ak.
Changes include:
- added zip_safe=False to setup.py - this enables installing kafkatest
module together with source code when running `python setup.py
develop/install`.
- install [debugpy](https://github.com/microsoft/debugpy) on ducker nodes
- expose 5678 (default debugpy port) on ducker01 node - ducker01 is the one
that actually executes tests, so that's where you'd connect to.
- added `-d|--debug` option to `ducker-ak test` command - if used, tests
will run via `python3.7 -m debugpy` command, which would listen on 5678 and
pause until debugger is connected.
- changed the logic of the `ducker-ak test` command so that ducktape args
are collected separately after `--` - otherwise any argument we add to the
`test` command in the future might potentially
shadow a similar ducktape argument.
- we don't really check that `ducktape_args` are args while
`test_name_args` are actual test names, so the difference between the two is
minimal actually - most importantly we do check that `test_name_args` is not
empty, but we are ok if `ducktape_args` is.
Reviewers: Ewen Cheslack-Postava <[email protected]>, Manikumar Reddy
<[email protected]>
---
tests/README.md | 34 +++++++++++++++++++++++++++++
tests/docker/Dockerfile | 4 ++--
tests/docker/ducker-ak | 55 ++++++++++++++++++++++++++++++++++++++---------
tests/docker/run_tests.sh | 3 +++
tests/setup.py | 3 ++-
5 files changed, 86 insertions(+), 13 deletions(-)
diff --git a/tests/README.md b/tests/README.md
index d2d9738..c2b756a 100644
--- a/tests/README.md
+++ b/tests/README.md
@@ -48,6 +48,40 @@ bash tests/docker/ducker-ak up -j 'openjdk:11';
tests/docker/run_tests.sh
```
REBUILD="t" bash tests/docker/run_tests.sh
```
+* Debug tests in VS Code:
+ - Run test with `--debug` flag (can be before or after file name):
+ ```
+ tests/docker/ducker-ak up; tests/docker/ducker-ak test
tests/kafkatest/tests/core/security_test.py --debug
+ ```
+ - Test will run in debug mode and wait for a debugger to attach.
+ - Launch VS Code debugger with `"attach"` request - here's an example:
+ ```json
+ {
+ "version": "0.2.0",
+ "configurations": [
+ {
+ "name": "Python: Attach to Ducker",
+ "type": "python",
+ "request": "attach",
+ "connect": {
+ "host": "localhost",
+ "port": 5678
+ },
+ "justMyCode": false,
+ "pathMappings": [
+ {
+ "localRoot": "${workspaceFolder}",
+ "remoteRoot": "."
+ }
+ ]
+ }
+ ]
+ }
+ ```
+ - To pass `--debug` flag to ducktape itself, use `--`:
+ ```
+ tests/docker/ducker-ak test tests/kafkatest/tests/core/security_test.py
--debug -- --debug
+ ```
* Notes
- The scripts to run tests creates and destroys docker network named *knw*.
diff --git a/tests/docker/Dockerfile b/tests/docker/Dockerfile
index 143de9b..21472a7 100644
--- a/tests/docker/Dockerfile
+++ b/tests/docker/Dockerfile
@@ -34,8 +34,8 @@ LABEL ducker.creator=$ducker_creator
# Update Linux and install necessary utilities.
# we have to install git since it is included in openjdk:8 but not openjdk:11
RUN apt update && apt install -y sudo git netcat iptables rsync unzip wget
curl jq coreutils openssh-server net-tools vim python3-pip python3-dev
libffi-dev libssl-dev cmake pkg-config libfuse-dev iperf traceroute && apt-get
-y clean
-RUN python3 -m pip install -U pip==20.2.2;
-RUN pip3 install --upgrade cffi virtualenv pyasn1 boto3 pycrypto pywinrm
ipaddress enum34 && pip3 install --upgrade ducktape==0.8.1
+RUN python3 -m pip install -U pip==21.1.1;
+RUN pip3 install --upgrade cffi virtualenv pyasn1 boto3 pycrypto pywinrm
ipaddress enum34 debugpy && pip3 install --upgrade ducktape==0.8.1
# Set up ssh
COPY ./ssh-config /root/.ssh/config
diff --git a/tests/docker/ducker-ak b/tests/docker/ducker-ak
index 0a7604a..5b6e01d 100755
--- a/tests/docker/ducker-ak
+++ b/tests/docker/ducker-ak
@@ -47,6 +47,9 @@ default_jdk="openjdk:8"
# The default ducker-ak image name.
default_image_name="ducker-ak"
+# Port to listen on when debugging
+debugpy_port=5678
+
# Display a usage message on the terminal and exit.
#
# $1: The exit status to use
@@ -74,11 +77,20 @@ up [-n|--num-nodes NUM_NODES] [-f|--force] [docker-image]
on the host. The argument can be a single port (like 5005), a port range
like (5005-5009)
or a combination of port/port-range separated by comma (like 2181,9092 or
2181,5005-5008).
By default no port is exposed. See README.md for more detail on this
option.
+
+ Note that port 5678 will be automatically exposed for ducker01 node and
will be mapped to 5678
+ on your local machine to enable debugging in VS Code.
-test [test-name(s)]
+test [-d|--debug] [test-name(s)] [-- [ducktape args]]
Run a test or set of tests inside the currently active Ducker nodes.
For example, to run the system test produce_bench_test, you would run:
./tests/docker/ducker-ak test
./tests/kafkatest/tests/core/produce_bench_test.py
+
+ If --debug is passed, the tests will wait for remote VS Code debugger to
connect on port 5678:
+ ./tests/docker/ducker-ak test --debug
./tests/kafkatest/tests/core/produce_bench_test.py
+
+ To pass arguments to underlying ducktape invocation, pass them after `--`,
e.g.:
+ ./tests/docker/ducker-ak test
./tests/kafkatest/tests/core/produce_bench_test.py -- --test-runner-timeout
1800000
ssh [node-name|user-name@node-name] [command]
Log in to a running ducker container. If node-name is not given, it prints
@@ -243,6 +255,7 @@ docker_run() {
local node=${1}
local image_name=${2}
local ports_option=${3}
+ local port_mapping=${4}
local expose_ports=""
if [[ -n ${ports_option} ]]; then
@@ -251,6 +264,9 @@ docker_run() {
expose_ports="${expose_ports} --expose ${expose_port}"
done
fi
+ if [[ -n ${port_mapping} ]]; then
+ expose_ports="${expose_ports} -p ${port_mapping}:${port_mapping}"
+ fi
# Invoke docker-run. We need privileged mode to be able to run iptables
# and mount FUSE filesystems inside the container. We also need it to
@@ -337,7 +353,8 @@ attempting to start new ones."
if [[ -n "${custom_ducktape}" ]]; then
setup_custom_ducktape "${custom_ducktape}" "${image_name}"
fi
- for n in $(seq -f %02g 1 ${num_nodes}); do
+ docker_run ducker01 "${image_name}" "${expose_ports}" "${debugpy_port}"
+ for n in $(seq -f %02g 2 ${num_nodes}); do
local node="ducker${n}"
docker_run "${node}" "${image_name}" "${expose_ports}"
done
@@ -422,23 +439,41 @@ ducker_test() {
require_commands docker
docker inspect ducker01 &>/dev/null || \
die "ducker_test: the ducker01 instance appears to be down. Did you
run 'ducker up'?"
- [[ $# -lt 1 ]] && \
+ declare -a test_name_args=()
+ local debug=0
+ while [[ $# -ge 1 ]]; do
+ case "${1}" in
+ -d|--debug) debug=1; shift;;
+ --) shift; break;;
+ *) test_name_args+=("${1}"); shift;;
+ esac
+ done
+ local ducktape_args=${*}
+
+ [[ ${#test_name_args} -lt 1 ]] && \
die "ducker_test: you must supply at least one system test to run.
Type --help for help."
- local args=""
- local kafka_test=0
- for arg in "${@}"; do
+ local test_names=""
+
+ for test_name in ${test_name_args[*]}; do
local regex=".*\/kafkatest\/(.*)"
- if [[ $arg =~ $regex ]]; then
+ if [[ $test_name =~ $regex ]]; then
local kpath=${BASH_REMATCH[1]}
- args="${args} ./tests/kafkatest/${kpath}"
+ test_names="${test_names} ./tests/kafkatest/${kpath}"
else
- args="${args} ${arg}"
+ test_names="${test_names} ${test_name}"
fi
done
+
must_pushd "${kafka_dir}"
(test -f ./gradlew || gradle) && ./gradlew systemTestLibs
must_popd
- cmd="cd /opt/kafka-dev && ducktape --cluster-file
/opt/kafka-dev/tests/docker/build/cluster.json $args"
+ if [[ "${debug}" -eq 1 ]]; then
+ local ducktape_cmd="python3.7 -m debugpy --listen
0.0.0.0:${debugpy_port} --wait-for-client /usr/local/bin/ducktape"
+ else
+ local ducktape_cmd="ducktape"
+ fi
+
+ cmd="cd /opt/kafka-dev && ${ducktape_cmd} --cluster-file
/opt/kafka-dev/tests/docker/build/cluster.json $test_names $ducktape_args"
echo "docker exec ducker01 bash -c \"${cmd}\""
exec docker exec --user=ducker ducker01 bash -c "${cmd}"
}
diff --git a/tests/docker/run_tests.sh b/tests/docker/run_tests.sh
index 063e24d..0128fd6 100755
--- a/tests/docker/run_tests.sh
+++ b/tests/docker/run_tests.sh
@@ -32,4 +32,7 @@ fi
if ${SCRIPT_DIR}/ducker-ak ssh | grep -q '(none)'; then
${SCRIPT_DIR}/ducker-ak up -n "${KAFKA_NUM_CONTAINERS}" || die "ducker-ak
up failed"
fi
+
+[[ -n ${_DUCKTAPE_OPTIONS} ]] && _DUCKTAPE_OPTIONS="-- ${_DUCKTAPE_OPTIONS}"
+
${SCRIPT_DIR}/ducker-ak test ${TC_PATHS} ${_DUCKTAPE_OPTIONS} || die
"ducker-ak test failed"
diff --git a/tests/setup.py b/tests/setup.py
index ec36bbc..fc85061 100644
--- a/tests/setup.py
+++ b/tests/setup.py
@@ -51,7 +51,8 @@ setup(name="kafkatest",
license="apache2.0",
packages=find_packages(),
include_package_data=True,
- install_requires=["ducktape==0.8.1", "requests==2.24.0"],
+ install_requires=["ducktape==0.8.8", "requests==2.24.0"],
tests_require=["pytest", "mock"],
cmdclass={'test': PyTest},
+ zip_safe=False
)
