| #!/bin/bash |
| # Copyright 2020 The Fuchsia Authors. All rights reserved. |
| # Use of this source code is governed by a BSD-style license that can be |
| # found in the LICENSE file. |
| # |
| # Test that verifies that fserve-remote builds the ssh commands correctly. |
| |
| set -e |
| |
| FSERVE_REMOTE_CMD="${BT_TEMP_DIR}/scripts/sdk/gn/base/bin/fserve-remote.sh" |
| FCONFIG_CMD="${BT_TEMP_DIR}/scripts/sdk/gn/base/bin/fconfig.sh" |
| |
| # Sets up an ssh mock binary on the $PATH of any subshell. |
| set_up_ssh() { |
| export PATH="${BT_TEMP_DIR}/isolated_path_for:${PATH}" |
| |
| SSH_MOCK_PATH="${BT_TEMP_DIR}/isolated_path_for" |
| |
| cat > "${SSH_MOCK_PATH}/ssh.mock_side_effects" <<"EOF" |
| if [[ "$*" =~ SSH_CONNECTION ]]; then |
| echo "172.20.100.10 38618 100.90.250.100 22" |
| elif [[ "$*" =~ "amber_ctl add_src" ]]; then |
| return 0 |
| else |
| echo "$@" |
| fi |
| EOF |
| } |
| |
| # Sets up a device-finder mock. The implemented mock aims to produce minimal |
| # output that parses correctly but is otherwise uninteresting. |
| set_up_device_finder() { |
| cat >"${MOCKED_DEVICE_FINDER}.mock_side_effects" <<"EOF" |
| if [[ "$*" =~ -local ]]; then |
| # Emit a different address than the default so the device and the host can |
| # have different IP addresses. |
| echo fe80::1234%coffee |
| elif [[ "$*" =~ -full ]]; then |
| echo fe80::c0ff:eec0:ffee%coffee coffee-coffee-coffee-coffee |
| else |
| echo fe80::c0ff:eec0:ffee%coffee |
| fi |
| EOF |
| } |
| |
| # Verify that pm serve was run correctly. |
| TEST_fserve_remote() { |
| set_up_ssh |
| set_up_device_finder |
| |
| REMOTE_PATH="/home/path_to_samples/third_party/fuchsia-sdk" |
| |
| # Create a mock that simulates a clean environment, nothing is running. |
| cat > "${SSH_MOCK_PATH}/ssh.mock_side_effects" <<"EOF" |
| # |
| # DO NOT RELY ON mock_state.n files for this SSH. |
| # |
| # There are some calls to SSH that are done in the background, |
| # The mock implementation does not handle these concurrent calls, |
| # so we write out our own state file to capture the parameters. |
| # The names of the state files are hard coded, so if you need |
| # a new call to ssh, you'll need to add the file. |
| # |
| source "$(dirname $0)/../scripts/sdk/gn/bash_tests/gn-bash-test-lib.sh" |
| |
| if [[ "$*" =~ SSH_CONNECTION ]]; then |
| echo "172.20.100.10 38618 100.90.250.100 22" |
| return 0 |
| elif [[ "$*" =~ "amber_ctl add_src" ]]; then |
| return 0 |
| fi |
| |
| # if the check is in quiet mode, return true if there is a tunnel |
| # file, indicating that the mock ssh call that creates the tunnel |
| # has finished. Since it happens in the background, this check avoids |
| # racing conditions and flake tests. |
| if [[ "$*" =~ "-q -O check" ]]; then |
| rc=0 |
| if [[ ! -f "$0.tunnel" ]]; then |
| rc=254 |
| fi |
| gn-test-log-mock "${0}.qcheck" $rc $@ |
| return $rc |
| fi |
| |
| # No existing session. |
| if [[ "$*" =~ "-O check" ]]; then |
| rc=255 |
| gn-test-log-mock "${0}.check" $rc $@ |
| return "${rc}" |
| fi |
| |
| # No existing forwarding for 8022 |
| if [[ "$*" =~ "grep :8022" ]]; then |
| rc=255 |
| gn-test-log-mock "${0}.check_for_8022" $rc $@ |
| return "${rc}" |
| fi |
| |
| |
| # No existing forwarding for 8083 |
| if [[ "$*" =~ "grep :8083" ]]; then |
| rc=2 |
| gn-test-log-mock "${0}.check_for_8083" $rc $@ |
| return "${rc}" |
| fi |
| |
| # log the call to make the tunnel |
| if [[ "$*" =~ "-6 -L" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.tunnel" $rc $@ |
| return "${rc}" |
| fi |
| |
| # log the call to configure the settings |
| if [[ "$*" =~ "fconfig.sh list" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.fconfig" $rc $@ |
| return "${rc}" |
| fi |
| # log the call to configure the settings |
| if [[ "$*" =~ "fserve.sh" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.fserve" $rc $@ |
| return "${rc}" |
| fi |
| echo $@ |
| EOF |
| |
| # Run command. |
| BT_EXPECT "${FSERVE_REMOTE_CMD}" desktop.example.com "${REMOTE_PATH}" --image image1 --ttl 0 > "${BT_TEMP_DIR}/fserve_remote_log.txt" 2>&1 |
| |
| # Common command line args, used for each call to ssh |
| ssh_common_expected=(_ANY_ "desktop.example.com" ) |
| ssh_common_expected+=(-S "${HOME}/.ssh/control-fuchsia-fx-remote") |
| ssh_common_expected+=(-o "ControlMaster=auto") |
| ssh_common_expected+=(-t) |
| |
| # First SSH is to check if the mux session to the host exists |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check.1" |
| expected=("${ssh_common_expected[@]}" -O check) |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # Check for an existing forwarding for port 8022. |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check_for_8022" |
| expected=("${ssh_common_expected[@]}" "ss" "-ln" "|" "grep" ":8022") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| # A loop in fserve-remote checks every second if the tunnel has been |
| # established, every time invocating the ssh mock to check it and thus |
| # creating a new mock_state file for ssh.qcheck. The gn-test-latest-mock |
| # function guarantees that we only check the latest invocation of the mock. |
| source "$(gn-test-latest-mock "${SSH_MOCK_PATH}/ssh.qcheck")" |
| expected=("${ssh_common_expected[@]}" -q -O check) |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.fconfig" |
| expected=("${ssh_common_expected[@]}" "cd" "\$HOME" "&&" |
| "cd" "/home/path_to_samples/third_party/fuchsia-sdk" "&&" |
| "./bin/fconfig.sh" "set" "device-ip" "127.0.0.1" "&&" |
| "./bin/fconfig.sh" "default" "device-name" "&&" |
| "./bin/fconfig.sh" "set" "image" "image1" "&&" |
| "./bin/fconfig.sh" "list" ) |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check_for_8083" |
| expected=("${ssh_common_expected[@]}" "ss" "-ln" "|" "grep" ":8083") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.fserve" |
| expected=("${ssh_common_expected[@]}" ) |
| expected+=("cd" "\$HOME" "&&" |
| "cd" "/home/path_to_samples/third_party/fuchsia-sdk" "&&" "./bin/fserve.sh" "&&" "sleep" "0") |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.tunnel" |
| expected=("${ssh_common_expected[@]}" ) |
| expected+=(-6 -L "\*:8083:localhost:8083" |
| -R "8022:[fe80::c0ff:eec0:ffee%coffee]:22" |
| -R "2345:[fe80::c0ff:eec0:ffee%coffee]:2345" |
| -R "8443:[fe80::c0ff:eec0:ffee%coffee]:8443" |
| -R "9080:[fe80::c0ff:eec0:ffee%coffee]:80" |
| -o "ExitOnForwardFailure=yes" "-nT" "sleep" "0") |
| gn-test-check-mock-args "${expected[@]}" |
| } |
| |
| TEST_fserve_remote_with_config() { |
| set_up_ssh |
| set_up_device_finder |
| |
| REMOTE_PATH="/home/path_to_samples/third_party/fuchsia-sdk" |
| |
| BT_EXPECT "${FCONFIG_CMD}" set image "test-image" |
| BT_EXPECT "${FCONFIG_CMD}" set bucket "custom-bucket" |
| |
| # Create a mock that simulates a clean environment, nothing is running. |
| cat > "${SSH_MOCK_PATH}/ssh.mock_side_effects" <<"EOF" |
| # |
| # DO NOT RELY ON mock_state.n files for this SSH. |
| # |
| # There are some calls to SSH that are done in the background, |
| # The mock implementation does not handle these concurrent calls, |
| # so we write out our own state file to capture the parameters. |
| # The names of the state files are hard coded, so if you need |
| # a new call to ssh, you'll need to add the file. |
| # |
| source "$(dirname $0)/../scripts/sdk/gn/bash_tests/gn-bash-test-lib.sh" |
| |
| if [[ "$*" =~ SSH_CONNECTION ]]; then |
| echo "172.20.100.10 38618 100.90.250.100 22" |
| return 0 |
| elif [[ "$*" =~ "amber_ctl add_src" ]]; then |
| return 0 |
| fi |
| |
| # if the check is in quiet mode, return true if there is a tunnel |
| # file, indicating that the mock ssh call that creates the tunnel |
| # has finished. Since it happens in the background, this check avoids |
| # racing conditions and flake tests. |
| if [[ "$*" =~ "-q -O check" ]]; then |
| rc=0 |
| if [[ ! -f "$0.tunnel" ]]; then |
| rc=254 |
| fi |
| gn-test-log-mock "${0}.qcheck" $rc $@ |
| return $rc |
| fi |
| |
| # No existing session. |
| if [[ "$*" =~ "-O check" ]]; then |
| rc=255 |
| gn-test-log-mock "${0}.check" $rc $@ |
| return $rc |
| fi |
| |
| # No existing forwarding for 8022 |
| if [[ "$*" =~ "grep :8022" ]]; then |
| rc=2 |
| gn-test-log-mock "${0}.check_for_8022" $rc $@ |
| return "${rc}" |
| fi |
| |
| # No existing forwarding for 8083 |
| if [[ "$*" =~ "grep :8083" ]]; then |
| rc=2 |
| gn-test-log-mock "${0}.check_for_8083" $rc $@ |
| return "${rc}" |
| fi |
| |
| # log the call to make the tunnel |
| if [[ "$*" =~ "-6 -L" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.tunnel" $rc $@ |
| return "${rc}" |
| fi |
| |
| # log the call to configure the settings |
| if [[ "$*" =~ "fconfig.sh list" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.fconfig" $rc $@ |
| return "${rc}" |
| fi |
| # log the call to configure the settings |
| if [[ "$*" =~ "fserve.sh" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.fserve" $rc $@ |
| return "${rc}" |
| fi |
| echo $@ |
| EOF |
| |
| |
| # Run command. |
| BT_EXPECT "${FSERVE_REMOTE_CMD}" desktop.example.com "${REMOTE_PATH}" --ttl 0 > "${BT_TEMP_DIR}/fserve_remote_with_config_log.txt" 2>&1 |
| |
| ssh_common_expected=(_ANY_ "desktop.example.com" ) |
| ssh_common_expected+=(-S "${HOME}/.ssh/control-fuchsia-fx-remote") |
| ssh_common_expected+=(-o "ControlMaster=auto") |
| ssh_common_expected+=(-t) |
| |
| # First SSH is to check if the mux session to the host exists |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check.1" |
| expected=("${ssh_common_expected[@]}" -O check) |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # Look for an existing port forwarding for 8022 |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check_for_8022" |
| expected=("${ssh_common_expected[@]}" "ss" "-ln" "|" "grep" ":8022") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| # A loop in fserve-remote checks every second if the tunnel has been |
| # established, every time invocating the ssh mock to check it and thus |
| # creating a new mock_state file for ssh.qcheck. The gn-test-latest-mock |
| # function guarantees that we only check the latest invocation of the mock. |
| source "$(gn-test-latest-mock "${SSH_MOCK_PATH}/ssh.qcheck")" |
| expected=("${ssh_common_expected[@]}" -q -O check) |
| gn-test-check-mock-args "${expected[@]}" |
| |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.fconfig" |
| expected=("${ssh_common_expected[@]}" "cd" "\$HOME" "&&" |
| "cd" "/home/path_to_samples/third_party/fuchsia-sdk" "&&" |
| "./bin/fconfig.sh" "set" "device-ip" "127.0.0.1" "&&" |
| "./bin/fconfig.sh" "default" "device-name" "&&" |
| "./bin/fconfig.sh" "set" "bucket" "custom-bucket" "&&" |
| "./bin/fconfig.sh" "set" "image" "test-image" "&&" |
| "./bin/fconfig.sh" "list") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check_for_8083" |
| expected=("${ssh_common_expected[@]}" "ss" "-ln" "|" "grep" ":8083") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.fserve" |
| expected=("${ssh_common_expected[@]}" ) |
| expected+=("cd" "\$HOME" "&&" "cd" "/home/path_to_samples/third_party/fuchsia-sdk" |
| "&&" "./bin/fserve.sh" "&&" "sleep" "0") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check.2" |
| expected=("${ssh_common_expected[@]}" -O "check") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.tunnel" |
| expected=("${ssh_common_expected[@]}" ) |
| expected+=(-6 -L "\*:8083:localhost:8083" |
| -R "8022:[fe80::c0ff:eec0:ffee%coffee]:22" |
| -R "2345:[fe80::c0ff:eec0:ffee%coffee]:2345" |
| -R "8443:[fe80::c0ff:eec0:ffee%coffee]:8443" |
| -R "9080:[fe80::c0ff:eec0:ffee%coffee]:80" |
| -o "ExitOnForwardFailure=yes" "-nT" "sleep" "0") |
| gn-test-check-mock-args "${expected[@]}" |
| } |
| |
| TEST_fserve_remote_existing_session() { |
| set_up_ssh |
| set_up_device_finder |
| |
| REMOTE_PATH="/home/path_to_samples/third_party/fuchsia-sdk" |
| |
| # Create a mock that simulates fserve-remote is already running. |
| cat > "${SSH_MOCK_PATH}/ssh.mock_side_effects" <<"EOF" |
| # |
| # DO NOT RELY ON mock_state.n files for this SSH. |
| # |
| # There are some calls to SSH that are done in the background, |
| # The mock implementation does not handle these concurrent calls, |
| # so we write out our own state file to capture the parameters. |
| # The names of the state files are hard coded, so if you need |
| # a new call to ssh, you'll need to add the file. |
| # |
| source "$(dirname $0)/../scripts/sdk/gn/bash_tests/gn-bash-test-lib.sh" |
| |
| if [[ "$*" =~ SSH_CONNECTION ]]; then |
| echo "172.20.100.10 38618 100.90.250.100 22" |
| return 0 |
| elif [[ "$*" =~ "amber_ctl add_src" ]]; then |
| return 0 |
| fi |
| |
| # if the check is in quiet mode, return true if there is a tunnel |
| # file, indicating that the mock ssh call that creates the tunnel |
| # has finished. Since it happens in the background, this check avoids |
| # racing conditions and flake tests. |
| if [[ "$*" =~ "-q -O check" ]]; then |
| rc=0 |
| if [[ ! -f "$0.tunnel" ]]; then |
| rc=254 |
| fi |
| gn-test-log-mock "${0}.qcheck" $rc $@ |
| return $rc |
| fi |
| |
| # existing session return 0. |
| if [[ "$*" =~ "-O check" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.check" $rc $@ |
| return $rc |
| fi |
| |
| # existing forwarding for 8022 |
| if [[ "$*" =~ "grep :8022" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.check_for_8022" $rc $@ |
| return "${rc}" |
| fi |
| |
| # No existing forwarding for 8083 |
| if [[ "$*" =~ "grep :8083" ]]; then |
| rc=2 |
| gn-test-log-mock "${0}.check_for_8083" $rc $@ |
| return "${rc}" |
| fi |
| |
| # simulate kill working successfully. |
| if [[ "$*" =~ "pkill -u \$USER sshd" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.pkill" $rc $@ |
| return "${rc}" |
| fi |
| |
| # log the call to make the tunnel |
| if [[ "$*" =~ "-6 -L" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.tunnel" $rc $@ |
| return "${rc}" |
| fi |
| |
| # log the call to configure the settings |
| if [[ "$*" =~ "fconfig.sh list" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.fconfig" $rc $@ |
| return "${rc}" |
| fi |
| # log the call to configure the settings |
| if [[ "$*" =~ "fserve.sh" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.fserve" $rc $@ |
| return "${rc}" |
| fi |
| # log the call to exit |
| if [[ "$*" =~ "-O exit" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.exit" $rc $@ |
| return "${rc}" |
| fi |
| echo $@ |
| EOF |
| |
| # Run command. |
| BT_EXPECT "${FSERVE_REMOTE_CMD}" desktop.example.com "${REMOTE_PATH}" --image image1 --ttl 0 > "${BT_TEMP_DIR}/fserve_remote_log.txt" 2>&1 |
| |
| # Common command line args, used for each call to ssh |
| ssh_common_expected=(_ANY_ "desktop.example.com" ) |
| ssh_common_expected+=(-S "${HOME}/.ssh/control-fuchsia-fx-remote") |
| ssh_common_expected+=(-o "ControlMaster=auto") |
| ssh_common_expected+=(-t) |
| |
| # First SSH is to check if the mux session to the host exists |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check.1" |
| expected=("${ssh_common_expected[@]}" -O check) |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # There is another check before calling exit. |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check.2" |
| expected=("${ssh_common_expected[@]}" -O "check") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.exit.1" |
| expected=("${ssh_common_expected[@]}" -O "exit") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # Check for an existing forwarding for port 8022. |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check_for_8022" |
| expected=("${ssh_common_expected[@]}" "ss" "-ln" "|" "grep" ":8022") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # Kill sshd. |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.pkill" |
| expected=("${ssh_common_expected[@]}" "pkill" "-u" "\$USER" "sshd") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| # A loop in fserve-remote checks every second if the tunnel has been |
| # established, every time invocating the ssh mock to check it and thus |
| # creating a new mock_state file for ssh.qcheck. The gn-test-latest-mock |
| # function guarantees that we only check the latest invocation of the mock. |
| source "$(gn-test-latest-mock "${SSH_MOCK_PATH}/ssh.qcheck")" |
| expected=("${ssh_common_expected[@]}" -q -O check) |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.fconfig" |
| expected=("${ssh_common_expected[@]}" "cd" "\$HOME" "&&" |
| "cd" "/home/path_to_samples/third_party/fuchsia-sdk" "&&" |
| "./bin/fconfig.sh" "set" "device-ip" "127.0.0.1" "&&" |
| "./bin/fconfig.sh" "default" "device-name" "&&" |
| "./bin/fconfig.sh" "set" "image" "image1" "&&" |
| "./bin/fconfig.sh" "list") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check_for_8083" |
| expected=("${ssh_common_expected[@]}" "ss" "-ln" "|" "grep" ":8083") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.fserve" |
| expected=("${ssh_common_expected[@]}" ) |
| expected+=("cd" "\$HOME" "&&" "cd" "/home/path_to_samples/third_party/fuchsia-sdk" |
| "&&" "./bin/fserve.sh" "&&" "sleep" "0") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check.3" |
| expected=("${ssh_common_expected[@]}" -O "check") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.exit.2" |
| expected=("${ssh_common_expected[@]}" -O "exit") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.tunnel" |
| expected=("${ssh_common_expected[@]}" ) |
| expected+=(-6 -L "\*:8083:localhost:8083" |
| -R "8022:[fe80::c0ff:eec0:ffee%coffee]:22" |
| -R "2345:[fe80::c0ff:eec0:ffee%coffee]:2345" |
| -R "8443:[fe80::c0ff:eec0:ffee%coffee]:8443" |
| -R "9080:[fe80::c0ff:eec0:ffee%coffee]:80" |
| -o "ExitOnForwardFailure=yes" "-nT" "sleep" "0") |
| gn-test-check-mock-args "${expected[@]}" |
| } |
| |
| TEST_fserve_remote_pm_running() { |
| set_up_ssh |
| set_up_device_finder |
| |
| REMOTE_PATH="/home/path_to_samples/third_party/fuchsia-sdk" |
| |
| # Create a mock that simulates a clean environment, except pm is running on the remote. |
| cat > "${SSH_MOCK_PATH}/ssh.mock_side_effects" <<"EOF" |
| # |
| # DO NOT RELY ON mock_state.n files for this SSH. |
| # |
| # There are some calls to SSH that are done in the background, |
| # The mock implementation does not handle these concurrent calls, |
| # so we write out our own state file to capture the parameters. |
| # The names of the state files are hard coded, so if you need |
| # a new call to ssh, you'll need to add the file. |
| # |
| source "$(dirname $0)/../scripts/sdk/gn/bash_tests/gn-bash-test-lib.sh" |
| |
| if [[ "$*" =~ SSH_CONNECTION ]]; then |
| echo "172.20.100.10 38618 100.90.250.100 22" |
| return 0 |
| elif [[ "$*" =~ "amber_ctl add_src" ]]; then |
| return 0 |
| fi |
| |
| # if the check is in quiet mode, return true if there is a tunnel |
| # file, indicating that the mock ssh call that creates the tunnel |
| # has finished. Since it happens in the background, this check avoids |
| # racing conditions and flake tests. |
| if [[ "$*" =~ "-q -O check" ]]; then |
| rc=0 |
| if [[ ! -f "$0.tunnel" ]]; then |
| rc=254 |
| fi |
| gn-test-log-mock "${0}.qcheck" $rc $@ |
| return $rc |
| fi |
| |
| # No existing session. |
| if [[ "$*" =~ "-O check" ]]; then |
| rc=255 |
| gn-test-log-mock "${0}.check" $rc $@ |
| return $rc |
| fi |
| |
| # No existing forwarding for 8022 |
| if [[ "$*" =~ "grep :8022" ]]; then |
| rc=2 |
| gn-test-log-mock "${0}.check_for_8022" $rc $@ |
| return "${rc}" |
| fi |
| |
| # Existing forwarding for 8083 |
| if [[ "$*" =~ "grep :8083" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.check_for_8083" $rc $@ |
| return "${rc}" |
| fi |
| |
| # simulate kill working successfully. |
| if [[ "$*" =~ "pkill -u \$USER" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.pkill" $rc $@ |
| return "${rc}" |
| fi |
| |
| # log the call to make the tunnel |
| if [[ "$*" =~ "-6 -L" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.tunnel" $rc $@ |
| return "${rc}" |
| fi |
| |
| # log the call to configure the settings |
| if [[ "$*" =~ "fconfig.sh list" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.fconfig" $rc $@ |
| return "${rc}" |
| fi |
| # log the call to configure the settings |
| if [[ "$*" =~ "fserve.sh" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.fserve" $rc $@ |
| return "${rc}" |
| fi |
| # log the call to exit |
| if [[ "$*" =~ "-O exit" ]]; then |
| rc=0 |
| gn-test-log-mock "${0}.exit" $rc $@ |
| return "${rc}" |
| fi |
| echo $@ |
| EOF |
| |
| # Run command. |
| BT_EXPECT "${FSERVE_REMOTE_CMD}" desktop.example.com "${REMOTE_PATH}" --image image1 --ttl 0 > "${BT_TEMP_DIR}/fserve_remote_log.txt" 2>&1 |
| |
| # Common command line args, used for each call to ssh |
| ssh_common_expected=(_ANY_ "desktop.example.com" ) |
| ssh_common_expected+=(-S "${HOME}/.ssh/control-fuchsia-fx-remote") |
| ssh_common_expected+=(-o "ControlMaster=auto") |
| ssh_common_expected+=(-t) |
| |
| # First SSH is to check if the mux session to the host exists |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check.1" |
| expected=("${ssh_common_expected[@]}" "-O" "check") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| |
| # Check for an existing forwarding for port 8022. |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check_for_8022" |
| expected=("${ssh_common_expected[@]}" "ss" "-ln" "|" "grep" ":8022") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| # A loop in fserve-remote checks every second if the tunnel has been |
| # established, every time invocating the ssh mock to check it and thus |
| # creating a new mock_state file for ssh.qcheck. The gn-test-latest-mock |
| # function guarantees that we only check the latest invocation of the mock. |
| source "$(gn-test-latest-mock "${SSH_MOCK_PATH}/ssh.qcheck")" |
| expected=("${ssh_common_expected[@]}" -q -O check) |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.fconfig" |
| expected=("${ssh_common_expected[@]}" "cd" "\$HOME" "&&" |
| "cd" "/home/path_to_samples/third_party/fuchsia-sdk" "&&" |
| "./bin/fconfig.sh" "set" "device-ip" "127.0.0.1" "&&" |
| "./bin/fconfig.sh" "default" "device-name" "&&" |
| "./bin/fconfig.sh" "set" "image" "image1" "&&" |
| "./bin/fconfig.sh" "list") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check_for_8083" |
| expected=("${ssh_common_expected[@]}" "ss" "-ln" "|" "grep" ":8083") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # Check for killing existing pm server running |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.pkill" |
| expected=("${ssh_common_expected[@]}" "pkill" "-u" "\$USER" "pm") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.fserve" |
| expected=("${ssh_common_expected[@]}" ) |
| expected+=("cd" "\$HOME" "&&" "cd" "/home/path_to_samples/third_party/fuchsia-sdk" |
| "&&" "./bin/fserve.sh" "&&" "sleep" "0") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.check.2" |
| expected=("${ssh_common_expected[@]}" -O "check") |
| gn-test-check-mock-args "${expected[@]}" |
| |
| # shellcheck disable=SC1090 |
| source "${SSH_MOCK_PATH}/ssh.tunnel" |
| expected=("${ssh_common_expected[@]}" ) |
| expected+=(-6 -L "\*:8083:localhost:8083" |
| -R "8022:[fe80::c0ff:eec0:ffee%coffee]:22" |
| -R "2345:[fe80::c0ff:eec0:ffee%coffee]:2345" |
| -R "8443:[fe80::c0ff:eec0:ffee%coffee]:8443" |
| -R "9080:[fe80::c0ff:eec0:ffee%coffee]:80" |
| -o "ExitOnForwardFailure=yes" "-nT" "sleep" "0") |
| gn-test-check-mock-args "${expected[@]}" |
| } |
| |
| # Test initialization. Note that we copy various tools/devshell files and need to replicate the |
| # behavior of generate.py by copying these files into scripts/sdk/gn/base/bin/devshell |
| # shellcheck disable=SC2034 |
| BT_FILE_DEPS=( |
| scripts/sdk/gn/base/bin/fconfig.sh |
| scripts/sdk/gn/base/bin/fserve-remote.sh |
| scripts/sdk/gn/base/bin/fuchsia-common.sh |
| scripts/sdk/gn/bash_tests/gn-bash-test-lib.sh |
| tools/devshell/tests/subcommands/data/fx_remote_test/verify-default-keys.sh |
| ) |
| # shellcheck disable=SC2034 |
| BT_MOCKED_TOOLS=( |
| scripts/sdk/gn/base/bin/gsutil |
| scripts/sdk/gn/base/tools/x64/device-finder |
| scripts/sdk/gn/base/tools/arm64/device-finder |
| scripts/sdk/gn/base/tools/x64/pm |
| scripts/sdk/gn/base/tools/arm64/pm |
| isolated_path_for/ssh |
| isolated_path_for/kill |
| ) |
| |
| BT_INIT_TEMP_DIR() { |
| mkdir -p "${BT_TEMP_DIR}/scripts/sdk/gn/base/meta" |
| |
| # Create a stub SDK manifest. |
| cp "${BT_DEPS_ROOT}/scripts/sdk/gn/testdata/meta/manifest.json" \ |
| "${BT_TEMP_DIR}/scripts/sdk/gn/base/meta/manifest.json" |
| |
| # Stage the files we copy from fx implementation, replicating behavior of generate.py |
| cp -r "${BT_TEMP_DIR}/tools/devshell" "${BT_TEMP_DIR}/scripts/sdk/gn/base/bin/" |
| |
| mkdir "${BT_TEMP_DIR}/scripts/sdk/gn/base/bin/devshell/lib" |
| mv "${BT_TEMP_DIR}/tools/devshell/tests/subcommands/data/fx_remote_test/verify-default-keys.sh" "${BT_TEMP_DIR}/scripts/sdk/gn/base/bin/devshell/lib/verify-default-keys.sh" |
| } |
| |
| BT_SET_UP() { |
| # shellcheck disable=SC1090 |
| source "${BT_TEMP_DIR}/scripts/sdk/gn/bash_tests/gn-bash-test-lib.sh" |
| |
| # Make "home" directory in the test dir so the paths are stable." |
| mkdir -p "${BT_TEMP_DIR}/test-home" |
| export HOME="${BT_TEMP_DIR}/test-home" |
| |
| MOCKED_DEVICE_FINDER="${BT_TEMP_DIR}/scripts/sdk/gn/base/$(gn-test-tools-subdir)/device-finder" |
| |
| if [[ "$(type -t kill)" == "builtin" ]]; then |
| kill() { |
| "${BT_TEMP_DIR}/isolated_path_for/kill" "$@" |
| } |
| fi |
| # Create mock "kill" command$. |
| # Need to embed "enable -n kill" into fuchsia-common.sh to disable the |
| # bash builtin kill so we can intercept it. |
| echo "enable -n kill" >> "${BT_TEMP_DIR}/scripts/sdk/gn/base/bin/fuchsia-common.sh" |
| } |
| |
| BT_RUN_TESTS "$@" |
