# Copyright 2021 Google LLC. All Rights Reserved.
#
# Licensed 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.
# =============================================================================

load("@npm//@bazel/concatjs:index.bzl", "ts_config")
load("//tools:tfjs_web_test.bzl", "grep_flag", "headless_flag")

package(default_visibility = ["//visibility:public"])

exports_files([
    "babel.config.json",
    "tsconfig.json",
    "tsconfig_es5.json",
    "tsconfig.test.json",
    "tsconfig_base.json",
    "tsconfig_ts_library.json",
])

ts_config(
    name = "tsconfig_ts_library",
    src = "tsconfig_ts_library.json",
    deps = [
        ":tsconfig.json",
        ":tsconfig_base.json",
    ],
)

grep_flag(
    name = "grep",
    build_setting_default = "",
    visibility = ["//visibility:public"],
)

headless_flag(
    name = "headless",
    build_setting_default = True,
    visibility = ["//visibility:public"],
)

test_suite(
    name = "tests_cpu",
    tests = [
        "//tfjs-backend-cpu:tests",
        "//tfjs-backend-wasm:tests",
        "//tfjs-converter:tests",
        "//tfjs-core:tests",
        "//tfjs-data:tests",
        "//tfjs-tfdf:tests",
        "//tfjs-tflite:tests",
    ],
)

test_suite(
    name = "tests_gpu",
    tests = [
        "//tfjs-backend-webgl:tests",
        "//tfjs-backend-webgpu:tests",
        "//tfjs-layers:tests",
    ],
)

test_suite(
    name = "tests",
    tests = [
        ":tests_cpu",
        ":tests_gpu",
    ],
)