227 lines
8.0 KiB
Rust
227 lines
8.0 KiB
Rust
use serialize::json::Json;
|
|
use std::collections::HashMap;
|
|
use std::f32;
|
|
use std::f64;
|
|
use std::fmt;
|
|
use std::num::Float;
|
|
use std::io::net::ip::IpAddr;
|
|
|
|
use postgres::{Connection, SslMode};
|
|
use postgres::types::{ToSql, FromSql};
|
|
|
|
#[cfg(feature = "uuid")]
|
|
mod uuid;
|
|
mod time;
|
|
|
|
fn test_type<T: PartialEq+FromSql+ToSql, S: fmt::Display>(sql_type: &str, checks: &[(T, S)]) {
|
|
let conn = or_panic!(Connection::connect("postgres://postgres@localhost", &SslMode::None));
|
|
for &(ref val, ref repr) in checks.iter() {
|
|
let stmt = or_panic!(conn.prepare(&*format!("SELECT {}::{}", *repr, sql_type)));
|
|
let result = or_panic!(stmt.query(&[])).next().unwrap().get(0us);
|
|
assert!(val == &result);
|
|
|
|
let stmt = or_panic!(conn.prepare(&*format!("SELECT $1::{}", sql_type)));
|
|
let result = or_panic!(stmt.query(&[val])).next().unwrap().get(0us);
|
|
assert!(val == &result);
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_bool_params() {
|
|
test_type("BOOL", &[(Some(true), "'t'"), (Some(false), "'f'"),
|
|
(None, "NULL")]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_i8_params() {
|
|
test_type("\"char\"", &[(Some('a' as i8), "'a'"), (None, "NULL")]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_name_params() {
|
|
test_type("NAME", &[(Some("hello world".to_string()), "'hello world'"),
|
|
(Some("イロハニホヘト チリヌルヲ".to_string()), "'イロハニホヘト チリヌルヲ'"),
|
|
(None, "NULL")]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_i16_params() {
|
|
test_type("SMALLINT", &[(Some(15001i16), "15001"),
|
|
(Some(-15001i16), "-15001"), (None, "NULL")]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_i32_params() {
|
|
test_type("INT", &[(Some(2147483548i32), "2147483548"),
|
|
(Some(-2147483548i32), "-2147483548"), (None, "NULL")]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_oid_params() {
|
|
test_type("OID", &[(Some(2147483548u32), "2147483548"),
|
|
(Some(4000000000), "4000000000"), (None, "NULL")]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_i64_params() {
|
|
test_type("BIGINT", &[(Some(9223372036854775708i64), "9223372036854775708"),
|
|
(Some(-9223372036854775708i64), "-9223372036854775708"),
|
|
(None, "NULL")]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_f32_params() {
|
|
test_type("REAL", &[(Some(f32::INFINITY), "'infinity'"),
|
|
(Some(f32::NEG_INFINITY), "'-infinity'"),
|
|
(Some(1000.55), "1000.55"), (None, "NULL")]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_f64_params() {
|
|
test_type("DOUBLE PRECISION", &[(Some(f64::INFINITY), "'infinity'"),
|
|
(Some(f64::NEG_INFINITY), "'-infinity'"),
|
|
(Some(10000.55), "10000.55"),
|
|
(None, "NULL")]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_varchar_params() {
|
|
test_type("VARCHAR", &[(Some("hello world".to_string()), "'hello world'"),
|
|
(Some("イロハニホヘト チリヌルヲ".to_string()), "'イロハニホヘト チリヌルヲ'"),
|
|
(None, "NULL")]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_text_params() {
|
|
test_type("TEXT", &[(Some("hello world".to_string()), "'hello world'"),
|
|
(Some("イロハニホヘト チリヌルヲ".to_string()), "'イロハニホヘト チリヌルヲ'"),
|
|
(None, "NULL")]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_bpchar_params() {
|
|
let conn = or_panic!(Connection::connect("postgres://postgres@localhost", &SslMode::None));
|
|
or_panic!(conn.execute("CREATE TEMPORARY TABLE foo (
|
|
id SERIAL PRIMARY KEY,
|
|
b CHAR(5)
|
|
)", &[]));
|
|
or_panic!(conn.execute("INSERT INTO foo (b) VALUES ($1), ($2), ($3)",
|
|
&[&Some("12345"), &Some("123"), &None::<&'static str>]));
|
|
let stmt = or_panic!(conn.prepare("SELECT b FROM foo ORDER BY id"));
|
|
let res = or_panic!(stmt.query(&[]));
|
|
|
|
assert_eq!(vec!(Some("12345".to_string()), Some("123 ".to_string()), None),
|
|
res.map(|row| row.get(0us)).collect::<Vec<_>>());
|
|
}
|
|
|
|
#[test]
|
|
fn test_citext_params() {
|
|
let conn = or_panic!(Connection::connect("postgres://postgres@localhost", &SslMode::None));
|
|
or_panic!(conn.execute("CREATE TEMPORARY TABLE foo (
|
|
id SERIAL PRIMARY KEY,
|
|
b CITEXT
|
|
)", &[]));
|
|
or_panic!(conn.execute("INSERT INTO foo (b) VALUES ($1), ($2), ($3)",
|
|
&[&Some("foobar"), &Some("FooBar"), &None::<&'static str>]));
|
|
let stmt = or_panic!(conn.prepare("SELECT id FROM foo WHERE b = 'FOOBAR' ORDER BY id"));
|
|
let res = or_panic!(stmt.query(&[]));
|
|
|
|
assert_eq!(vec!(Some(1i32), Some(2i32)), res.map(|row| row.get(0us)).collect::<Vec<_>>());
|
|
}
|
|
|
|
#[test]
|
|
fn test_bytea_params() {
|
|
test_type("BYTEA", &[(Some(vec!(0u8, 1, 2, 3, 254, 255)), "'\\x00010203feff'"),
|
|
(None, "NULL")]);
|
|
}
|
|
|
|
#[test]
|
|
fn test_json_params() {
|
|
test_type("JSON", &[(Some(Json::from_str("[10, 11, 12]").unwrap()),
|
|
"'[10, 11, 12]'"),
|
|
(Some(Json::from_str("{\"f\": \"asd\"}").unwrap()),
|
|
"'{\"f\": \"asd\"}'"),
|
|
(None, "NULL")])
|
|
}
|
|
|
|
#[test]
|
|
fn test_jsonb_params() {
|
|
if option_env!("TRAVIS").is_some() { return } // Travis doesn't have Postgres 9.4 yet
|
|
test_type("JSONB", &[(Some(Json::from_str("[10, 11, 12]").unwrap()),
|
|
"'[10, 11, 12]'"),
|
|
(Some(Json::from_str("{\"f\": \"asd\"}").unwrap()),
|
|
"'{\"f\": \"asd\"}'"),
|
|
(None, "NULL")])
|
|
}
|
|
|
|
#[test]
|
|
fn test_inet_params() {
|
|
test_type("INET", &[(Some("127.0.0.1".parse::<IpAddr>().unwrap()),
|
|
"'127.0.0.1'"),
|
|
(Some("2001:0db8:85a3:0000:0000:8a2e:0370:7334".parse::<IpAddr>().unwrap()),
|
|
"'2001:0db8:85a3:0000:0000:8a2e:0370:7334'"),
|
|
(None, "NULL")])
|
|
}
|
|
|
|
#[test]
|
|
fn test_cidr_params() {
|
|
test_type("CIDR", &[(Some("127.0.0.1".parse::<IpAddr>().unwrap()),
|
|
"'127.0.0.1'"),
|
|
(Some("2001:0db8:85a3:0000:0000:8a2e:0370:7334".parse::<IpAddr>().unwrap()),
|
|
"'2001:0db8:85a3:0000:0000:8a2e:0370:7334'"),
|
|
(None, "NULL")])
|
|
}
|
|
|
|
#[test]
|
|
fn test_hstore_params() {
|
|
macro_rules! make_map {
|
|
($($k:expr => $v:expr),+) => ({
|
|
let mut map = HashMap::new();
|
|
$(map.insert($k, $v);)+
|
|
map
|
|
})
|
|
}
|
|
test_type("hstore",
|
|
&[(Some(make_map!("a".to_string() => Some("1".to_string()))), "'a=>1'"),
|
|
(Some(make_map!("hello".to_string() => Some("world!".to_string()),
|
|
"hola".to_string() => Some("mundo!".to_string()),
|
|
"what".to_string() => None)),
|
|
"'hello=>world!,hola=>mundo!,what=>NULL'"),
|
|
(None, "NULL")]);
|
|
}
|
|
|
|
fn test_nan_param<T: Float+ToSql+FromSql>(sql_type: &str) {
|
|
let conn = or_panic!(Connection::connect("postgres://postgres@localhost", &SslMode::None));
|
|
let stmt = or_panic!(conn.prepare(&*format!("SELECT 'NaN'::{}", sql_type)));
|
|
let mut result = or_panic!(stmt.query(&[]));
|
|
let val: T = result.next().unwrap().get(0us);
|
|
assert!(val.is_nan());
|
|
|
|
let nan: T = Float::nan();
|
|
let stmt = or_panic!(conn.prepare(&*format!("SELECT $1::{}", sql_type)));
|
|
let mut result = or_panic!(stmt.query(&[&nan]));
|
|
let val: T = result.next().unwrap().get(0us);
|
|
assert!(val.is_nan())
|
|
}
|
|
|
|
#[test]
|
|
fn test_f32_nan_param() {
|
|
test_nan_param::<f32>("REAL");
|
|
}
|
|
|
|
#[test]
|
|
fn test_f64_nan_param() {
|
|
test_nan_param::<f64>("DOUBLE PRECISION");
|
|
}
|
|
|
|
#[test]
|
|
fn test_pg_database_datname() {
|
|
let conn = or_panic!(Connection::connect("postgres://postgres@localhost", &SslMode::None));
|
|
let stmt = or_panic!(conn.prepare("SELECT datname FROM pg_database"));
|
|
let mut result = or_panic!(stmt.query(&[]));
|
|
|
|
let next = result.next().unwrap();
|
|
or_panic!(next.get_opt::<usize, String>(0));
|
|
or_panic!(next.get_opt::<&str, String>("datname"));
|
|
}
|