tests/toolchains/python_toolchain_test.py

import json
import os
import pathlib
import pprint
import sys
import unittest

from python.runfiles import runfiles


class PythonToolchainTest(unittest.TestCase):
    def test_expected_toolchain_matches(self):
        expect_version = os.environ["EXPECT_PYTHON_VERSION"]

        rf = runfiles.Create()
        assert rf is not None  # type assert
        settings_path = rf.Rlocation(
            "rules_python/tests/support/current_build_settings.json"
        )
        assert settings_path is not None  # type assert
        settings = json.loads(pathlib.Path(settings_path).read_text())

        expected = "python_{}".format(expect_version.replace(".", "_"))
        msg = (
            "Expected toolchain not found\n"
            + f"Expected toolchain label to contain: {expected}\n"
            + "Actual build settings:\n"
            + pprint.pformat(settings)
        )
        self.assertIn(expected, settings["toolchain_label"], msg)

        actual = "{v.major}.{v.minor}.{v.micro}".format(v=sys.version_info)
        if sys.version_info.releaselevel != "final":
            release_prefix = (
                "rc"
                if sys.version_info.releaselevel == "candidate"
                else sys.version_info.releaselevel[0]
            )
            actual = f"{actual}{release_prefix}{sys.version_info.serial}"
        self.assertEqual(actual, expect_version)


if __name__ == "__main__":
    unittest.main()