sqlx/tests/postgres-derives.rs

use sqlx_test::test_type;
use std::fmt::Debug;
use sqlx::Postgres;

// 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,
}

// 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
));

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()),
    }
));