docker-connect-test.sh

#!/usr/bin/env bash

# This script runs @trezor/connect tests.
# It spins up trezor-user-env and sets required evironment variables.

set -euo pipefail

# Running standalone instance of trezor-user-env
# docker run -it -e SDL_VIDEODRIVER="dummy" -p "9001:9001" -p "21326:21326" -p "21325:21326" registry.gitlab.com/satoshilabs/trezor/trezor-user-env/trezor-user-env

# Tweaking trezor-user-env locally
# docker run -it -e SDL_VIDEODRIVER="dummy" -p "9001:9001" -p "21326:21326" -p "21325:21326" registry.gitlab.com/satoshilabs/trezor/trezor-user-env/trezor-user-env nix-shell
# do your changes using `vi` and run:
# [nix-shell:/trezor-user-env]# ./run.sh

# Ports
# 9001  - websocket server, communication test case > user-env (setup etc...)
# 21326 - trezord proxy. beacuse of trezord CORS check
# 21325 - original trezord port redirected to trezor-user-env proxy

ENVIRONMENT=$1

if [[ "$ENVIRONMENT" != "web" && "$ENVIRONMENT" != "node" ]];
    then
        echo "either 'web' or 'node' must be specified as the first argument"
        exit 1
fi

show_usage() {
  echo "Usage: run [OPTIONS] [ARGS]"
  echo ""
  echo "Options:"
  echo "  -c       Disable backend cache. default: enabled"
  echo "  -d       Disable docker. Useful when running own instance of trezor-user-env. default: enabled"
  echo "  -D PATH  Set path to docker executable. Can be replaced with \`podman\`. default: docker"
  echo "  -e       All methods except excluded, example: applySettings,signTransaction"
  echo "  -p       Test pattern"
  echo "  -f       Use specific firmware version, example: 2.1.4, 1.8.0 default: 2-main"
  echo "  -i       Included methods only, example: applySettings,signTransaction"
  echo "  -s       actual test script. default: 'yarn test:integration'"
  echo "  -u       Firmware url"
  echo "  -m       Firmware model, example: R'"
}

# default options
FIRMWARE=""
FIRMWARE_URL=""
INCLUDED_METHODS=""
EXCLUDED_METHODS=""
DOCKER=true
# TODO: DOCKER_PATH appears unused. Remove or export?
DOCKER_PATH="docker"
USE_TX_CACHE=true
USE_WS_CACHE=true
PATTERN=""
FIRMWARE_MODEL=""

# user options
OPTIND=2
while getopts ":p:i:e:f:u:m:D:hdc" opt; do
  case $opt in
  d)
    DOCKER=false
    ;;
  D)
    DOCKER_PATH="$OPTARG"
    ;;
  c)
    USE_TX_CACHE=false
    USE_WS_CACHE=false
    ;;
  f)
    FIRMWARE=$OPTARG
    ;;
  u)
    FIRMWARE_URL=$OPTARG
    ;;
  i)
    INCLUDED_METHODS=$OPTARG
    ;;
  e)
    EXCLUDED_METHODS=$OPTARG
    ;;
  p)
    PATTERN=$OPTARG
    ;;
  m)
    FIRMWARE_MODEL=$OPTARG
    ;;
  h) # Script usage
    show_usage
    exit 0
    ;;
  \?)
    echo "invalid option $OPTARG"
    exit 1
    ;;
  esac
done
shift $((OPTIND - 1))

export DOCKER_PATH

if [[ $ENVIRONMENT == "node" ]];
  then
    SCRIPT="yarn workspace @trezor/connect test:e2e:node"
  else
    SCRIPT="yarn workspace @trezor/connect test:e2e:web"
fi

# export essential process.env variables
export TESTS_FIRMWARE=$FIRMWARE
export TESTS_INCLUDED_METHODS=$INCLUDED_METHODS
export TESTS_EXCLUDED_METHODS=$EXCLUDED_METHODS
export TESTS_USE_TX_CACHE=$USE_TX_CACHE
export TESTS_USE_WS_CACHE=$USE_WS_CACHE
export TESTS_PATTERN=$PATTERN
export TESTS_SCRIPT=$SCRIPT
export TESTS_FIRMWARE_URL=$FIRMWARE_URL
export TESTS_FIRMWARE_MODEL=$FIRMWARE_MODEL

runDocker() {
  docker compose -f ./docker/docker-compose.connect-test.yml up --abort-on-container-exit
}

run() {

  echo "Testing env: ${ENVIRONMENT}. Using: ${SCRIPT} ${PATTERN}"
  echo "  Firmware: ${TESTS_FIRMWARE}"
  echo "  Firmware model: ${TESTS_FIRMWARE_MODEL}"
  echo "  Firmware from url: ${FIRMWARE_URL}"
  echo "  Test pattern: $PATTERN"
  echo "  Included methods: ${INCLUDED_METHODS}"
  echo "  Excluded methods: ${EXCLUDED_METHODS}"
  echo "  TxCache: ${USE_TX_CACHE}"
  echo "  WsCache: ${USE_WS_CACHE}"

  if [ $DOCKER = true ]; then
    runDocker
  else
    $SCRIPT "$PATTERN"
  fi

}

run