load("//third_party/bazel_rules/rules_python/python:py_test.bzl", "py_test")

package(
    default_applicable_licenses = ["//third_party/py/cloud_goodput:license"],
    default_visibility = ["//visibility:private"],
    licenses = ["notice"],
)

py_test(
    name = "goodput_test",
    srcs = ["goodput_test.py"],
    strict_deps = False,
    deps = [
        "//testing/pybase",
        "//third_party/py/cloud_goodput/ml_goodput_measurement/src:cloud_goodput",
        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip
    ],
)

py_test(
    name = "checkpoint_badput_calculator_test",
    srcs = ["checkpoint_badput_calculator_test.py"],
    strict_deps = False,
    deps = [
        "//testing/pybase",
        "//third_party/py/cloud_goodput/ml_goodput_measurement/src:cloud_goodput",
        "//third_party/py/google/cloud/logging",
        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip,
        "//third_party/py/mock",
    ],
)

py_test(
    name = "goodput_cache_test",
    srcs = ["goodput_cache_test.py"],
    strict_deps = False,
    deps = [
        "//testing/pybase",
        "//third_party/py/cloud_goodput/ml_goodput_measurement/src:cloud_goodput",
        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip,
    ],
)

py_test(
    name = "goodput_elastic_test",
    srcs = ["goodput_elastic_test.py"],
    strict_deps = False,
    deps = [
        "//testing/pybase",
        "//third_party/py/cloud_goodput/ml_goodput_measurement/src:cloud_goodput",
        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip,
    ],
)

py_test(
    name = "monitoring_test",
    srcs = ["monitoring_test.py"],
    strict_deps = False,
    deps = [
        "//testing/pybase",
        "//third_party/py/absl/testing:absltest",
        "//third_party/py/cloud_goodput/ml_goodput_measurement/src:cloud_goodput",
        "//third_party/py/google/cloud/monitoring_v3",
        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip,
    ],
)

py_test(
    name = "gcp_metrics_test",
    srcs = ["gcp_metrics_test.py"],
    strict_deps = False,
    deps = [
        "//testing/pybase",
        "//third_party/py/cloud_goodput/ml_goodput_measurement/src:cloud_goodput",
        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip,
    ],
)

py_test(
    name = "exclusion_utils_test",
    srcs = ["exclusion_utils_test.py"],
    strict_deps = False,
    deps = [
        "//testing/pybase",
        "//third_party/py/absl/testing:absltest",
        "//third_party/py/cloud_goodput/ml_goodput_measurement/src:cloud_goodput",
        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip,
    ],
)

py_test(
    name = "cloud_logging_test",
    srcs = ["cloud_logging_test.py"],
    strict_deps = False,
    deps = [
        "//testing/pybase",
        "//third_party/py/absl/testing:absltest",
        "//third_party/py/cloud_goodput/ml_goodput_measurement/src:cloud_goodput",
        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip,
    ],
)

py_test(
    name = "goodput_utils_test",
    srcs = ["goodput_utils_test.py"],
    strict_deps = False,
    deps = [
        "//testing/pybase",
        "//third_party/py/absl/testing:absltest",
        "//third_party/py/cloud_goodput/ml_goodput_measurement/src:cloud_goodput",
        "//third_party/py/requests",
    ],
)

py_test(
    name = "monitoring_elastic_test",
    srcs = ["monitoring_elastic_test.py"],
    strict_deps = False,
    deps = [
        "//testing/pybase",
        "//third_party/py/absl/testing:absltest",
        "//third_party/py/cloud_goodput/ml_goodput_measurement/src:cloud_goodput",
        "//third_party/py/google/cloud/monitoring_v3",
        "//third_party/py/google/protobuf:use_fast_cpp_protos",  # Automatically added go/proto_python_upb_flip,
    ],
)
