use super::*;
use pretty_assertions::assert_eq;
#[test]
fn create_1() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(
ColumnDef::new(Glyph::Id)
.integer()
.not_null()
.auto_increment()
.primary_key()
)
.col(ColumnDef::new(Glyph::Aspect).double().not_null())
.col(ColumnDef::new(Glyph::Image).text())
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "glyph" ("#,
r#""id" integer GENERATED BY DEFAULT AS IDENTITY NOT NULL PRIMARY KEY,"#,
r#""aspect" double precision NOT NULL,"#,
r#""image" text"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_2() {
assert_eq!(
Table::create()
.table(Font::Table)
.col(
ColumnDef::new(Font::Id)
.integer()
.not_null()
.primary_key()
.auto_increment()
)
.col(ColumnDef::new(Font::Name).string().not_null())
.col(ColumnDef::new(Font::Variant).string_len(255).not_null())
.col(ColumnDef::new(Font::Language).string_len(255).not_null())
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "font" ("#,
r#""id" integer GENERATED BY DEFAULT AS IDENTITY NOT NULL PRIMARY KEY,"#,
r#""name" varchar NOT NULL,"#,
r#""variant" varchar(255) NOT NULL,"#,
r#""language" varchar(255) NOT NULL"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_3() {
assert_eq!(
Table::create()
.table(Char::Table)
.if_not_exists()
.col(
ColumnDef::new(Char::Id)
.integer()
.not_null()
.primary_key()
.auto_increment()
)
.col(ColumnDef::new(Char::FontSize).integer().not_null())
.col(ColumnDef::new(Char::Character).string_len(255).not_null())
.col(ColumnDef::new(Char::SizeW).unsigned().not_null())
.col(ColumnDef::new(Char::SizeH).unsigned().not_null())
.col(
ColumnDef::new(Char::FontId)
.integer()
.default(Value::Int(None))
)
.foreign_key(
ForeignKey::create()
.name("FK_2e303c3a712662f1fc2a4d0aad6")
.from(Char::Table, Char::FontId)
.to(Font::Table, Font::Id)
.on_delete(ForeignKeyAction::Cascade)
.on_update(ForeignKeyAction::Cascade)
)
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE IF NOT EXISTS "character" ("#,
r#""id" integer GENERATED BY DEFAULT AS IDENTITY NOT NULL PRIMARY KEY,"#,
r#""font_size" integer NOT NULL,"#,
r#""character" varchar(255) NOT NULL,"#,
r#""size_w" bigint NOT NULL,"#,
r#""size_h" bigint NOT NULL,"#,
r#""font_id" integer DEFAULT NULL,"#,
r#"CONSTRAINT "FK_2e303c3a712662f1fc2a4d0aad6""#,
r#"FOREIGN KEY ("font_id") REFERENCES "font" ("id")"#,
r#"ON DELETE CASCADE ON UPDATE CASCADE"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_4() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(ColumnDef::new(Glyph::Image).custom(Glyph::Aspect))
.to_string(PostgresQueryBuilder),
r#"CREATE TABLE "glyph" ( "image" aspect )"#
);
}
#[test]
fn create_5() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(ColumnDef::new(Glyph::Image).json())
.col(ColumnDef::new(Glyph::Aspect).json_binary())
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "glyph" ("#,
r#""image" json,"#,
r#""aspect" jsonb"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_6() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(
ColumnDef::new(Glyph::Id)
.integer()
.not_null()
.extra("ANYTHING I WANT TO SAY".to_owned())
)
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "glyph" ("#,
r#""id" integer NOT NULL ANYTHING I WANT TO SAY"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_7() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(
ColumnDef::new(Glyph::Aspect)
.interval(None, None)
.not_null()
)
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "glyph" ("#,
r#""aspect" interval NOT NULL"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_8() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(
ColumnDef::new(Glyph::Aspect)
.interval(Some(PgInterval::YearToMonth), None)
.not_null()
)
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "glyph" ("#,
r#""aspect" interval YEAR TO MONTH NOT NULL"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_9() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(
ColumnDef::new(Glyph::Aspect)
.interval(None, Some(42))
.not_null()
)
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "glyph" ("#,
r#""aspect" interval(42) NOT NULL"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_10() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(
ColumnDef::new(Glyph::Aspect)
.interval(Some(PgInterval::Hour), Some(43))
.not_null()
)
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "glyph" ("#,
r#""aspect" interval HOUR(43) NOT NULL"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_11() {
assert_eq!(
Table::create()
.table(Char::Table)
.col(
ColumnDef::new(Char::CreatedAt)
.timestamp_with_time_zone()
.not_null()
)
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "character" ("#,
r#""created_at" timestamp with time zone NOT NULL"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_12() {
assert_eq!(
Table::create()
.table(BinaryType::Table)
.col(ColumnDef::new(BinaryType::BinaryLen).binary_len(32))
.col(ColumnDef::new(BinaryType::Binary).binary())
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "binary_type" ("#,
r#""binlen" bytea,"#,
r#""bin" bytea"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_13() {
assert_eq!(
Table::create()
.table(Char::Table)
.col(ColumnDef::new(Char::Character).binary())
.col(ColumnDef::new(Char::FontSize).binary_len(10))
.col(ColumnDef::new(Char::SizeW).var_binary(10))
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "character" ("#,
r#""character" bytea,"#,
r#""font_size" bytea,"#,
r#""size_w" bytea"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_14() {
assert_eq!(
Table::create()
.table(("schema", Glyph::Table))
.col(ColumnDef::new(Glyph::Image).custom(Glyph::Aspect))
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "schema"."glyph" ("#,
r#""image" aspect"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_15() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(ColumnDef::new(Glyph::Image).json())
.col(ColumnDef::new(Glyph::Aspect).json_binary())
.index(
Index::create()
.unique()
.nulls_not_distinct()
.name("idx-glyph-aspect-image")
.table(Glyph::Table)
.col(Glyph::Aspect)
.col(Glyph::Image)
)
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "glyph" ("#,
r#""image" json,"#,
r#""aspect" jsonb,"#,
r#"CONSTRAINT "idx-glyph-aspect-image" UNIQUE NULLS NOT DISTINCT ("aspect", "image")"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn drop_1() {
assert_eq!(
Table::drop()
.table(Glyph::Table)
.table(Char::Table)
.cascade()
.to_string(PostgresQueryBuilder),
r#"DROP TABLE "glyph", "character" CASCADE"#
);
}
#[test]
fn drop_2() {
assert_eq!(
Table::drop()
.table(("schema1", Glyph::Table))
.table(("schema2", Char::Table))
.cascade()
.to_string(PostgresQueryBuilder),
r#"DROP TABLE "schema1"."glyph", "schema2"."character" CASCADE"#
);
}
#[test]
fn truncate_1() {
assert_eq!(
Table::truncate()
.table(Font::Table)
.to_string(PostgresQueryBuilder),
r#"TRUNCATE TABLE "font""#
);
}
#[test]
fn truncate_2() {
assert_eq!(
Table::truncate()
.table(("schema", Font::Table))
.to_string(PostgresQueryBuilder),
r#"TRUNCATE TABLE "schema"."font""#
);
}
#[test]
fn alter_1() {
assert_eq!(
Table::alter()
.table(Font::Table)
.add_column(ColumnDef::new("new_col").integer().not_null().default(100))
.to_string(PostgresQueryBuilder),
r#"ALTER TABLE "font" ADD COLUMN "new_col" integer NOT NULL DEFAULT 100"#
);
}
#[test]
fn alter_2() {
assert_eq!(
Table::alter()
.table(Font::Table)
.modify_column(ColumnDef::new("new_col").big_integer().default(999))
.to_string(PostgresQueryBuilder),
[
r#"ALTER TABLE "font""#,
r#"ALTER COLUMN "new_col" TYPE bigint,"#,
r#"ALTER COLUMN "new_col" SET DEFAULT 999"#,
]
.join(" ")
);
}
#[test]
fn alter_3() {
assert_eq!(
Table::alter()
.table(Font::Table)
.rename_column("new_col", "new_column")
.to_string(PostgresQueryBuilder),
r#"ALTER TABLE "font" RENAME COLUMN "new_col" TO "new_column""#
);
}
#[test]
fn alter_4() {
assert_eq!(
Table::alter()
.table(Font::Table)
.drop_column("new_column")
.to_string(PostgresQueryBuilder),
r#"ALTER TABLE "font" DROP COLUMN "new_column""#
);
}
#[test]
fn alter_5() {
assert_eq!(
Table::alter()
.table(("schema", Font::Table))
.rename_column("new_col", "new_column")
.to_string(PostgresQueryBuilder),
r#"ALTER TABLE "schema"."font" RENAME COLUMN "new_col" TO "new_column""#
);
}
#[test]
#[should_panic(expected = "No alter option found")]
fn alter_6() {
Table::alter().to_string(PostgresQueryBuilder);
}
#[test]
fn alter_7() {
assert_eq!(
Table::alter()
.table(Font::Table)
.add_column(ColumnDef::new("new_col").integer())
.rename_column(Font::Name, "name_new")
.to_string(PostgresQueryBuilder),
r#"ALTER TABLE "font" ADD COLUMN "new_col" integer, RENAME COLUMN "name" TO "name_new""#
);
}
#[test]
fn alter_8() {
assert_eq!(
Table::alter()
.table(Font::Table)
.modify_column(ColumnDef::new(Font::Language).null())
.to_string(PostgresQueryBuilder),
[
r#"ALTER TABLE "font""#,
r#"ALTER COLUMN "language" DROP NOT NULL"#,
]
.join(" ")
);
}
#[test]
fn alter_9() {
assert_eq!(
Table::alter()
.table(Glyph::Table)
.modify_column(
ColumnDef::new(Glyph::Aspect)
.integer()
.auto_increment()
.not_null()
.unique_key()
.primary_key()
)
.to_string(PostgresQueryBuilder),
[
r#"ALTER TABLE "glyph""#,
r#"ALTER COLUMN "aspect" TYPE integer,"#,
r#"ALTER COLUMN "aspect" SET NOT NULL,"#,
r#"ADD UNIQUE ("aspect"),"#,
r#"ADD PRIMARY KEY ("aspect")"#,
]
.join(" ")
);
}
#[test]
fn alter_10() {
assert_eq!(
Table::alter()
.table(Glyph::Table)
.add_column(
ColumnDef::new(Glyph::Aspect)
.integer()
.auto_increment()
.not_null()
.unique_key()
.primary_key()
)
.to_string(PostgresQueryBuilder),
[
r#"ALTER TABLE "glyph""#,
r#"ADD COLUMN "aspect" integer GENERATED BY DEFAULT AS IDENTITY NOT NULL PRIMARY KEY UNIQUE"#,
]
.join(" ")
);
}
#[test]
fn rename_1() {
assert_eq!(
Table::rename()
.table(Font::Table, "font_new")
.to_string(PostgresQueryBuilder),
r#"ALTER TABLE "font" RENAME TO "font_new""#
);
}
#[test]
fn rename_2() {
assert_eq!(
Table::rename()
.table(("schema", Font::Table), ("schema", "font_new"),)
.to_string(PostgresQueryBuilder),
r#"ALTER TABLE "schema"."font" RENAME TO "schema"."font_new""#
);
}
#[test]
fn create_with_check_constraint() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(
ColumnDef::new(Glyph::Id)
.integer()
.not_null()
.check(Expr::col(Glyph::Id).gt(10))
)
.check(Expr::col(Glyph::Id).lt(20))
.check(Expr::col(Glyph::Id).ne(15))
.to_string(PostgresQueryBuilder),
r#"CREATE TABLE "glyph" ( "id" integer NOT NULL CHECK ("id" > 10), CHECK ("id" < 20), CHECK ("id" <> 15) )"#,
);
}
#[test]
fn create_with_named_check_constraint() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(
ColumnDef::new(Glyph::Id)
.integer()
.not_null()
.check(("positive_id", Expr::col(Glyph::Id).gt(10)))
)
.check(("id_range", Expr::col(Glyph::Id).lt(20)))
.check(Expr::col(Glyph::Id).ne(15))
.to_string(PostgresQueryBuilder),
r#"CREATE TABLE "glyph" ( "id" integer NOT NULL CONSTRAINT "positive_id" CHECK ("id" > 10), CONSTRAINT "id_range" CHECK ("id" < 20), CHECK ("id" <> 15) )"#,
);
}
#[test]
fn alter_with_check_constraint() {
assert_eq!(
Table::alter()
.table(Glyph::Table)
.add_column(
ColumnDef::new(Glyph::Aspect)
.integer()
.not_null()
.default(101)
.check(Expr::col(Glyph::Aspect).gt(100))
)
.to_string(PostgresQueryBuilder),
r#"ALTER TABLE "glyph" ADD COLUMN "aspect" integer NOT NULL DEFAULT 101 CHECK ("aspect" > 100)"#,
);
}
#[test]
fn alter_with_named_check_constraint() {
assert_eq!(
Table::alter()
.table(Glyph::Table)
.add_column(
ColumnDef::new(Glyph::Aspect)
.integer()
.not_null()
.default(101)
.check(("positive_aspect", Expr::col(Glyph::Aspect).gt(100)))
)
.to_string(PostgresQueryBuilder),
r#"ALTER TABLE "glyph" ADD COLUMN "aspect" integer NOT NULL DEFAULT 101 CONSTRAINT "positive_aspect" CHECK ("aspect" > 100)"#,
);
}
#[test]
fn create_16() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(
ColumnDef::new(Glyph::Id)
.integer()
.not_null()
.auto_increment()
.primary_key()
)
.col(ColumnDef::new(Glyph::Tokens).ltree())
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "glyph" ("#,
r#""id" integer GENERATED BY DEFAULT AS IDENTITY NOT NULL PRIMARY KEY,"#,
r#""tokens" ltree"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_17() {
assert_eq!(
Table::create()
.table(Font::Table)
.col(
ColumnDef::new(Font::Id)
.integer()
.not_null()
.primary_key()
.auto_increment(),
)
.col(ColumnDef::new(Font::Name).string())
.col(ColumnDef::new(Font::Variant).string_len(255).not_null())
.col(ColumnDef::new(Font::Language).string_len(255).not_null())
.index(
Index::create()
.name("idx-font-name-include-language")
.unique()
.nulls_not_distinct()
.col(Font::Name)
.include(Font::Language),
)
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE "font" ("#,
r#""id" integer GENERATED BY DEFAULT AS IDENTITY NOT NULL PRIMARY KEY,"#,
r#""name" varchar,"#,
r#""variant" varchar(255) NOT NULL,"#,
r#""language" varchar(255) NOT NULL,"#,
r#"CONSTRAINT "idx-font-name-include-language" UNIQUE NULLS NOT DISTINCT ("name") INCLUDE ("language")"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_18() {
assert_eq!(
Table::create()
.table(Font::Table)
.temporary()
.col(
ColumnDef::new(Font::Id)
.integer()
.not_null()
.primary_key()
.auto_increment()
)
.col(ColumnDef::new(Font::Name).string().not_null())
.to_string(PostgresQueryBuilder),
[
r#"CREATE TEMPORARY TABLE "font" ("#,
r#""id" integer GENERATED BY DEFAULT AS IDENTITY NOT NULL PRIMARY KEY,"#,
r#""name" varchar NOT NULL"#,
r#")"#,
]
.join(" ")
);
}
#[test]
fn create_19() {
assert_eq!(
Table::create()
.table(Char::Table)
.if_not_exists()
.col(
ColumnDef::new(Char::Id)
.integer()
.not_null()
.primary_key()
.auto_increment(),
)
.col(ColumnDef::new(Char::FontSize).integer().not_null())
.col(ColumnDef::new(Char::Character).string_len(255).not_null())
.col(ColumnDef::new(Char::SizeW).unsigned().not_null())
.col(ColumnDef::new(Char::SizeH).unsigned().not_null())
.col(
ColumnDef::new(Char::FontId)
.integer()
.default(Value::Int(None)),
)
.index(
Index::create()
.name("idx-character-area")
.table(Character::Table)
.col(Expr::col(Character::SizeH).mul(Expr::col(Character::SizeW))),
)
.to_string(PostgresQueryBuilder),
[
r#"CREATE TABLE IF NOT EXISTS "character" ("#,
r#""id" integer GENERATED BY DEFAULT AS IDENTITY NOT NULL PRIMARY KEY,"#,
r#""font_size" integer NOT NULL,"#,
r#""character" varchar(255) NOT NULL,"#,
r#""size_w" bigint NOT NULL,"#,
r#""size_h" bigint NOT NULL,"#,
r#""font_id" integer DEFAULT NULL,"#,
r#"CONSTRAINT "idx-character-area" (("size_h" * "size_w"))"#,
r#")"#,
]
.join(" "),
);
}
#[test]
fn create_partition_master_range() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(ColumnDef::new(Glyph::Id).integer().not_null())
.col(ColumnDef::new(Glyph::Aspect).integer().not_null())
.partition_by_range([Glyph::Aspect])
.to_string(PostgresQueryBuilder),
r#"CREATE TABLE "glyph" ( "id" integer NOT NULL, "aspect" integer NOT NULL ) PARTITION BY RANGE ("aspect")"#
);
}
#[test]
fn create_partition_master_range_multiple_columns() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(ColumnDef::new(Glyph::Id).integer().not_null())
.col(ColumnDef::new(Glyph::Aspect).integer().not_null())
.partition_by_range([Glyph::Id, Glyph::Aspect])
.to_string(PostgresQueryBuilder),
r#"CREATE TABLE "glyph" ( "id" integer NOT NULL, "aspect" integer NOT NULL ) PARTITION BY RANGE ("id", "aspect")"#
);
}
#[test]
fn create_partition_child_range() {
assert_eq!(
Table::create()
.table(Alias::new("glyph_1"))
.partition_of(Glyph::Table)
.values_from_to([1], [10])
.to_string(PostgresQueryBuilder),
r#"CREATE TABLE "glyph_1" PARTITION OF "glyph" FOR VALUES FROM (1) TO (10)"#
);
}
#[test]
fn create_partition_child_range_multiple_columns() {
assert_eq!(
Table::create()
.table(Alias::new("glyph_1"))
.partition_of(Glyph::Table)
.values_from_to([1, 100], [10, 200])
.to_string(PostgresQueryBuilder),
r#"CREATE TABLE "glyph_1" PARTITION OF "glyph" FOR VALUES FROM (1, 100) TO (10, 200)"#
);
}
#[test]
fn create_partition_master_list() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(ColumnDef::new(Glyph::Id).integer().not_null())
.partition_by_list([Glyph::Id])
.to_string(PostgresQueryBuilder),
r#"CREATE TABLE "glyph" ( "id" integer NOT NULL ) PARTITION BY LIST ("id")"#
);
}
#[test]
fn create_partition_child_list() {
assert_eq!(
Table::create()
.table(Alias::new("glyph_p1"))
.partition_of(Glyph::Table)
.values_in([1, 2, 3])
.to_string(PostgresQueryBuilder),
r#"CREATE TABLE "glyph_p1" PARTITION OF "glyph" FOR VALUES IN (1, 2, 3)"#
);
}
#[test]
fn create_partition_master_hash() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(ColumnDef::new(Glyph::Id).integer().not_null())
.partition_by_hash([Glyph::Id])
.to_string(PostgresQueryBuilder),
r#"CREATE TABLE "glyph" ( "id" integer NOT NULL ) PARTITION BY HASH ("id")"#
);
}
#[test]
fn create_partition_master_hash_multiple_columns() {
assert_eq!(
Table::create()
.table(Glyph::Table)
.col(ColumnDef::new(Glyph::Id).integer().not_null())
.col(ColumnDef::new(Glyph::Aspect).integer().not_null())
.partition_by_hash([Glyph::Id, Glyph::Aspect])
.to_string(PostgresQueryBuilder),
r#"CREATE TABLE "glyph" ( "id" integer NOT NULL, "aspect" integer NOT NULL ) PARTITION BY HASH ("id", "aspect")"#
);
}
#[test]
fn create_partition_child_hash() {
assert_eq!(
Table::create()
.table(Alias::new("glyph_p1"))
.partition_of(Glyph::Table)
.values_with(4, 0)
.to_string(PostgresQueryBuilder),
r#"CREATE TABLE "glyph_p1" PARTITION OF "glyph" FOR VALUES WITH (MODULUS 4, REMAINDER 0)"#
);
}
#[test]
#[should_panic(expected = "Postgres does not support PARTITION BY KEY")]
fn create_partition_key_panics() {
Table::create()
.table(Glyph::Table)
.col(ColumnDef::new(Glyph::Id).integer().not_null())
.partition_by_key([Glyph::Id])
.to_string(PostgresQueryBuilder);
}
#[test]
#[should_panic(expected = "Postgres does not support VALUES LESS THAN")]
fn create_partition_values_less_than_panics() {
Table::create()
.table(Alias::new("glyph_p1"))
.partition_of(Glyph::Table)
.values_less_than([10])
.to_string(PostgresQueryBuilder);
}