drizzle 0.1.11

A type-safe SQL query builder for Rust
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

//! PostgreSQL array operator tests
//!
//! Tests for PostgreSQL-specific array operators (@>, <@, &&).

#![cfg(any(feature = "postgres-sync", feature = "tokio-postgres"))]

use crate::common::schema::postgres::*;
use drizzle::postgres::expr::{array_contained, array_contains, array_overlaps};
use drizzle::postgres::prelude::*;

// Test SQL generation for array_contains (@>) operator
#[drizzle::test]
fn array_contains_sql_generation(db: &mut TestDb<SimpleSchema>) {
    let SimpleSchema { simple } = schema;

    // Note: This test verifies SQL generation. The simple.name column is TEXT,
    // not an array type, but the SQL generation should still work correctly.
    // In production, this would be used with actual TEXT[] columns.
    let stmt = db
        .select(())
        .from(simple)
        .r#where(array_contains(simple.name, "test"));

    let sql = stmt.to_sql().sql();

    // Verify the @> operator is present in the generated SQL
    assert!(sql.contains("@>"), "Expected @> operator in SQL: {}", sql);
}

// Test SQL generation for array_contained (<@) operator
#[drizzle::test]
fn array_contained_sql_generation(db: &mut TestDb<SimpleSchema>) {
    let SimpleSchema { simple } = schema;

    let stmt = db
        .select(())
        .from(simple)
        .r#where(array_contained(simple.name, "test"));

    let sql = stmt.to_sql().sql();

    // Verify the <@ operator is present in the generated SQL
    assert!(sql.contains("<@"), "Expected <@ operator in SQL: {}", sql);
}

// Test SQL generation for array_overlaps (&&) operator
#[drizzle::test]
fn array_overlaps_sql_generation(db: &mut TestDb<SimpleSchema>) {
    let SimpleSchema { simple } = schema;

    let stmt = db
        .select(())
        .from(simple)
        .r#where(array_overlaps(simple.name, "test"));

    let sql = stmt.to_sql().sql();

    // Verify the && operator is present in the generated SQL
    assert!(sql.contains("&&"), "Expected && operator in SQL: {}", sql);
}

// Test that array operators work with method syntax via ArrayExprExt trait
#[drizzle::test]
fn array_ops_method_syntax(db: &mut TestDb<SimpleSchema>) {
    use drizzle::postgres::expr::ArrayExprExt;

    let SimpleSchema { simple } = schema;

    // Test method syntax for array_contains
    let stmt = db
        .select(())
        .from(simple)
        .r#where(simple.name.array_contains("test"));

    let sql = stmt.to_sql().sql();
    assert!(sql.contains("@>"), "Expected @> operator in SQL: {}", sql);

    // Test method syntax for array_contained
    let stmt = db
        .select(())
        .from(simple)
        .r#where(simple.name.array_contained("test"));

    let sql = stmt.to_sql().sql();
    assert!(sql.contains("<@"), "Expected <@ operator in SQL: {}", sql);

    // Test method syntax for array_overlaps
    let stmt = db
        .select(())
        .from(simple)
        .r#where(simple.name.array_overlaps("test"));

    let sql = stmt.to_sql().sql();
    assert!(sql.contains("&&"), "Expected && operator in SQL: {}", sql);
}