Function test_time

src/environmentd/tests/sql.rs:316–362

()

#[mz_ore::test]
#[allow(clippy::disallowed_methods)]
fn test_time() {
    let server = test_util::TestHarness::default()
        .unsafe_mode()
        .start_blocking();
    server.enable_feature_flags(&["unsafe_enable_unsafe_functions"]);
    let mut client = server.connect(postgres::NoTls).unwrap();

    // Confirm that `now()` and `current_timestamp()` both return a
    // DateTime<Utc>, but don't assert specific times.
    let row = client
        .query_one("SELECT now(), current_timestamp()", &[])
        .unwrap();
    let _ = row.get::<_, DateTime<Utc>>(0);
    let _ = row.get::<_, DateTime<Utc>>(1);

    // Confirm calls to now() return the same DateTime<Utc> both inside and
    // outside of subqueries.
    let row = client
        .query_one("SELECT now(), (SELECT now())", &[])
        .unwrap();
    assert_eq!(
        row.get::<_, DateTime<Utc>>(0),
        row.get::<_, DateTime<Utc>>(1)
    );

    // Ensure that EXPLAIN selects a timestamp for `now()` and
    // `current_timestamp()`, though we don't care what the timestamp is.
    let rows = client
        .query(
            "EXPLAIN OPTIMIZED PLAN AS VERBOSE TEXT FOR SELECT now(), current_timestamp()",
            &[],
        )
        .unwrap();
    assert_eq!(1, rows.len());

    // Test that `mz_sleep` causes a delay of at least the appropriate time.
    let start = Instant::now();
    client
        .batch_execute("SELECT mz_unsafe.mz_sleep(0.3)")
        .unwrap();
    let elapsed = start.elapsed();
    assert!(
        elapsed >= Duration::from_millis(300),
        "start.elapsed() = {:?}",
        elapsed
    );
}

#[mz_ore::test]
#[allow(clippy::disallowed_methods)]