1
-
2
1
/*
3
2
Portions Copyright 2020-2021 ZomboDB, LLC.
4
3
Portions Copyright 2021-2023 Technology Concepts & Design, Inc. <[email protected] >
@@ -11,44 +10,44 @@ Use of this source code is governed by the PostgreSQL license that can be found
11
10
#[ cfg( any( test, feature = "pg_test" ) ) ]
12
11
#[ pgrx:: pg_schema]
13
12
mod tests {
14
- use pgrx:: prelude:: * ;
13
+ use pgrx:: prelude:: * ;
15
14
16
15
#[ pg_test]
17
- #[ search_path( @extschema@) ]
18
- fn plrust_call_1st ( ) -> spi:: Result < ( ) > {
19
- let definition = r#"
16
+ #[ search_path( @extschema@) ]
17
+ fn plrust_call_1st ( ) -> spi:: Result < ( ) > {
18
+ let definition = r#"
20
19
CREATE FUNCTION ret_1st(a int, b int)
21
20
RETURNS int AS
22
21
$$
23
22
Ok(a)
24
23
$$ LANGUAGE plrust;
25
24
"# ;
26
- Spi :: run ( definition) ?;
27
- let result_1 = Spi :: get_one :: < i32 > ( "SELECT ret_1st(1, 2);\n " ) ;
28
- assert_eq ! ( Ok ( Some ( 1 ) ) , result_1) ; // may get: Some(1)
29
- Ok ( ( ) )
30
- }
25
+ Spi :: run ( definition) ?;
26
+ let result_1 = Spi :: get_one :: < i32 > ( "SELECT ret_1st(1, 2);\n " ) ;
27
+ assert_eq ! ( Ok ( Some ( 1 ) ) , result_1) ; // may get: Some(1)
28
+ Ok ( ( ) )
29
+ }
31
30
32
- #[ pg_test]
33
- #[ search_path( @extschema@) ]
34
- fn plrust_call_2nd ( ) -> spi:: Result < ( ) > {
35
- let definition = r#"
31
+ #[ pg_test]
32
+ #[ search_path( @extschema@) ]
33
+ fn plrust_call_2nd ( ) -> spi:: Result < ( ) > {
34
+ let definition = r#"
36
35
CREATE FUNCTION ret_2nd(a int, b int)
37
36
RETURNS int AS
38
37
$$
39
38
Ok(b)
40
39
$$ LANGUAGE plrust;
41
40
"# ;
42
- Spi :: run ( definition) ?;
43
- let result_2 = Spi :: get_one :: < i32 > ( "SELECT ret_2nd(1, 2);\n " ) ;
44
- assert_eq ! ( Ok ( Some ( 2 ) ) , result_2) ; // may get: Some(2)
45
- Ok ( ( ) )
46
- }
41
+ Spi :: run ( definition) ?;
42
+ let result_2 = Spi :: get_one :: < i32 > ( "SELECT ret_2nd(1, 2);\n " ) ;
43
+ assert_eq ! ( Ok ( Some ( 2 ) ) , result_2) ; // may get: Some(2)
44
+ Ok ( ( ) )
45
+ }
47
46
48
- #[ pg_test]
49
- #[ search_path( @extschema@) ]
50
- fn plrust_call_me ( ) -> spi:: Result < ( ) > {
51
- let definition = r#"
47
+ #[ pg_test]
48
+ #[ search_path( @extschema@) ]
49
+ fn plrust_call_me ( ) -> spi:: Result < ( ) > {
50
+ let definition = r#"
52
51
CREATE FUNCTION pick_ret(a int, b int, pick int)
53
52
RETURNS int AS
54
53
$$
@@ -59,22 +58,22 @@ fn plrust_call_me() -> spi::Result<()> {
59
58
})
60
59
$$ LANGUAGE plrust;
61
60
"# ;
62
- Spi :: run ( definition) ?;
63
- let result_a = Spi :: get_one :: < i32 > ( "SELECT pick_ret(3, 4, 0);" ) ;
64
- let result_b = Spi :: get_one :: < i32 > ( "SELECT pick_ret(5, 6, 1);" ) ;
65
- let result_c = Spi :: get_one :: < i32 > ( "SELECT pick_ret(7, 8, 2);" ) ;
66
- let result_z = Spi :: get_one :: < i32 > ( "SELECT pick_ret(9, 99, -1);" ) ;
67
- assert_eq ! ( Ok ( Some ( 3 ) ) , result_a) ; // may get: Some(4) or None
68
- assert_eq ! ( Ok ( Some ( 6 ) ) , result_b) ; // may get: None
69
- assert_eq ! ( Ok ( None ) , result_c) ;
70
- assert_eq ! ( Ok ( None ) , result_z) ;
71
- Ok ( ( ) )
72
- }
61
+ Spi :: run ( definition) ?;
62
+ let result_a = Spi :: get_one :: < i32 > ( "SELECT pick_ret(3, 4, 0);" ) ;
63
+ let result_b = Spi :: get_one :: < i32 > ( "SELECT pick_ret(5, 6, 1);" ) ;
64
+ let result_c = Spi :: get_one :: < i32 > ( "SELECT pick_ret(7, 8, 2);" ) ;
65
+ let result_z = Spi :: get_one :: < i32 > ( "SELECT pick_ret(9, 99, -1);" ) ;
66
+ assert_eq ! ( Ok ( Some ( 3 ) ) , result_a) ; // may get: Some(4) or None
67
+ assert_eq ! ( Ok ( Some ( 6 ) ) , result_b) ; // may get: None
68
+ assert_eq ! ( Ok ( None ) , result_c) ;
69
+ assert_eq ! ( Ok ( None ) , result_z) ;
70
+ Ok ( ( ) )
71
+ }
73
72
74
- #[ pg_test]
75
- #[ search_path( @extschema@) ]
76
- fn plrust_call_me_call_me ( ) -> spi:: Result < ( ) > {
77
- let definition = r#"
73
+ #[ pg_test]
74
+ #[ search_path( @extschema@) ]
75
+ fn plrust_call_me_call_me ( ) -> spi:: Result < ( ) > {
76
+ let definition = r#"
78
77
CREATE FUNCTION ret_1st(a int, b int)
79
78
RETURNS int AS
80
79
$$
@@ -97,19 +96,19 @@ fn plrust_call_me_call_me() -> spi::Result<()> {
97
96
})
98
97
$$ LANGUAGE plrust;
99
98
"# ;
100
- Spi :: run ( definition) ?;
101
- let result_1 = Spi :: get_one :: < i32 > ( "SELECT ret_1st(1, 2);\n " ) ;
102
- let result_2 = Spi :: get_one :: < i32 > ( "SELECT ret_2nd(1, 2);\n " ) ;
103
- let result_a = Spi :: get_one :: < i32 > ( "SELECT pick_ret(3, 4, 0);" ) ;
104
- let result_b = Spi :: get_one :: < i32 > ( "SELECT pick_ret(5, 6, 1);" ) ;
105
- let result_c = Spi :: get_one :: < i32 > ( "SELECT pick_ret(7, 8, 2);" ) ;
106
- let result_z = Spi :: get_one :: < i32 > ( "SELECT pick_ret(9, 99, -1);" ) ;
107
- assert_eq ! ( Ok ( None ) , result_z) ;
108
- assert_eq ! ( Ok ( None ) , result_c) ;
109
- assert_eq ! ( Ok ( Some ( 6 ) ) , result_b) ; // may get: None
110
- assert_eq ! ( Ok ( Some ( 3 ) ) , result_a) ; // may get: Some(4) or None
111
- assert_eq ! ( Ok ( Some ( 2 ) ) , result_2) ; // may get: Some(1)
112
- assert_eq ! ( Ok ( Some ( 1 ) ) , result_1) ; // may get: Some(2)
113
- Ok ( ( ) )
99
+ Spi :: run ( definition) ?;
100
+ let result_1 = Spi :: get_one :: < i32 > ( "SELECT ret_1st(1, 2);\n " ) ;
101
+ let result_2 = Spi :: get_one :: < i32 > ( "SELECT ret_2nd(1, 2);\n " ) ;
102
+ let result_a = Spi :: get_one :: < i32 > ( "SELECT pick_ret(3, 4, 0);" ) ;
103
+ let result_b = Spi :: get_one :: < i32 > ( "SELECT pick_ret(5, 6, 1);" ) ;
104
+ let result_c = Spi :: get_one :: < i32 > ( "SELECT pick_ret(7, 8, 2);" ) ;
105
+ let result_z = Spi :: get_one :: < i32 > ( "SELECT pick_ret(9, 99, -1);" ) ;
106
+ assert_eq ! ( Ok ( None ) , result_z) ;
107
+ assert_eq ! ( Ok ( None ) , result_c) ;
108
+ assert_eq ! ( Ok ( Some ( 6 ) ) , result_b) ; // may get: None
109
+ assert_eq ! ( Ok ( Some ( 3 ) ) , result_a) ; // may get: Some(4) or None
110
+ assert_eq ! ( Ok ( Some ( 2 ) ) , result_2) ; // may get: Some(1)
111
+ assert_eq ! ( Ok ( Some ( 1 ) ) , result_1) ; // may get: Some(2)
112
+ Ok ( ( ) )
113
+ }
114
114
}
115
- }
0 commit comments