uqa-sql 0.5.0

PostgreSQL-compatible SQL compiler built on libpg_query
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

//
// Unified Query Algebra
//
// Copyright (c) 2023-2026 Cognica, Inc.
//

use super::{flatten_set_arguments, SetArgument};

fn text(value: &str) -> SetArgument {
    SetArgument::Text(value.into())
}

#[test]
fn list_parameters_join_their_arguments_and_quote_identifiers() {
    assert_eq!(
        flatten_set_arguments(
            "search_path",
            &[text("a$b"), text("public"), text("My S"), text("$user")]
        )
        .unwrap(),
        "\"a$b\", public, \"My S\", \"$user\""
    );
    assert_eq!(
        flatten_set_arguments("search_path", &[text("")]).unwrap(),
        "\"\""
    );
    assert_eq!(
        flatten_set_arguments("search_path", &[text("a,b")]).unwrap(),
        "\"a,b\""
    );
    assert_eq!(
        flatten_set_arguments("DateStyle", &[text("ISO"), text("DMY")]).unwrap(),
        "ISO, DMY"
    );
}

#[test]
fn other_parameters_take_one_argument_as_written() {
    assert_eq!(
        flatten_set_arguments("statement_timeout", &[SetArgument::Number("1.5".into())]).unwrap(),
        "1.5"
    );
    assert_eq!(
        flatten_set_arguments("work_mem", &[SetArgument::Integer(65536)]).unwrap(),
        "65536"
    );
    assert_eq!(
        flatten_set_arguments("my.var", &[text("My Value")]).unwrap(),
        "My Value"
    );
    let error = flatten_set_arguments(
        "statement_timeout",
        &[SetArgument::Integer(1), SetArgument::Integer(2)],
    )
    .unwrap_err();
    assert_eq!(error.sqlstate(), Some("22023"));
    assert_eq!(
        error.to_string(),
        "SET statement_timeout takes only one argument"
    );
}