| #!/usr/bin/env bash |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| if [[ ${VERBOSE_COMMANDS:="false"} == "true" ]]; then |
| set -x |
| fi |
| |
| # shellcheck source=scripts/in_container/_in_container_script_init.sh |
| . /opt/airflow/scripts/in_container/_in_container_script_init.sh |
| |
| # This one is to workaround https://github.com/apache/airflow/issues/17546 |
| # issue with /usr/lib/<MACHINE>-linux-gnu/libstdc++.so.6: cannot allocate memory in static TLS block |
| # We do not yet a more "correct" solution to the problem but in order to avoid raising new issues |
| # by users of the prod image, we implement the workaround now. |
| # The side effect of this is slightly (in the range of 100s of milliseconds) slower load for any |
| # binary started and a little memory used for Heap allocated by initialization of libstdc++ |
| # This overhead is not happening for binaries that already link dynamically libstdc++ |
| LD_PRELOAD="/usr/lib/$(uname -m)-linux-gnu/libstdc++.so.6" |
| export LD_PRELOAD |
| |
| # Add "other" and "group" write permission to the tmp folder |
| # Note that it will also change permissions in the /tmp folder on the host |
| # but this is necessary to enable some of our CLI tools to work without errors |
| chmod 1777 /tmp |
| |
| AIRFLOW_SOURCES=$(cd "${IN_CONTAINER_DIR}/../.." || exit 1; pwd) |
| |
| PYTHON_MAJOR_MINOR_VERSION=${PYTHON_MAJOR_MINOR_VERSION:=3.7} |
| |
| export AIRFLOW_HOME=${AIRFLOW_HOME:=${HOME}} |
| |
| : "${AIRFLOW_SOURCES:?"ERROR: AIRFLOW_SOURCES not set !!!!"}" |
| |
| if [[ ${SKIP_ENVIRONMENT_INITIALIZATION=} != "true" ]]; then |
| |
| if [[ $(uname -m) == "arm64" || $(uname -m) == "aarch64" ]]; then |
| if [[ ${BACKEND:=} == "mysql" || ${BACKEND} == "mssql" ]]; then |
| echo "${COLOR_RED}ARM platform is not supported for ${BACKEND} backend. Exiting.${COLOR_RESET}" |
| exit 1 |
| fi |
| fi |
| |
| echo |
| echo "${COLOR_BLUE}Running Initialization. Your basic configuration is:${COLOR_RESET}" |
| echo |
| echo " * ${COLOR_BLUE}Airflow home:${COLOR_RESET} ${AIRFLOW_HOME}" |
| echo " * ${COLOR_BLUE}Airflow sources:${COLOR_RESET} ${AIRFLOW_SOURCES}" |
| echo " * ${COLOR_BLUE}Airflow core SQL connection:${COLOR_RESET} ${AIRFLOW__CORE__SQL_ALCHEMY_CONN:=}" |
| echo |
| |
| RUN_TESTS=${RUN_TESTS:="false"} |
| CI=${CI:="false"} |
| USE_AIRFLOW_VERSION="${USE_AIRFLOW_VERSION:=""}" |
| |
| if [[ ${USE_AIRFLOW_VERSION} == "" ]]; then |
| export PYTHONPATH=${AIRFLOW_SOURCES} |
| echo |
| echo "${COLOR_BLUE}Using airflow version from current sources${COLOR_RESET}" |
| echo |
| # Cleanup the logs, tmp when entering the environment |
| sudo rm -rf "${AIRFLOW_SOURCES}"/logs/* |
| sudo rm -rf "${AIRFLOW_SOURCES}"/tmp/* |
| mkdir -p "${AIRFLOW_SOURCES}"/logs/ |
| mkdir -p "${AIRFLOW_SOURCES}"/tmp/ |
| elif [[ ${USE_AIRFLOW_VERSION} == "none" ]]; then |
| echo |
| echo "${COLOR_BLUE}Skip installing airflow - only install wheel/tar.gz packages that are present locally.${COLOR_RESET}" |
| echo |
| echo |
| echo "${COLOR_BLUE}Uninstalling airflow and providers" |
| echo |
| uninstall_airflow_and_providers |
| elif [[ ${USE_AIRFLOW_VERSION} == "wheel" ]]; then |
| echo |
| echo "${COLOR_BLUE}Uninstalling airflow and providers" |
| echo |
| uninstall_airflow_and_providers |
| if [[ ${SKIP_CONSTRAINTS,,=} == "true" ]]; then |
| echo "${COLOR_BLUE}Install airflow from wheel package with extras: '${AIRFLOW_EXTRAS}' with no constraints.${COLOR_RESET}" |
| echo |
| install_airflow_from_wheel "${AIRFLOW_EXTRAS}" "none" |
| else |
| echo "${COLOR_BLUE}Install airflow from wheel package with extras: '${AIRFLOW_EXTRAS}' and constraints reference ${AIRFLOW_CONSTRAINTS_REFERENCE}.${COLOR_RESET}" |
| echo |
| install_airflow_from_wheel "${AIRFLOW_EXTRAS}" "${AIRFLOW_CONSTRAINTS_REFERENCE}" |
| fi |
| uninstall_providers |
| elif [[ ${USE_AIRFLOW_VERSION} == "sdist" ]]; then |
| echo |
| echo "${COLOR_BLUE}Uninstalling airflow and providers" |
| echo |
| uninstall_airflow_and_providers |
| echo |
| if [[ ${SKIP_CONSTRAINTS,,=} == "true" ]]; then |
| echo "${COLOR_BLUE}Install airflow from sdist package with extras: '${AIRFLOW_EXTRAS}' with no constraints.${COLOR_RESET}" |
| echo |
| install_airflow_from_sdist "${AIRFLOW_EXTRAS}" "none" |
| else |
| echo "${COLOR_BLUE}Install airflow from sdist package with extras: '${AIRFLOW_EXTRAS}' and constraints reference ${AIRFLOW_CONSTRAINTS_REFERENCE}.${COLOR_RESET}" |
| echo |
| install_airflow_from_sdist "${AIRFLOW_EXTRAS}" "${AIRFLOW_CONSTRAINTS_REFERENCE}" |
| fi |
| uninstall_providers |
| else |
| echo |
| echo "${COLOR_BLUE}Uninstalling airflow and providers" |
| echo |
| uninstall_airflow_and_providers |
| echo |
| if [[ ${SKIP_CONSTRAINTS,,=} == "true" ]]; then |
| echo "${COLOR_BLUE}Install released airflow from PyPI with extras: '${AIRFLOW_EXTRAS}' with no constraints.${COLOR_RESET}" |
| echo |
| install_released_airflow_version "${USE_AIRFLOW_VERSION}" "none" |
| else |
| echo "${COLOR_BLUE}Install released airflow from PyPI with extras: '${AIRFLOW_EXTRAS}' and constraints reference ${AIRFLOW_CONSTRAINTS_REFERENCE}.${COLOR_RESET}" |
| echo |
| install_released_airflow_version "${USE_AIRFLOW_VERSION}" "${AIRFLOW_CONSTRAINTS_REFERENCE}" |
| fi |
| if [[ "${USE_AIRFLOW_VERSION}" =~ ^2\.2\..*|^2\.1\..*|^2\.0\..* && "${AIRFLOW__DATABASE__SQL_ALCHEMY_CONN=}" != "" ]]; then |
| # make sure old variable is used for older airflow versions |
| export AIRFLOW__CORE__SQL_ALCHEMY_CONN="${AIRFLOW__DATABASE__SQL_ALCHEMY_CONN}" |
| fi |
| fi |
| if [[ ${USE_PACKAGES_FROM_DIST=} == "true" ]]; then |
| echo |
| echo "${COLOR_BLUE}Install all packages from dist folder${COLOR_RESET}" |
| if [[ ${USE_AIRFLOW_VERSION} == "wheel" ]]; then |
| echo "${COLOR_BLUE}(except apache-airflow)${COLOR_RESET}" |
| fi |
| if [[ ${PACKAGE_FORMAT} == "both" ]]; then |
| echo |
| echo "${COLOR_RED}ERROR:You can only specify 'wheel' or 'sdist' as PACKAGE_FORMAT not 'both'.${COLOR_RESET}" |
| echo |
| exit 1 |
| fi |
| echo |
| installable_files=() |
| for file in /dist/*.{whl,tar.gz} |
| do |
| if [[ ${USE_AIRFLOW_VERSION} == "wheel" && ${file} == "/dist/apache?airflow-[0-9]"* ]]; then |
| # Skip Apache Airflow package - it's just been installed above with extras |
| echo "Skipping ${file}" |
| continue |
| fi |
| if [[ ${PACKAGE_FORMAT} == "wheel" && ${file} == *".whl" ]]; then |
| echo "Adding ${file} to install" |
| installable_files+=( "${file}" ) |
| fi |
| if [[ ${PACKAGE_FORMAT} == "sdist" && ${file} == *".tar.gz" ]]; then |
| echo "Adding ${file} to install" |
| installable_files+=( "${file}" ) |
| fi |
| done |
| if (( ${#installable_files[@]} )); then |
| pip install --root-user-action ignore "${installable_files[@]}" |
| fi |
| fi |
| |
| # Added to have run-tests on path |
| export PATH=${PATH}:${AIRFLOW_SOURCES} |
| |
| # This is now set in conftest.py - only for pytest tests |
| unset AIRFLOW__CORE__UNIT_TEST_MODE |
| |
| mkdir -pv "${AIRFLOW_HOME}/logs/" |
| cp -f "${IN_CONTAINER_DIR}/airflow_ci.cfg" "${AIRFLOW_HOME}/unittests.cfg" |
| |
| # Change the default worker_concurrency for tests |
| export AIRFLOW__CELERY__WORKER_CONCURRENCY=8 |
| |
| set +e |
| |
| "${IN_CONTAINER_DIR}/check_environment.sh" |
| ENVIRONMENT_EXIT_CODE=$? |
| set -e |
| if [[ ${ENVIRONMENT_EXIT_CODE} != 0 ]]; then |
| echo |
| echo "Error: check_environment returned ${ENVIRONMENT_EXIT_CODE}. Exiting." |
| echo |
| exit ${ENVIRONMENT_EXIT_CODE} |
| fi |
| # Create symbolic link to fix possible issues with kubectl config cmd-path |
| mkdir -p /usr/lib/google-cloud-sdk/bin |
| touch /usr/lib/google-cloud-sdk/bin/gcloud |
| ln -s -f /usr/bin/gcloud /usr/lib/google-cloud-sdk/bin/gcloud |
| |
| in_container_fix_ownership |
| |
| if [[ ${SKIP_SSH_SETUP="false"} == "false" ]]; then |
| # Set up ssh keys |
| echo 'yes' | ssh-keygen -t rsa -C your_email@youremail.com -m PEM -P '' -f ~/.ssh/id_rsa \ |
| >"${AIRFLOW_HOME}/logs/ssh-keygen.log" 2>&1 |
| |
| cat ~/.ssh/id_rsa.pub >> ~/.ssh/authorized_keys |
| ln -s -f ~/.ssh/authorized_keys ~/.ssh/authorized_keys2 |
| chmod 600 ~/.ssh/* |
| |
| # SSH Service |
| sudo service ssh restart >/dev/null 2>&1 |
| |
| # Sometimes the server is not quick enough to load the keys! |
| while [[ $(ssh-keyscan -H localhost 2>/dev/null | wc -l) != "3" ]] ; do |
| echo "Not all keys yet loaded by the server" |
| sleep 0.05 |
| done |
| |
| ssh-keyscan -H localhost >> ~/.ssh/known_hosts 2>/dev/null |
| fi |
| |
| # shellcheck source=scripts/in_container/configure_environment.sh |
| . "${IN_CONTAINER_DIR}/configure_environment.sh" |
| |
| # shellcheck source=scripts/in_container/run_init_script.sh |
| . "${IN_CONTAINER_DIR}/run_init_script.sh" |
| |
| cd "${AIRFLOW_SOURCES}" |
| |
| if [[ ${START_AIRFLOW:="false"} == "true" || ${START_AIRFLOW} == "True" ]]; then |
| export AIRFLOW__DATABASE__LOAD_DEFAULT_CONNECTIONS=${LOAD_DEFAULT_CONNECTIONS} |
| export AIRFLOW__CORE__LOAD_EXAMPLES=${LOAD_EXAMPLES} |
| # shellcheck source=scripts/in_container/bin/run_tmux |
| exec run_tmux |
| fi |
| fi |
| |
| set +u |
| # If we do not want to run tests, we simply drop into bash |
| if [[ "${RUN_TESTS}" != "true" ]]; then |
| exec /bin/bash "${@}" |
| fi |
| set -u |
| |
| export RESULT_LOG_FILE="/files/test_result-${TEST_TYPE/\[*\]/}-${BACKEND}.xml" |
| |
| EXTRA_PYTEST_ARGS=( |
| "--verbosity=0" |
| "--strict-markers" |
| "--durations=100" |
| "--maxfail=50" |
| "--color=yes" |
| "--junitxml=${RESULT_LOG_FILE}" |
| # timeouts in seconds for individual tests |
| "--timeouts-order" |
| "moi" |
| "--setup-timeout=60" |
| "--execution-timeout=60" |
| "--teardown-timeout=60" |
| # Only display summary for non-expected case |
| # f - failed |
| # E - error |
| # X - xpassed (passed even if expected to fail) |
| # The following cases are not displayed: |
| # s - skipped |
| # x - xfailed (expected to fail and failed) |
| # p - passed |
| # P - passed with output |
| "-rfEX" |
| ) |
| |
| if [[ "${TEST_TYPE}" == "Helm" ]]; then |
| _cpus="$(grep -c 'cpu[0-9]' /proc/stat)" |
| echo "Running tests with ${_cpus} CPUs in parallel" |
| # Enable parallelism |
| EXTRA_PYTEST_ARGS+=( |
| "-n" "${_cpus}" |
| ) |
| else |
| EXTRA_PYTEST_ARGS+=( |
| "--with-db-init" |
| ) |
| fi |
| |
| if [[ ${ENABLE_TEST_COVERAGE:="false"} == "true" ]]; then |
| EXTRA_PYTEST_ARGS+=( |
| "--cov=airflow/" |
| "--cov-config=.coveragerc" |
| "--cov-report=xml:/files/coverage-${TEST_TYPE/\[*\]/}-${BACKEND}.xml" |
| ) |
| fi |
| |
| declare -a SELECTED_TESTS CLI_TESTS API_TESTS PROVIDERS_TESTS CORE_TESTS WWW_TESTS \ |
| ALL_TESTS ALL_PRESELECTED_TESTS ALL_OTHER_TESTS |
| |
| # Finds all directories that are not on the list of tests |
| # - so that we do not skip any in the future if new directories are added |
| function find_all_other_tests() { |
| local all_tests_dirs |
| all_tests_dirs=$(find "tests" -type d) |
| all_tests_dirs=$(echo "${all_tests_dirs}" | sed "/tests$/d" ) |
| all_tests_dirs=$(echo "${all_tests_dirs}" | sed "/tests\/dags/d" ) |
| local path |
| for path in "${ALL_PRESELECTED_TESTS[@]}" |
| do |
| escaped_path="${path//\//\\\/}" |
| all_tests_dirs=$(echo "${all_tests_dirs}" | sed "/${escaped_path}/d" ) |
| done |
| for path in ${all_tests_dirs} |
| do |
| ALL_OTHER_TESTS+=("${path}") |
| done |
| } |
| |
| if [[ ${#@} -gt 0 && -n "$1" ]]; then |
| SELECTED_TESTS=("${@}") |
| else |
| CLI_TESTS=("tests/cli") |
| API_TESTS=("tests/api" "tests/api_connexion") |
| PROVIDERS_TESTS=("tests/providers") |
| ALWAYS_TESTS=("tests/always") |
| CORE_TESTS=( |
| "tests/core" |
| "tests/executors" |
| "tests/jobs" |
| "tests/models" |
| "tests/serialization" |
| "tests/ti_deps" |
| "tests/utils" |
| ) |
| WWW_TESTS=("tests/www") |
| HELM_CHART_TESTS=("tests/charts") |
| ALL_TESTS=("tests") |
| ALL_PRESELECTED_TESTS=( |
| "${CLI_TESTS[@]}" |
| "${API_TESTS[@]}" |
| "${HELM_CHART_TESTS[@]}" |
| "${PROVIDERS_TESTS[@]}" |
| "${CORE_TESTS[@]}" |
| "${ALWAYS_TESTS[@]}" |
| "${WWW_TESTS[@]}" |
| ) |
| |
| if [[ ${TEST_TYPE:=""} == "CLI" ]]; then |
| SELECTED_TESTS=("${CLI_TESTS[@]}") |
| elif [[ ${TEST_TYPE:=""} == "API" ]]; then |
| SELECTED_TESTS=("${API_TESTS[@]}") |
| elif [[ ${TEST_TYPE:=""} == "Providers" ]]; then |
| SELECTED_TESTS=("${PROVIDERS_TESTS[@]}") |
| elif [[ ${TEST_TYPE:=""} == "Core" ]]; then |
| SELECTED_TESTS=("${CORE_TESTS[@]}") |
| elif [[ ${TEST_TYPE:=""} == "Always" ]]; then |
| SELECTED_TESTS=("${ALWAYS_TESTS[@]}") |
| elif [[ ${TEST_TYPE:=""} == "WWW" ]]; then |
| SELECTED_TESTS=("${WWW_TESTS[@]}") |
| elif [[ ${TEST_TYPE:=""} == "Helm" ]]; then |
| SELECTED_TESTS=("${HELM_CHART_TESTS[@]}") |
| elif [[ ${TEST_TYPE:=""} == "Other" ]]; then |
| find_all_other_tests |
| SELECTED_TESTS=("${ALL_OTHER_TESTS[@]}") |
| elif [[ ${TEST_TYPE:=""} == "All" || ${TEST_TYPE} == "Quarantined" || \ |
| ${TEST_TYPE} == "Always" || \ |
| ${TEST_TYPE} == "Postgres" || ${TEST_TYPE} == "MySQL" || \ |
| ${TEST_TYPE} == "Long" || \ |
| ${TEST_TYPE} == "Integration" ]]; then |
| SELECTED_TESTS=("${ALL_TESTS[@]}") |
| elif [[ ${TEST_TYPE} =~ Providers\[(.*)\] ]]; then |
| SELECTED_TESTS=() |
| for provider in ${BASH_REMATCH[1]//,/ } |
| do |
| SELECTED_TESTS+=("tests/providers/${provider//./\/}") |
| done |
| else |
| echo |
| echo "${COLOR_RED}ERROR: Wrong test type ${TEST_TYPE} ${COLOR_RESET}" |
| echo |
| exit 1 |
| fi |
| fi |
| readonly SELECTED_TESTS CLI_TESTS API_TESTS PROVIDERS_TESTS CORE_TESTS WWW_TESTS \ |
| ALL_TESTS ALL_PRESELECTED_TESTS |
| |
| if [[ -n ${LIST_OF_INTEGRATION_TESTS_TO_RUN=} ]]; then |
| # Integration tests |
| for INT in ${LIST_OF_INTEGRATION_TESTS_TO_RUN} |
| do |
| EXTRA_PYTEST_ARGS+=("--integration" "${INT}") |
| done |
| elif [[ ${TEST_TYPE:=""} == "Long" ]]; then |
| EXTRA_PYTEST_ARGS+=( |
| "-m" "long_running" |
| "--include-long-running" |
| ) |
| elif [[ ${TEST_TYPE:=""} == "Postgres" ]]; then |
| EXTRA_PYTEST_ARGS+=( |
| "--backend" |
| "postgres" |
| ) |
| elif [[ ${TEST_TYPE:=""} == "MySQL" ]]; then |
| EXTRA_PYTEST_ARGS+=( |
| "--backend" |
| "mysql" |
| ) |
| elif [[ ${TEST_TYPE:=""} == "Quarantined" ]]; then |
| EXTRA_PYTEST_ARGS+=( |
| "-m" "quarantined" |
| "--include-quarantined" |
| ) |
| fi |
| |
| echo |
| echo "Running tests ${SELECTED_TESTS[*]}" |
| echo |
| |
| ARGS=("${EXTRA_PYTEST_ARGS[@]}" "${SELECTED_TESTS[@]}") |
| |
| if [[ ${RUN_SYSTEM_TESTS:="false"} == "true" ]]; then |
| "${IN_CONTAINER_DIR}/run_system_tests.sh" "${ARGS[@]}" |
| else |
| "${IN_CONTAINER_DIR}/run_ci_tests.sh" "${ARGS[@]}" |
| fi |
