uqa-sql 0.5.4

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
62
63
64

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

use super::*;

#[test]
fn boolean_test_analysis_matches_postgresql_errors_before_evaluation() {
    let cases: serde_json::Value =
        serde_json::from_str(include_str!("boolean_tests.json")).unwrap();
    for case in cases.as_array().unwrap() {
        let sql = case["sql"].as_str().unwrap();
        let plan = UnifiedPlan::lower(crate::compile(sql).unwrap().remove(0));
        let error = infer_prepared_parameter_types(&NoRoutines, &plan, &[], &assignment_context())
            .unwrap_err();
        assert_eq!(
            error.sqlstate(),
            case["error"]["sqlstate"].as_str(),
            "{sql}: {error}"
        );
        assert_eq!(
            error.to_string(),
            case["error"]["message"].as_str().unwrap(),
            "{sql}"
        );
    }
}

#[test]
fn boolean_tests_infer_parameters_without_retyping_declared_inputs() {
    use crate::ast::BooleanTest;
    for test in [
        BooleanTest::IsTrue,
        BooleanTest::IsNotTrue,
        BooleanTest::IsFalse,
        BooleanTest::IsNotFalse,
        BooleanTest::IsUnknown,
        BooleanTest::IsNotUnknown,
    ] {
        let sql = format!("SELECT $1 {}", test.label());
        let plan = UnifiedPlan::lower(crate::compile(&sql).unwrap().remove(0));
        let types =
            infer_prepared_parameter_types(&NoRoutines, &plan, &[None], &assignment_context())
                .unwrap();
        assert_eq!(types, vec![Some(ColumnType::Boolean)]);
        let error = infer_prepared_parameter_types(
            &NoRoutines,
            &plan,
            &[Some(ColumnType::Text)],
            &assignment_context(),
        )
        .unwrap_err();
        assert_eq!(error.sqlstate(), Some("42804"));
        assert_eq!(
            error.to_string(),
            format!(
                "argument of {} must be type boolean, not type text",
                test.label()
            )
        );
    }
}