Function workflow_default

test/launchdarkly/mzcompose.py:109–284

(c: Composition)

def workflow_default(c: Composition) -> None:
    if LAUNCHDARKLY_API_TOKEN is None:
        raise UIError("Missing LAUNCHDARKLY_API_TOKEN environment variable")
    if LAUNCHDARKLY_SDK_KEY is None:
        raise UIError("Missing LAUNCHDARKLY_SDK_KEY environment variable")

    # Create a LaunchDarkly client that simulates somebody interacting
    # with the LaunchDarkly frontend.
    ld_client = LaunchDarklyClient(
        configuration=launchdarkly_api.Configuration(
            api_key=dict(ApiKey=LAUNCHDARKLY_API_TOKEN),
        ),
        project_key=environ.get("LAUNCHDARKLY_PROJECT_KEY", "default"),
        environment_key=environ.get("LAUNCHDARKLY_ENVIRONMENT_KEY", "ci-cd"),
    )

    try:
        c.up(Service("testdrive", idle=True))

        # Assert that the default max_result_size is served when sync is disabled.
        with c.override(Materialized(external_metadata_store=True)):
            c.up("materialized")
            c.testdrive("\n".join(["> SHOW max_result_size", "1GB"]))
            c.stop("materialized")

        # Create a test feature flag unique for this test run. Based on the
        # MZ_LAUNCHDARKLY_KEY_MAP value, the test feature will be mapped to the
        # max_result_size system parameter.
        ld_client.create_flag(
            LD_FEATURE_FLAG_KEY,
            tags=(
                ["ci-test", f"gh-{BUILDKITE_PULL_REQUEST}"]
                if BUILDKITE_PULL_REQUEST
                else ["ci-test"]
            ),
        )
        # Turn on targeting. The default rule will now serve 2GiB for the test
        # feature.
        ld_client.update_targeting(
            LD_FEATURE_FLAG_KEY,
            on=True,
        )

        # 3 seconds should be enough to avoid race conditions between the update
        # above and the query below.
        sleep(3)

        # Assert that the value is as expected after the initial parameter sync.
        with c.override(
            Materialized(
                environment_extra=[
                    f"MZ_LAUNCHDARKLY_SDK_KEY={LAUNCHDARKLY_SDK_KEY}",
                    f"MZ_LAUNCHDARKLY_KEY_MAP=max_result_size={LD_FEATURE_FLAG_KEY}",
                ],
                additional_system_parameter_defaults={
                    "log_filter": "mz_adapter::catalog=debug,mz_adapter::config=debug",
                },
                external_metadata_store=True,