common: &common
  artifact_paths:
    - "/tmp/bazel_event_logs/*"
  env:
    BUILDKITE: "true"
    CI: "true"
    PYTHON: "3.6"
    RAY_USE_RANDOM_PORTS: "1"
    RAY_DEFAULT_BUILD: "1"
    LC_ALL: en_US.UTF-8
    LANG: en_US.UTF-8
    # TODO(simon): Can't figure out how get the JDK working yet.
    # RAY_INSTALL_JAVA: "1"

prelude_commands: &prelude_commands |-
  ./ci/travis/upload_build_info.sh
  (which bazel && bazel clean) || true;
  . ./ci/travis/ci.sh init && source ~/.zshrc
  . ./ci/travis/ci.sh build
  ./ci/travis/install-dependencies.sh

epilogue_commands: &epilogue_commands |-
  # Cleanup runtime environment to save storage
  rm -rf /tmp/ray
  # Cleanup local caches (this shouldn't clean up global disk cache)
  bazel clean

steps:
- label: ":mac: :apple: Ray C++ and Libraries"
  <<: *common
  commands:
    - *prelude_commands
    - TORCH_VERSION=1.6 ./ci/travis/install-dependencies.sh
    - bazel test --config=ci $(./scripts/bazel_export_options) --build_tests_only --test_tag_filters=-flaky,-flaky-mac --
      //:all python/ray/serve/... python/ray/new_dashboard/... -rllib/... -core_worker_test
    - *epilogue_commands

- label: ":mac: :apple: Small and Large"
  <<: *common
  commands:
    - *prelude_commands
    - bazel test $(./scripts/bazel_export_options) --config=ci
      --test_env=CONDA_EXE --test_env=CONDA_PYTHON_EXE --test_env=CONDA_SHLVL --test_env=CONDA_PREFIX
      --test_env=CONDA_DEFAULT_ENV --test_env=CONDA_PROMPT_MODIFIER
      --test_tag_filters=-kubernetes,-jenkins_only,-medium_size_python_tests_a_to_j,-medium_size_python_tests_k_to_z,-flaky,-flaky-mac
      python/ray/tests/...
    - *epilogue_commands

- label: ":mac: :apple: Medium A-J"
  <<: *common
  commands:
    - *prelude_commands
    - bazel test --config=ci $(./scripts/bazel_export_options)
      --test_tag_filters=-kubernetes,-jenkins_only,medium_size_python_tests_a_to_j,-flaky,-flaky-mac
      python/ray/tests/...
    - *epilogue_commands

- label: ":mac: :apple: Medium K-Z"
  <<: *common
  commands:
    - *prelude_commands
    - bazel test --config=ci $(./scripts/bazel_export_options)
      --test_tag_filters=-kubernetes,-jenkins_only,medium_size_python_tests_k_to_z,-flaky,-flaky-mac
      python/ray/tests/...
    - *epilogue_commands

- label: ":mac: :apple: :snowflake: Flaky"
  <<: *common
  commands:
    - *prelude_commands
    - ./ci/travis/install-dependencies.sh
    - bazel test --config=ci $(./scripts/bazel_export_options)
      --test_tag_filters=-kubernetes,-jenkins_only,flaky,flaky-mac
      --test_env=CONDA_EXE
      --test_env=CONDA_PYTHON_EXE
      --test_env=CONDA_SHLVL
      --test_env=CONDA_PREFIX
      --test_env=CONDA_DEFAULT_ENV
      python/ray/tests/... python/ray/serve/...
    - *epilogue_commands

    # The follow is running in Travis for now.
    # - bazel test --test_env=CONDA_EXE --test_env=CONDA_PYTHON_EXE --test_env=CONDA_SHLVL --test_env=CONDA_PREFIX --test_env=CONDA_DEFAULT_ENV --test_env=CONDA_PROMPT_MODIFIER --config=ci $(./scripts/bazel_export_options) --test_tag_filters=-kubernetes,-jenkins_only,-medium_size_python_tests_a_to_j,-medium_size_python_tests_k_to_z,-flaky python/ray/tests/...
    # - bazel test --config=ci $(./scripts/bazel_export_options) --test_tag_filters=-kubernetes,-jenkins_only,medium_size_python_tests_k_to_z,-flaky python/ray/tests/...
    # - bazel test --config=ci $(./scripts/bazel_export_options)
    #       --test_tag_filters=-kubernetes,-jenkins_only,flaky
    #       --test_env=CONDA_EXE
    #       --test_env=CONDA_PYTHON_EXE
    #       --test_env=CONDA_SHLVL
    #       --test_env=CONDA_PREFIX
    #       --test_env=CONDA_DEFAULT_ENV
    #       python/ray/tests/... python/ray/serve/... python/ray/tune/... rllib/...