mirror of https://github.com/k3s-io/k3s
50 lines
1.8 KiB
Python
50 lines
1.8 KiB
Python
# Copyright 2018 The Kubernetes Authors.
|
|
#
|
|
# 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(":code_generation.bzl", "bazel_go_library", "go_pkg")
|
|
load("@bazel_skylib//lib:unittest.bzl", "asserts", "unittest")
|
|
|
|
def _bazel_go_library_test_impl(ctx):
|
|
env = unittest.begin(ctx)
|
|
test_cases = [
|
|
("pkg/kubectl/util", "//pkg/kubectl/util:go_default_library"),
|
|
("vendor/some/third/party", "//vendor/some/third/party:go_default_library"),
|
|
("staging/src/k8s.io/apimachinery/api", "//staging/src/k8s.io/apimachinery/api:go_default_library"),
|
|
]
|
|
for input, expected in test_cases:
|
|
asserts.equals(env, expected, bazel_go_library(input))
|
|
unittest.end(env)
|
|
|
|
bazel_go_library_test = unittest.make(_bazel_go_library_test_impl)
|
|
|
|
def _go_pkg_test_impl(ctx):
|
|
env = unittest.begin(ctx)
|
|
test_cases = [
|
|
("pkg/kubectl/util", "k8s.io/kubernetes/pkg/kubectl/util"),
|
|
("vendor/some/third/party", "k8s.io/kubernetes/vendor/some/third/party"),
|
|
("staging/src/k8s.io/apimachinery/api", "k8s.io/kubernetes/vendor/k8s.io/apimachinery/api"),
|
|
]
|
|
for input, expected in test_cases:
|
|
asserts.equals(env, expected, go_pkg(input))
|
|
unittest.end(env)
|
|
|
|
go_pkg_test = unittest.make(_go_pkg_test_impl)
|
|
|
|
def code_generation_test_suite(name):
|
|
unittest.suite(
|
|
name,
|
|
bazel_go_library_test,
|
|
go_pkg_test,
|
|
)
|