xref: /aosp_15_r20/external/grpc-grpc/bazel/internal_python_rules.bzl (revision cc02d7e222339f7a4f6ba5f422e6413f4bd931f2)
1*cc02d7e2SAndroid Build Coastguard Worker# Copyright 2021 The gRPC Authors
2*cc02d7e2SAndroid Build Coastguard Worker#
3*cc02d7e2SAndroid Build Coastguard Worker# Licensed under the Apache License, Version 2.0 (the "License");
4*cc02d7e2SAndroid Build Coastguard Worker# you may not use this file except in compliance with the License.
5*cc02d7e2SAndroid Build Coastguard Worker# You may obtain a copy of the License at
6*cc02d7e2SAndroid Build Coastguard Worker#
7*cc02d7e2SAndroid Build Coastguard Worker#     http://www.apache.org/licenses/LICENSE-2.0
8*cc02d7e2SAndroid Build Coastguard Worker#
9*cc02d7e2SAndroid Build Coastguard Worker# Unless required by applicable law or agreed to in writing, software
10*cc02d7e2SAndroid Build Coastguard Worker# distributed under the License is distributed on an "AS IS" BASIS,
11*cc02d7e2SAndroid Build Coastguard Worker# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12*cc02d7e2SAndroid Build Coastguard Worker# See the License for the specific language governing permissions and
13*cc02d7e2SAndroid Build Coastguard Worker# limitations under the License.
14*cc02d7e2SAndroid Build Coastguard Worker"""Python-related rules intended only for use internal to the repo."""
15*cc02d7e2SAndroid Build Coastguard Worker
16*cc02d7e2SAndroid Build Coastguard Workerload("//bazel:gevent_test.bzl", "py_grpc_gevent_test")
17*cc02d7e2SAndroid Build Coastguard Worker
18*cc02d7e2SAndroid Build Coastguard Workerdef internal_py_grpc_test(name, **kwargs):
19*cc02d7e2SAndroid Build Coastguard Worker    """Runs a test under all supported environments.
20*cc02d7e2SAndroid Build Coastguard Worker
21*cc02d7e2SAndroid Build Coastguard Worker    Args:
22*cc02d7e2SAndroid Build Coastguard Worker      name: The name of the test.
23*cc02d7e2SAndroid Build Coastguard Worker      **kwargs: Any additional arguments to add to the test.
24*cc02d7e2SAndroid Build Coastguard Worker    """
25*cc02d7e2SAndroid Build Coastguard Worker    native.py_test(
26*cc02d7e2SAndroid Build Coastguard Worker        name = name + ".native",
27*cc02d7e2SAndroid Build Coastguard Worker        python_version = "PY3",
28*cc02d7e2SAndroid Build Coastguard Worker        **kwargs
29*cc02d7e2SAndroid Build Coastguard Worker    )
30*cc02d7e2SAndroid Build Coastguard Worker    py_grpc_gevent_test(name, **kwargs)
31*cc02d7e2SAndroid Build Coastguard Worker
32*cc02d7e2SAndroid Build Coastguard Worker    suite_kwargs = {}
33*cc02d7e2SAndroid Build Coastguard Worker    if "visibility" in kwargs:
34*cc02d7e2SAndroid Build Coastguard Worker        suite_kwargs["visibility"] = kwargs["visibility"]
35*cc02d7e2SAndroid Build Coastguard Worker
36*cc02d7e2SAndroid Build Coastguard Worker    native.test_suite(
37*cc02d7e2SAndroid Build Coastguard Worker        name = name,
38*cc02d7e2SAndroid Build Coastguard Worker        tests = [
39*cc02d7e2SAndroid Build Coastguard Worker            name + ".native",
40*cc02d7e2SAndroid Build Coastguard Worker            name + ".gevent",
41*cc02d7e2SAndroid Build Coastguard Worker        ],
42*cc02d7e2SAndroid Build Coastguard Worker        **suite_kwargs
43*cc02d7e2SAndroid Build Coastguard Worker    )
44