Function workflow_default

test/cloud-canary/mzcompose.py:223–310

Deploy the current source to the cloud and run tests.

(c: Composition, parser: WorkflowArgumentParser)

def workflow_default(c: Composition, parser: WorkflowArgumentParser) -> None:
    """Deploy the current source to the cloud and run tests."""

    parser.add_argument(
        "--cleanup",
        default=True,
        action=argparse.BooleanOptionalAction,
        help="Destroy the region at the end of the workflow.",
    )
    parser.add_argument(
        "--version-check",
        default=True,
        action=argparse.BooleanOptionalAction,
        help="Perform a version check.",
    )

    parser.add_argument(
        "td_files",
        nargs="*",
        default=["*.td"],
        help="run against the specified files",
    )

    args = parser.parse_args()

    if not os.getenv("BUILDKITE_COMMIT"):
        raise UIError(
            "BUILDKITE_COMMIT must be set (and valid) when running cloud-canary"
        )

    files = list(
        itertools.chain.from_iterable(
            [
                glob.glob(file_glob, root_dir=MZ_ROOT / "test" / "cloud-canary")
                for file_glob in args.td_files
            ]
        )
    )

    if args.cleanup:
        disable_region(c)

    try:
        test_failed = True
        print(f"Enabling region using Mz version {VERSION} ...")
        try:
            c.run("mz", "region", "enable", "--version", VERSION)
        except UIError:
            # Work around https://github.com/MaterializeInc/database-issues/issues/4989
            pass

        time.sleep(10)

        assert "materialize.cloud" in c.cloud_hostname()
        wait_for_cloud(c)

        if args.version_check:
            version_check(c)