load("@rules_python//python:defs.bzl", "py_library", "py_test")
load("//bazel:python.bzl", "py_test_run_all_subdirectory")

# This is a dummy test dependency that causes the above tests to be
# re-run if any of these files changes.
py_library(
    name = "dashboard_lib",
    srcs = glob(
        ["**/*.py"],
        exclude = ["tests/*"],
    ),
)

py_library(
    name = "conftest",
    srcs = ["tests/conftest.py"],
    visibility = [
        "//python/ray/dashboard:__subpackages__",
    ],
    deps = ["//python/ray/tests:conftest"],
)

py_test_run_all_subdirectory(
    size = "medium",
    include = ["**/test*.py"],
    data = [
        "modules/job/tests/backwards_compatibility_scripts/script.py",
        "modules/job/tests/backwards_compatibility_scripts/test_backwards_compatibility.sh",
        "modules/job/tests/pip_install_test-0.5-py3-none-any.whl",
        "modules/tests/test_config_files/basic_runtime_env.yaml",
    ] + glob([
        "modules/job/tests/subprocess_driver_scripts/*.py",
    ]),
    exclude = [
        "client/node_modules/**",
        "modules/aggregator/tests/test_aggregator_agent.py",
        "modules/job/tests/test_cli_integration.py",
        "modules/job/tests/test_http_job_server.py",
        "modules/job/tests/test_job_agent.py",
        "modules/node/tests/test_node.py",
        "tests/test_dashboard.py",
        "tests/test_state_head.py",
        "modules/serve/tests/**/*.py",
        "modules/job/tests/test_job_manager.py",
    ],
    extra_srcs = [],
    tags = [
        "exclusive",
        "team:core",
    ],
    deps = [":conftest"],
)

py_test(
    name = "test_cli_integration",
    size = "large",
    srcs = ["modules/job/tests/test_cli_integration.py"],
    tags = [
        "exclusive",
        "team:core",
    ],
    deps = [":conftest"],
)

py_test(
    name = "test_job_manager",
    size = "large",
    timeout = "eternal",
    srcs = ["modules/job/tests/test_job_manager.py"],
    tags = [
        "exclusive",
        "team:core",
    ],
    deps = [":conftest"],
)

py_test(
    name = "test_aggregator_agent",
    size = "large",
    srcs = ["modules/aggregator/tests/test_aggregator_agent.py"],
    tags = [
        "exclusive",
        "team:core",
    ],
    deps = [":conftest"],
)

py_test(
    name = "test_http_job_server",
    size = "large",
    srcs = ["modules/job/tests/test_http_job_server.py"],
    data = [
        "modules/job/tests/backwards_compatibility_scripts/script.py",
        "modules/job/tests/backwards_compatibility_scripts/test_backwards_compatibility.sh",
        "modules/job/tests/pip_install_test-0.5-py3-none-any.whl",
        "modules/tests/test_config_files/basic_runtime_env.yaml",
    ] + glob([
        "modules/job/tests/subprocess_driver_scripts/*.py",
    ]),
    tags = [
        "exclusive",
        "team:core",
    ],
    deps = [":conftest"],
)

py_test(
    name = "test_job_agent",
    size = "large",
    srcs = ["modules/job/tests/test_job_agent.py"],
    data = [
        "modules/job/tests/backwards_compatibility_scripts/script.py",
        "modules/job/tests/backwards_compatibility_scripts/test_backwards_compatibility.sh",
        "modules/job/tests/pip_install_test-0.5-py3-none-any.whl",
        "modules/tests/test_config_files/basic_runtime_env.yaml",
    ] + glob([
        "modules/job/tests/subprocess_driver_scripts/*.py",
    ]),
    tags = [
        "exclusive",
        "team:core",
    ],
    deps = [":conftest"],
)

py_test(
    name = "test_node",
    size = "medium",
    srcs = ["modules/node/tests/test_node.py"],
    tags = [
        "exclusive",
        "team:core",
    ],
    deps = [":conftest"],
)

py_test(
    name = "test_dashboard",
    size = "large",
    srcs = ["tests/test_dashboard.py"],
    tags = [
        "exclusive",
        "minimal",
        "team:core",
    ],
    deps = [":conftest"],
)

py_test(
    name = "test_dashboard_auth",
    size = "large",
    srcs = ["tests/test_dashboard_auth.py"],
    tags = [
        "exclusive",
        "team:core",
    ],
    deps = [":conftest"],
)

py_test(
    name = "test_metrics_integration",
    size = "medium",
    srcs = ["modules/tests/test_metrics_integration.py"],
    tags = [
        "exclusive",
        "team:clusters",
    ],
    deps = [":conftest"],
)

py_test(
    name = "test_state_head",
    size = "small",
    srcs = ["tests/test_state_head.py"],
    tags = ["team:core"],
    deps = [":conftest"],
)

py_test(
    name = "test_serve_dashboard",
    size = "enormous",
    srcs = [
        "modules/serve/tests/deploy_imperative_serve_apps.py",
        "modules/serve/tests/test_serve_dashboard.py",
    ],
    tags = ["team:serve"],
    deps = [":conftest"],
)

py_test(
    name = "test_serve_dashboard_2",
    size = "enormous",
    srcs = ["modules/serve/tests/test_serve_dashboard_2.py"],
    tags = ["team:serve"],
    deps = [":conftest"],
)

py_test(
    name = "test_data_head",
    size = "small",
    srcs = ["modules/data/tests/test_data_head.py"],
    tags = ["team:data"],
    deps = [":conftest"],
)