Add test case

SeaQL · Jan 27, 2024 · 7d2604f · 7d2604f
1 parent acc54c9
commit 7d2604f
Showing 1 changed file with 32 additions and 0 deletions.
diff --git a/tests/postgres/types.rs b/tests/postgres/types.rs
@@ -24,6 +24,38 @@ fn create_2() {
  );
 }
 
+#[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, s: &mut dyn std::fmt::Write) {
+ write!(
+ s,
+ "{}",
+ match self {
+ Self::Enum => "tea",
+ Self::EverydayTea => "EverydayTea",
+ Self::BreakfastTea => "BreakfastTea",
+ }
+ )
+ .unwrap();
+ }
+ }
+}
+
 #[test]
 fn drop_1() {
  assert_eq!(