Vendor dependencies

This commit is contained in:
2026-08-01 16:11:49 +03:00
parent 7f139a0241
commit 6b5e7f0f8b
29706 changed files with 9575646 additions and 0 deletions
@@ -0,0 +1,77 @@
use super::*;
use pretty_assertions::assert_eq;
#[test]
fn explain_postgres_select_with_options() {
assert_eq!(
ExplainStatement::new()
.analyze()
.verbose(false)
.costs(true)
.settings(false)
.generic_plan(true)
.buffers(true)
.serialize_text()
.wal(true)
.timing(false)
.summary(true)
.memory(true)
.format(ExplainFormat::Json)
.statement(
Query::select()
.column(Char::Character)
.from(Char::Table)
.to_owned(),
)
.to_string(PostgresQueryBuilder),
r#"EXPLAIN (ANALYZE, VERBOSE 0, COSTS, SETTINGS 0, GENERIC_PLAN, BUFFERS, SERIALIZE TEXT, WAL, TIMING 0, SUMMARY, MEMORY, FORMAT JSON) SELECT "character" FROM "character""#
);
}
#[test]
fn explain_postgres_serialize_text() {
assert_eq!(
ExplainStatement::new()
.serialize_text()
.statement(
Query::select()
.column(Char::Character)
.from(Char::Table)
.to_owned(),
)
.to_string(PostgresQueryBuilder),
r#"EXPLAIN (SERIALIZE TEXT) SELECT "character" FROM "character""#
);
}
#[test]
fn explain_postgres_serialize_binary() {
assert_eq!(
ExplainStatement::new()
.serialize_binary()
.statement(
Query::select()
.column(Char::Character)
.from(Char::Table)
.to_owned(),
)
.to_string(PostgresQueryBuilder),
r#"EXPLAIN (SERIALIZE BINARY) SELECT "character" FROM "character""#
);
}
#[test]
fn explain_postgres_serialize_none() {
assert_eq!(
ExplainStatement::new()
.serialize_none()
.statement(
Query::select()
.column(Char::Character)
.from(Char::Table)
.to_owned(),
)
.to_string(PostgresQueryBuilder),
r#"EXPLAIN (SERIALIZE NONE) SELECT "character" FROM "character""#
);
}
@@ -0,0 +1,62 @@
use super::*;
use pretty_assertions::assert_eq;
#[test]
fn create_1() {
assert_eq!(
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#"ALTER TABLE "character" ADD CONSTRAINT "FK_2e303c3a712662f1fc2a4d0aad6""#,
r#"FOREIGN KEY ("font_id") REFERENCES "font" ("id")"#,
r#"ON DELETE CASCADE ON UPDATE CASCADE"#,
]
.join(" ")
);
}
#[test]
fn create_2() {
assert_eq!(
ForeignKey::create()
.name("FK_2e303c3a712662f1fc2a4d0aad6")
.from(("schema", Char::Table), Char::FontId)
.to(Font::Table, Font::Id)
.on_delete(ForeignKeyAction::Cascade)
.on_update(ForeignKeyAction::Cascade)
.to_string(PostgresQueryBuilder),
[
r#"ALTER TABLE "schema"."character" ADD CONSTRAINT "FK_2e303c3a712662f1fc2a4d0aad6""#,
r#"FOREIGN KEY ("font_id") REFERENCES "font" ("id")"#,
r#"ON DELETE CASCADE ON UPDATE CASCADE"#,
]
.join(" ")
);
}
#[test]
fn drop_1() {
assert_eq!(
ForeignKey::drop()
.name("FK_2e303c3a712662f1fc2a4d0aad6")
.table(Char::Table)
.to_string(PostgresQueryBuilder),
r#"ALTER TABLE "character" DROP CONSTRAINT "FK_2e303c3a712662f1fc2a4d0aad6""#
);
}
#[test]
fn drop_2() {
assert_eq!(
ForeignKey::drop()
.name("FK_2e303c3a712662f1fc2a4d0aad6")
.table(("schema", Char::Table))
.to_string(PostgresQueryBuilder),
r#"ALTER TABLE "schema"."character" DROP CONSTRAINT "FK_2e303c3a712662f1fc2a4d0aad6""#
);
}
+227
View File
@@ -0,0 +1,227 @@
use super::*;
use pretty_assertions::assert_eq;
#[test]
fn create_1() {
assert_eq!(
Index::create()
.name("idx-glyph-aspect")
.table(Glyph::Table)
.col(Glyph::Aspect)
.to_string(PostgresQueryBuilder),
r#"CREATE INDEX "idx-glyph-aspect" ON "glyph" ("aspect")"#
);
}
#[test]
fn create_2() {
assert_eq!(
Index::create()
.unique()
.name("idx-glyph-aspect-image")
.table(Glyph::Table)
.col(Glyph::Aspect)
.col(Glyph::Image)
.to_string(PostgresQueryBuilder),
r#"CREATE UNIQUE INDEX "idx-glyph-aspect-image" ON "glyph" ("aspect", "image")"#
);
}
#[test]
fn create_3() {
assert_eq!(
Index::create()
.full_text()
.name("idx-glyph-image")
.table(Glyph::Table)
.col(Glyph::Image)
.to_string(PostgresQueryBuilder),
r#"CREATE INDEX "idx-glyph-image" ON "glyph" USING GIN ("image")"#
);
}
#[test]
fn create_4() {
assert_eq!(
Index::create()
.full_text()
.name("idx-glyph-image")
.concurrently()
.table(Glyph::Table)
.col(Glyph::Image)
.to_string(PostgresQueryBuilder),
r#"CREATE INDEX CONCURRENTLY "idx-glyph-image" ON "glyph" USING GIN ("image")"#
);
}
#[test]
fn create_5() {
assert_eq!(
Index::create()
.if_not_exists()
.full_text()
.name("idx-glyph-image")
.table(Glyph::Table)
.col(Glyph::Image)
.to_string(PostgresQueryBuilder),
r#"CREATE INDEX IF NOT EXISTS "idx-glyph-image" ON "glyph" USING GIN ("image")"#
);
}
#[test]
fn create_6() {
assert_eq!(
Index::create()
.unique()
.name("idx-glyph-aspect-image")
.table(("schema", Glyph::Table))
.col(Glyph::Aspect)
.col(Glyph::Image)
.to_string(PostgresQueryBuilder),
r#"CREATE UNIQUE INDEX "idx-glyph-aspect-image" ON "schema"."glyph" ("aspect", "image")"#
);
}
#[test]
fn create_7() {
assert_eq!(
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 UNIQUE INDEX "idx-glyph-aspect-image" ON "glyph" ("aspect", "image") NULLS NOT DISTINCT"#
);
}
#[test]
fn create_8() {
assert_eq!(
Index::create()
.unique()
.nulls_not_distinct()
.name("partial-index-glyph-image-not-null")
.table(Glyph::Table)
.col(Glyph::Image)
.and_where(Expr::col(Glyph::Image).is_not_null())
.to_string(PostgresQueryBuilder),
r#"CREATE UNIQUE INDEX "partial-index-glyph-image-not-null" ON "glyph" ("image") NULLS NOT DISTINCT WHERE "image" IS NOT NULL"#
);
}
#[test]
fn create_9() {
assert_eq!(
Index::create()
.name("idx-font-name-include-id-language")
.table(Font::Table)
.col(Font::Name)
.include(Font::Id)
.include(Font::Language)
.unique()
.nulls_not_distinct()
.to_string(PostgresQueryBuilder),
r#"CREATE UNIQUE INDEX "idx-font-name-include-id-language" ON "font" ("name") INCLUDE ("id", "language") NULLS NOT DISTINCT"#
);
}
#[test]
fn create_10() {
assert_eq!(
Index::create()
.name("idx-character-area")
.table(Character::Table)
.col(Expr::col(Character::SizeH).mul(Expr::col(Character::SizeW)))
.to_string(PostgresQueryBuilder),
r#"CREATE INDEX "idx-character-area" ON "character" (("size_h" * "size_w"))"#
)
}
#[test]
fn create_11() {
assert_eq!(
Index::create()
.name("idx-character-character-area-desc-created_at")
.table(Character::Table)
.col(Func::upper(Expr::col(Character::Character)))
.col((
Expr::col(Character::SizeH).mul(Expr::col(Character::SizeW)),
IndexOrder::Desc,
))
.col(Character::CreatedAt)
.to_string(PostgresQueryBuilder),
r#"CREATE INDEX "idx-character-character-area-desc-created_at" ON "character" ((UPPER("character")), ("size_h" * "size_w") DESC, "created_at")"#
)
}
#[test]
fn create_12() {
assert_eq!(
Index::create()
.if_not_exists()
.name("idx-glyph-image")
.table(Glyph::Table)
.col(
Glyph::Image
.into_index_column()
.with_operator_class("text_pattern_ops")
)
.to_string(PostgresQueryBuilder),
r#"CREATE INDEX IF NOT EXISTS "idx-glyph-image" ON "glyph" ("image" text_pattern_ops)"#
);
}
#[test]
fn drop_1() {
assert_eq!(
Index::drop()
.name("idx-glyph-aspect")
.to_string(PostgresQueryBuilder),
r#"DROP INDEX "idx-glyph-aspect""#
);
}
#[test]
fn drop_2() {
assert_eq!(
Index::drop()
.name("idx-glyph-aspect")
.concurrently()
.to_string(PostgresQueryBuilder),
r#"DROP INDEX CONCURRENTLY "idx-glyph-aspect""#
);
}
#[test]
fn drop_3() {
assert_eq!(
Index::drop()
.name("idx-glyph-aspect")
.table(("schema", Glyph::Table))
.to_string(PostgresQueryBuilder),
r#"DROP INDEX "schema"."idx-glyph-aspect""#
);
}
#[test]
fn drop_4() {
assert_eq!(
Index::drop()
.name("idx-glyph-aspect")
.table(Glyph::Table)
.to_string(PostgresQueryBuilder),
r#"DROP INDEX "idx-glyph-aspect""#
);
}
#[test]
#[should_panic(expected = "Not supported")]
fn drop_5() {
Index::drop()
.name("idx-glyph-aspect")
.table(("database", "schema", Glyph::Table))
.to_string(PostgresQueryBuilder);
}
@@ -0,0 +1,126 @@
use super::*;
use pretty_assertions::assert_eq;
use sea_query::extension::postgres::{PgFunc, json_table as pg_json_table};
#[test]
fn json_table_basic_columns() {
let query = Query::select()
.from_function(
PgFunc::json_table(
Expr::val(r#"[{"name": "John", "age": 30}, {"name": "Jane", "age": 25}]"#),
"$[*]",
)
.for_ordinality("row_number")
.column(pg_json_table::Column::new("name", "text").path("$.name"))
.column(pg_json_table::Column::new("age", "int").path("$.age"))
.build(),
"people",
)
.column("*")
.to_owned();
assert_eq!(
query.to_string(PostgresQueryBuilder),
r#"SELECT "*" FROM JSON_TABLE('[{"name": "John", "age": 30}, {"name": "Jane", "age": 25}]', '$[*]' COLUMNS (row_number FOR ORDINALITY, name "text" PATH '$.name', age "int" PATH '$.age')) AS "people""#
);
}
#[test]
fn json_table_passing_and_error_handling() {
let query = Query::select()
.from_function(
PgFunc::json_table(
Expr::val(r#"{"users": [{"id": 1, "name": "John"}, {"id": 2, "name": "Jane"}]}"#),
"$.users[*] ? (@.id > $min_id)",
)
.passing(0, "min_id")
.column(
pg_json_table::Column::new("user_id", "int")
.path("$.id")
.null_on_error(),
)
.column(
pg_json_table::Column::new("user_name", "text")
.path("$.name")
.default_on_empty(Expr::val("Unknown")),
)
.error_on_error()
.build(),
"filtered_users",
)
.column("*")
.to_owned();
assert_eq!(
query.to_string(PostgresQueryBuilder),
r#"SELECT "*" FROM JSON_TABLE('{"users": [{"id": 1, "name": "John"}, {"id": 2, "name": "Jane"}]}', '$.users[*] ? (@.id > $min_id)' PASSING 0 AS min_id COLUMNS (user_id "int" PATH '$.id' NULL ON ERROR, user_name "text" PATH '$.name' DEFAULT 'Unknown' ON EMPTY) ERROR ON ERROR) AS "filtered_users""#
);
}
#[test]
fn json_table_nested_path() {
let query = Query::select()
.from_function(
PgFunc::json_table(
Expr::val(
r#"{"users": [{"name": "John", "phones": ["123", "456"]}, {"name": "Jane", "phones": ["789"]}]}"#,
),
"$.users[*]",
)
.column(pg_json_table::Column::new("user_name", "text").path("$.name"))
.nested(
pg_json_table::NestedPath::new("$.phones[*]")
.column(pg_json_table::Column::new("phone", "text").path("$")),
)
.build(),
"user_phones",
)
.column("*")
.to_owned();
assert_eq!(
query.to_string(PostgresQueryBuilder),
r#"SELECT "*" FROM JSON_TABLE('{"users": [{"name": "John", "phones": ["123", "456"]}, {"name": "Jane", "phones": ["789"]}]}', '$.users[*]' COLUMNS (user_name "text" PATH '$.name', NESTED PATH '$.phones[*]' COLUMNS (phone "text" PATH '$'))) AS "user_phones""#
);
}
#[test]
fn json_table_nested_path_recursion_and_siblings() {
let query = Query::select()
.from_function(
PgFunc::json_table(
Expr::cust(
r#"'{"favorites":[{"movies":[{"name":"One","director":"John Doe"}],"books":[{"name":"Mystery","authors":[{"name":"Brown Dan"}]}]}]}'::json"#,
),
"$.favorites[*]",
)
.for_ordinality("user_id")
.nested(
pg_json_table::NestedPath::new("$.movies[*]")
.for_ordinality("movie_id")
.column(pg_json_table::Column::new("mname", "text").path("$.name"))
.column(pg_json_table::Column::new("director", "text")),
)
.nested(
pg_json_table::NestedPath::new("$.books[*]")
.for_ordinality("book_id")
.column(pg_json_table::Column::new("bname", "text").path("$.name"))
.nested(
pg_json_table::NestedPath::new("$.authors[*]")
.for_ordinality("author_id")
.column(
pg_json_table::Column::new("author_name", "text").path("$.name"),
),
),
)
.build(),
"jt",
)
.column("*")
.to_owned();
assert_eq!(
query.to_string(PostgresQueryBuilder),
r#"SELECT "*" FROM JSON_TABLE('{"favorites":[{"movies":[{"name":"One","director":"John Doe"}],"books":[{"name":"Mystery","authors":[{"name":"Brown Dan"}]}]}]}'::json, '$.favorites[*]' COLUMNS (user_id FOR ORDINALITY, NESTED PATH '$.movies[*]' COLUMNS (movie_id FOR ORDINALITY, mname "text" PATH '$.name', director "text"), NESTED PATH '$.books[*]' COLUMNS (book_id FOR ORDINALITY, bname "text" PATH '$.name', NESTED PATH '$.authors[*]' COLUMNS (author_id FOR ORDINALITY, author_name "text" PATH '$.name')))) AS "jt""#
);
}
+13
View File
@@ -0,0 +1,13 @@
use sea_query::{tests_cfg::*, *};
mod explain;
mod foreign_key;
mod index;
mod json_table;
mod query;
mod table;
mod types;
#[path = "../common.rs"]
mod common;
use common::*;
File diff suppressed because it is too large Load Diff
+873
View File
@@ -0,0 +1,873 @@
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() {
// https://dbfiddle.uk/98Vd8pmn
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() {
// https://dbfiddle.uk/BeiZPvBe
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);
}
+179
View File
@@ -0,0 +1,179 @@
use super::*;
use pretty_assertions::assert_eq;
use sea_query::{PostgresQueryBuilder, extension::postgres::Type};
#[test]
fn create_1() {
assert_eq!(
Type::create()
.as_enum(Font::Table)
.values([Font::Name, Font::Variant, Font::Language])
.to_string(PostgresQueryBuilder),
r#"CREATE TYPE "font" AS ENUM ('name', 'variant', 'language')"#
);
}
#[test]
fn create_2() {
assert_eq!(
Type::create()
.as_enum(("schema", Font::Table))
.values([Font::Name, Font::Variant, Font::Language])
.to_string(PostgresQueryBuilder),
r#"CREATE TYPE "schema"."font" AS ENUM ('name', 'variant', 'language')"#
);
}
#[test]
fn create_3() {
assert_eq!(
Type::create()
.as_enum(Tea::Enum)
.values([Tea::EverydayTea, Tea::BreakfastTea])
.to_string(PostgresQueryBuilder),
r#"CREATE TYPE "tea" AS ENUM ('EverydayTea', 'BreakfastTea')"#
);
enum Tea {
Enum,
EverydayTea,
BreakfastTea,
}
impl sea_query::Iden for Tea {
fn unquoted(&self) -> &str {
match self {
Self::Enum => "tea",
Self::EverydayTea => "EverydayTea",
Self::BreakfastTea => "BreakfastTea",
}
}
}
}
#[test]
fn drop_1() {
assert_eq!(
Type::drop()
.if_exists()
.name(Font::Table)
.restrict()
.to_string(PostgresQueryBuilder),
r#"DROP TYPE IF EXISTS "font" RESTRICT"#
)
}
#[test]
fn drop_2() {
assert_eq!(
Type::drop()
.name(Font::Table)
.to_string(PostgresQueryBuilder),
r#"DROP TYPE "font""#
);
}
#[test]
fn drop_3() {
assert_eq!(
Type::drop()
.if_exists()
.name(Font::Table)
.cascade()
.to_string(PostgresQueryBuilder),
r#"DROP TYPE IF EXISTS "font" CASCADE"#
);
}
#[test]
fn drop_4() {
assert_eq!(
Type::drop()
.name(("schema", Font::Table))
.to_string(PostgresQueryBuilder),
r#"DROP TYPE "schema"."font""#
);
}
#[test]
fn alter_1() {
assert_eq!(
Type::alter()
.name(Font::Table)
.add_value("weight")
.to_string(PostgresQueryBuilder),
r#"ALTER TYPE "font" ADD VALUE 'weight'"#
)
}
#[test]
fn alter_2() {
assert_eq!(
Type::alter()
.name(Font::Table)
.add_value("weight")
.before(Font::Variant)
.to_string(PostgresQueryBuilder),
r#"ALTER TYPE "font" ADD VALUE 'weight' BEFORE 'variant'"#
)
}
#[test]
fn alter_3() {
assert_eq!(
Type::alter()
.name(Font::Table)
.add_value("weight")
.after(Font::Variant)
.to_string(PostgresQueryBuilder),
r#"ALTER TYPE "font" ADD VALUE 'weight' AFTER 'variant'"#
)
}
#[test]
fn alter_4() {
assert_eq!(
Type::alter()
.name(Font::Table)
.rename_to("typeface")
.to_string(PostgresQueryBuilder),
r#"ALTER TYPE "font" RENAME TO "typeface""#
)
}
#[test]
fn alter_5() {
assert_eq!(
Type::alter()
.name(Font::Table)
.rename_value(Font::Variant, Font::Language)
.to_string(PostgresQueryBuilder),
r#"ALTER TYPE "font" RENAME VALUE 'variant' TO 'language'"#
)
}
#[test]
fn alter_6() {
assert_eq!(
Type::alter()
.name(("schema", Font::Table))
.rename_to("typeface")
.to_string(PostgresQueryBuilder),
r#"ALTER TYPE "schema"."font" RENAME TO "typeface""#
)
}
#[test]
fn unsigned_types() {
let query_builder = PostgresQueryBuilder {};
let column_to_string = |column_type| {
let mut out = String::new();
query_builder.prepare_column_type(column_type, &mut out);
out
};
assert_eq!(column_to_string(&ColumnType::TinyUnsigned), "smallint");
assert_eq!(column_to_string(&ColumnType::SmallUnsigned), "integer");
assert_eq!(column_to_string(&ColumnType::Unsigned), "bigint");
assert_eq!(column_to_string(&ColumnType::BigUnsigned), "bigint");
}