1 files changed, 51 insertions, 0 deletions
diff --git a/python/tests/toolchains/versions_test.bzl b/python/tests/toolchains/versions_test.bzl
new file mode 100644
index 0000000..b885d22
--- /dev/null
+++ b/python/tests/toolchains/versions_test.bzl
@@ -0,0 +1,51 @@
+# Copyright 2022 The Bazel Authors. 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.
+
+"""Unit tests for starlark helpers
+See https://docs.bazel.build/versions/main/skylark/testing.html#for-testing-starlark-utilities
+"""
+
+load("@bazel_skylib//lib:unittest.bzl", "asserts", "unittest")
+load("//python:versions.bzl", "MINOR_MAPPING", "TOOL_VERSIONS")
+
+required_platforms = [
+    "x86_64-apple-darwin",
+    "x86_64-unknown-linux-gnu",
+]
+
+def _smoke_test_impl(ctx):
+    env = unittest.begin(ctx)
+    for version in TOOL_VERSIONS.keys():
+        platforms = TOOL_VERSIONS[version]["sha256"]
+        for required_platform in required_platforms:
+            asserts.true(
+                env,
+                required_platform in platforms.keys(),
+                "Missing platform {} for version {}".format(required_platform, version),
+            )
+    for minor in MINOR_MAPPING:
+        version = MINOR_MAPPING[minor]
+        asserts.true(
+            env,
+            version in TOOL_VERSIONS.keys(),
+            "Missing version {} in TOOL_VERSIONS".format(version),
+        )
+    return unittest.end(env)
+
+# The unittest library requires that we export the test cases as named test rules,
+# but their names are arbitrary and don't appear anywhere.
+_t0_test = unittest.make(_smoke_test_impl)
+
+def versions_test_suite(name):
+    unittest.suite(name, _t0_test)