Function test_utilization_hold

src/environmentd/tests/sql.rs:1451–1579

()

#[ignore] // TODO: Reenable when https://linear.app/materializeinc/issue/SQL-407 is fixed
#[allow(clippy::disallowed_methods)]
async fn test_utilization_hold() {
    const THIRTY_DAYS_MS: u64 = 30 * 24 * 60 * 60 * 1000;
    // `mz_catalog_server` tests indexes, `quickstart` tests tables.
    // The bool determines whether we are testing indexes.
    const CLUSTERS_TO_TRY: &[&str] = &["mz_catalog_server", "quickstart"];
    const QUERIES_TO_TRY: &[&str] = &["SELECT * FROM mz_internal.mz_cluster_replica_statuses"];

    let now_millis = 619388520000;
    let past_millis = now_millis - THIRTY_DAYS_MS;
    let past_since = Timestamp::from(past_millis);

    let now = Arc::new(Mutex::new(past_millis));
    let now_fn = {
        let timestamp = Arc::clone(&now);
        NowFn::from(move || *timestamp.lock().unwrap())
    };
    let data_dir = tempfile::tempdir().unwrap();

    // Create the server with the past time, to make sure the table is created.
    let server = test_util::TestHarness::default()
        .with_now(now_fn)
        .data_directory(data_dir.path())
        .start()
        .await;

    // Fast-forward time to make sure the table is still readable at the old time.
    *now.lock().unwrap() = now_millis;

    let client = server.connect().await.unwrap();

    for q in QUERIES_TO_TRY {
        let explain_q = &format!("EXPLAIN TIMESTAMP AS JSON FOR {q}");
        for cluster in CLUSTERS_TO_TRY {
            client
                .execute(&format!("SET cluster={cluster}"), &[])
                .await
                .unwrap();

            let row = client.query_one(explain_q, &[]).await.unwrap();
            let explain: String = row.get(0);
            let explain: TimestampExplanation = serde_json::from_str(&explain).unwrap();

            // Assert that we actually used the indexes/tables, as required
            for s in &explain.sources {
                assert!(s.name.ends_with("compute)"));
            }

            // If we're not in EpochMilliseconds, the timestamp math below is invalid, so assert that here.
            assert!(matches!(
                explain.determination.timestamp_context,
                TimestampContext::TimelineTimestamp {
                    timeline: Timeline::EpochMilliseconds,
                    ..
                }
            ));
            let since = explain
                .determination
                .since