simplex-chat/scripts/nix/test-select-intel-setup.py

import importlib.util
import pathlib
import unittest


spec = importlib.util.spec_from_file_location(
    "select_intel_setup", pathlib.Path(__file__).with_name("select-intel-setup.py")
)
selector = importlib.util.module_from_spec(spec)
spec.loader.exec_module(selector)

OLD = "/nix/store/vghjgkr64qjigddh3gnf3c103wbvpxmz-ghc-boot-lib-ghc-boot-9.14.drv"
NEW = "/nix/store/" + "0" * 32 + "-ghc-boot-lib-ghc-boot-9.14.drv"


class FailedComponentSelection(unittest.TestCase):
    def test_current_and_rebuilt_component(self):
        for path in [OLD, NEW]:
            with self.subTest(path=path):
                self.assertEqual(
                    selector.select_failed_component(f"error: Cannot build '{path}'."),
                    path,
                )

    def test_old_nix_error_and_repeated_path(self):
        log = f"error: builder for '{NEW}' failed with exit code 134\n"
        log += f"error: Cannot build '{NEW}'.\n"
        self.assertEqual(selector.select_failed_component(log), NEW)

    def test_unrelated_or_progress_paths_are_ignored(self):
        for log in [
            f"building '{NEW}'...",
            f"error: Cannot build '{NEW.replace('-9.14.drv', '-9.12.drv')}'.",
            f"error: Cannot build '{NEW.replace('-lib-', '-exe-')}'.",
            f"error: Cannot build '{NEW.replace('/nix/store/', '/tmp/')}'.",
            f"error: Cannot build '{NEW.replace('0' * 32, 'e' * 32)}'.",
        ]:
            with self.subTest(log=log):
                self.assertIsNone(selector.select_failed_component(log))

    def test_multiple_components_are_rejected(self):
        with self.assertRaisesRegex(ValueError, "multiple"):
            selector.select_failed_component(
                f"error: Cannot build '{OLD}'.\nerror: Cannot build '{NEW}'."
            )


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