Function test_disk_label

test/cloudtest/test_compute.py:128–142

Test that cluster replicas have the correct materialize.cloud/disk labels

(mz: MaterializeApplication)

def test_disk_label(mz: MaterializeApplication) -> None:
    """Test that cluster replicas have the correct materialize.cloud/disk labels"""

    mz.environmentd.sql("CREATE CLUSTER disk SIZE = 'scale=1,workers=2'")

    cluster_id, replica_id = mz.environmentd.sql_query(
        "SELECT mz_clusters.id, mz_cluster_replicas.id FROM mz_cluster_replicas JOIN mz_clusters ON mz_cluster_replicas.cluster_id = mz_clusters.id WHERE mz_clusters.name = 'disk'"
    )[0]
    assert cluster_id is not None
    assert replica_id is not None

    node_selectors = get_node_selector(mz, cluster_id, replica_id)
    assert node_selectors == '\'{"materialize.cloud/disk":"true"}\'', node_selectors

    mz.environmentd.sql("DROP CLUSTER disk CASCADE")


@pytest.mark.skip(