.assign_test_template:
  image: $CI_DOCKER_REGISTRY/ubuntu-test-env$BOT_DOCKER_IMAGE_TAG
  stage: assign_test
  tags:
    - assign_test
  variables:
    SUBMODULES_TO_FETCH: components/esptool_py/esptool
  artifacts:
    paths:
      - ${TEST_DIR}/test_configs
      - ${BUILD_DIR}/artifact_index.json
    when: always
    expire_in: 1 week
  script:
    - python tools/ci/python_packages/ttfw_idf/IDFAssignTest.py $TEST_TYPE $TEST_DIR -c $CI_TARGET_TEST_CONFIG_FILE -o $TEST_DIR/test_configs

assign_example_test:
  extends:
    - .assign_test_template
    - .rules:build:example_test
  needs:
    - job: build_examples_cmake_esp32
      artifacts: false
      optional: true
    - job: build_examples_cmake_esp32s2
      artifacts: false
      optional: true
    - job: build_examples_cmake_esp32c3
      artifacts: false
      optional: true
    - job: build_examples_cmake_esp32s3
      artifacts: false
      optional: true
  variables:
    TEST_TYPE: example_test
    TEST_DIR: ${CI_PROJECT_DIR}/examples
    BUILD_DIR: ${CI_PROJECT_DIR}/build_examples

assign_custom_test:
  extends:
    - .assign_test_template
    - .rules:build:custom_test
  needs:
    - job: build_test_apps_esp32
      artifacts: false
      optional: true
    - job: build_test_apps_esp32s2
      artifacts: false
      optional: true
    - job: build_test_apps_esp32c3
      artifacts: false
      optional: true
    - job: build_test_apps_esp32s3
      artifacts: false
      optional: true
  variables:
    TEST_TYPE: custom_test
    TEST_DIR: ${CI_PROJECT_DIR}/tools/test_apps
    BUILD_DIR: ${CI_PROJECT_DIR}/build_test_apps

assign_component_ut:
  extends:
    - .assign_test_template
    - .rules:build:component_ut
  needs:
    - job: build_component_ut_esp32
      artifacts: false
      optional: true
    - job: build_component_ut_esp32s2
      artifacts: false
      optional: true
    - job: build_component_ut_esp32c3
      artifacts: false
      optional: true
    - job: build_component_ut_esp32s3
      artifacts: false
      optional: true
  variables:
    TEST_TYPE: component_ut
    TEST_DIR: ${CI_PROJECT_DIR}/component_ut
    BUILD_DIR: ${CI_PROJECT_DIR}/build_component_ut
  script:
    # COMPONENT_UT_DIRS is set by `set_component_ut_vars` in `utils.sh`
    - set_component_ut_vars
    - python tools/ci/python_packages/ttfw_idf/IDFAssignTest.py $TEST_TYPE $COMPONENT_UT_DIRS -c $CI_TARGET_TEST_CONFIG_FILE -o $TEST_DIR/test_configs

assign_unit_test:
  extends:
    - .assign_test_template
    - .rules:build:unit_test
  needs:
    - job: build_esp_idf_tests_cmake_esp32
      optional: true
    - job: build_esp_idf_tests_cmake_esp32s2
      optional: true
    - job: build_esp_idf_tests_cmake_esp32c3
      optional: true
    - job: build_esp_idf_tests_cmake_esp32s3
      optional: true
  variables:
    TEST_TYPE: unit_test
    TEST_DIR: ${CI_PROJECT_DIR}/components/idf_test/unit_test
    BUILD_DIR: ${CI_PROJECT_DIR}/tools/unit-test-app/builds
  script:
    - python tools/ci/python_packages/ttfw_idf/IDFAssignTest.py $TEST_TYPE $TEST_DIR -c $CI_TARGET_TEST_CONFIG_FILE -o $TEST_DIR/test_configs