Function test_http_sql

src/environmentd/tests/server.rs:1042–1188

()

#[mz_ore::test]
#[allow(clippy::disallowed_methods)]
fn test_http_sql() {
    // Datadriven directives for WebSocket are "ws-text" and "ws-binary" to send
    // text or binary websocket messages that are the input. Output is
    // everything until and including the next ReadyForQuery message. An
    // optional "rows=N" argument can be given in the directive to produce
    // datadriven output after N rows. Any directive with rows=N should be the
    // final directive in a file, since it leaves the websocket in a
    // mid-statement state.
    // A "fixtimestamp=true" argument can be given to replace timestamps with "<TIMESTAMP>".
    // A "fixid=true" argument can be given to replace IDs with "<ID>".
    //
    // Datadriven directive for HTTP POST is "http". Input and output are the
    // documented JSON formats.

    let fixtimestamp_replacements = [(Regex::new(r#"\d{13}(\.0)?"#).unwrap(), "<TIMESTAMP>")];
    let fixid_replacements = [
        (
            Regex::new(r#"\\"(User|System|Transient)\\": \d+"#).unwrap(),
            "<ID>",
        ),
        (Regex::new(r#"\b[ust]\d+\b"#).unwrap(), "<ID>"),
        (Regex::new(r#"\\n[ust]\d+\b"#).unwrap(), "\\n<ID>"),
    ];

    datadriven::walk("tests/testdata/http", |f| {
        let server = test_util::TestHarness::default().start_blocking();

        // Grant all privileges to default http user.
        // TODO(jkosh44) The HTTP endpoint has a special default user while the WS endpoint just
        //  uses the "materialize" role. This is probably wrong. They should either both user
        //  materialize, both use the same special default user, each have their own special
        //  default user.
        {
            let mut super_user = server
                .pg_config_internal()
                .user(&SYSTEM_USER.name)
                .connect(postgres::NoTls)
                .unwrap();
            super_user
                .batch_execute(&format!("CREATE ROLE {}", &HTTP_DEFAULT_USER.name))
                .unwrap();
            super_user
                .batch_execute(&format!(
                    "GRANT ALL PRIVILEGES ON SYSTEM TO {}",
                    &HTTP_DEFAULT_USER.name
                ))
                .unwrap();
            super_user
                .batch_execute(&format!(
                    "GRANT ALL PRIVILEGES ON CLUSTER quickstart TO {}",
                    &HTTP_DEFAULT_USER.name
                ))
                .unwrap();
            super_user
                .batch_execute(&format!(
                    "GRANT ALL PRIVILEGES ON DATABASE materialize TO {}",
                    &HTTP_DEFAULT_USER.name
                ))