28 lines
741 B
Rust
28 lines
741 B
Rust
|
#![cfg(test)]
|
||
|
|
|
||
|
|
use postgres::Client;
|
||
|
|
use postgres_types::{FromSqlOwned, ToSql};
|
||
|
|
use std::fmt;
|
||
|
|
|
||
|
|
mod composites;
|
||
|
|
mod domains;
|
||
|
|
mod enums;
|
||
|
|
|
||
|
|
pub fn test_type<T, S>(conn: &mut Client, sql_type: &str, checks: &[(T, S)])
|
||
|
|
where
|
||
|
|
T: PartialEq + FromSqlOwned + ToSql + Sync,
|
||
|
|
S: fmt::Display,
|
||
|
|
{
|
||
|
|
for &(ref val, ref repr) in checks.iter() {
|
||
|
|
let stmt = conn
|
||
|
|
.prepare(&*format!("SELECT {}::{}", *repr, sql_type))
|
||
|
|
.unwrap();
|
||
|
|
let result = conn.query_one(&stmt, &[]).unwrap().get(0);
|
||
|
|
assert_eq!(val, &result);
|
||
|
|
|
||
|
|
let stmt = conn.prepare(&*format!("SELECT $1::{}", sql_type)).unwrap();
|
||
|
|
let result = conn.query_one(&stmt, &[val]).unwrap().get(0);
|
||
|
|
assert_eq!(val, &result);
|
||
|
|
}
|
||
|
|
}
|