mirror of
https://github.com/launchbadge/sqlx.git
synced 2025-10-02 23:35:20 +00:00
87 lines
2.0 KiB
Rust
87 lines
2.0 KiB
Rust
use sqlx::Postgres;
|
|
use sqlx_test::test_type;
|
|
use std::fmt::Debug;
|
|
|
|
// Transparent types are rust-side wrappers over DB types
|
|
#[derive(PartialEq, Debug, sqlx::Type)]
|
|
#[sqlx(transparent)]
|
|
struct Transparent(i32);
|
|
|
|
// "Weak" enums map to an integer type indicated by #[repr]
|
|
#[derive(PartialEq, Copy, Clone, Debug, sqlx::Type)]
|
|
#[repr(i32)]
|
|
enum Weak {
|
|
One = 0,
|
|
Two = 2,
|
|
Three = 4,
|
|
}
|
|
|
|
// "Strong" enums can map to TEXT (25) or a custom enum type
|
|
#[derive(PartialEq, Debug, sqlx::Type)]
|
|
#[sqlx(postgres(oid = 25))]
|
|
#[sqlx(rename_all = "lowercase")]
|
|
enum Strong {
|
|
One,
|
|
Two,
|
|
|
|
#[sqlx(rename = "four")]
|
|
Three,
|
|
}
|
|
|
|
// TODO: Figure out a good solution for custom type testing
|
|
// Records must map to a custom type
|
|
// Note that all types are types in Postgres
|
|
// #[derive(PartialEq, Debug, sqlx::Type)]
|
|
// #[sqlx(postgres(oid = 12184))]
|
|
// struct PgConfig {
|
|
// name: String,
|
|
// setting: Option<String>,
|
|
// }
|
|
|
|
test_type!(transparent(
|
|
Postgres,
|
|
Transparent,
|
|
"0" == Transparent(0),
|
|
"23523" == Transparent(23523)
|
|
));
|
|
|
|
test_type!(weak_enum(
|
|
Postgres,
|
|
Weak,
|
|
"0::int4" == Weak::One,
|
|
"2::int4" == Weak::Two,
|
|
"4::int4" == Weak::Three
|
|
));
|
|
|
|
test_type!(strong_enum(
|
|
Postgres,
|
|
Strong,
|
|
"'one'::text" == Strong::One,
|
|
"'two'::text" == Strong::Two,
|
|
"'four'::text" == Strong::Three
|
|
));
|
|
|
|
// TODO: Figure out a good solution for custom type testing
|
|
// test_type!(record_pg_config(
|
|
// Postgres,
|
|
// PgConfig,
|
|
// // (CC,gcc)
|
|
// "(SELECT ROW('CC', 'gcc')::pg_config)"
|
|
// == PgConfig {
|
|
// name: "CC".to_owned(),
|
|
// setting: Some("gcc".to_owned()),
|
|
// },
|
|
// // (CC,)
|
|
// "(SELECT '(\"CC\",)'::pg_config)"
|
|
// == PgConfig {
|
|
// name: "CC".to_owned(),
|
|
// setting: None,
|
|
// },
|
|
// // (CC,"")
|
|
// "(SELECT '(\"CC\",\"\")'::pg_config)"
|
|
// == PgConfig {
|
|
// name: "CC".to_owned(),
|
|
// setting: Some("".to_owned()),
|
|
// }
|
|
// ));
|