Transaction with BEGIN

SeaQL · Oct 14, 2021 · 75882b3 · 75882b3
1 parent 9abfdf4
commit 75882b3
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/src/database/mock.rs b/src/database/mock.rs
@@ -227,7 +227,10 @@ impl Transaction {
 impl OpenTransaction {
     fn init() -> Self {
         Self {
-            stmts: Vec::new(),
+            stmts: vec![Statement::from_string(
+                DbBackend::Postgres,
+                "BEGIN".to_owned(),
+            )],
             transaction_depth: 0,
         }
     }
@@ -321,6 +324,7 @@ mod tests {
             db.into_transaction_log(),
             vec![
                 Transaction::many(vec![
+                    Statement::from_string(DbBackend::Postgres, "BEGIN".to_owned()),
                     Statement::from_sql_and_values(
                         DbBackend::Postgres,
                         r#"SELECT "cake"."id", "cake"."name" FROM "cake" LIMIT $1"#,
@@ -365,6 +369,7 @@ mod tests {
         assert_eq!(
             db.into_transaction_log(),
             vec![Transaction::many(vec![
+                Statement::from_string(DbBackend::Postgres, "BEGIN".to_owned()),
                 Statement::from_sql_and_values(
                     DbBackend::Postgres,
                     r#"SELECT "cake"."id", "cake"."name" FROM "cake" LIMIT $1"#,
@@ -402,6 +407,7 @@ mod tests {
         assert_eq!(
             db.into_transaction_log(),
             vec![Transaction::many(vec![
+                Statement::from_string(DbBackend::Postgres, "BEGIN".to_owned()),
                 Statement::from_sql_and_values(
                     DbBackend::Postgres,
                     r#"SELECT "cake"."id", "cake"."name" FROM "cake" LIMIT $1"#,
@@ -459,6 +465,7 @@ mod tests {
         assert_eq!(
             db.into_transaction_log(),
             vec![Transaction::many(vec![
+                Statement::from_string(DbBackend::Postgres, "BEGIN".to_owned()),
                 Statement::from_sql_and_values(
                     DbBackend::Postgres,
                     r#"SELECT "cake"."id", "cake"."name" FROM "cake" LIMIT $1"#,

diff --git a/tests/uuid_tests.rs b/tests/uuid_tests.rs
@@ -1,7 +1,7 @@
 pub mod common;
 
 pub use common::{bakery_chain::*, setup::*, TestContext};
-use sea_orm::{entity::*, entity::prelude::*, DatabaseConnection};
+use sea_orm::{entity::prelude::*, entity::*, DatabaseConnection};
 
 #[sea_orm_macros::test]
 #[cfg(any(
@@ -27,7 +27,8 @@ pub async fn insert_metadata(db: &DatabaseConnection) -> Result<(), DbErr> {
         bytes: vec![1, 2, 3],
         date: Some(Date::from_ymd(2021, 9, 27)),
         time: Some(Time::from_hms(11, 32, 55)),
-    }.into_active_model();
+    }
+    .into_active_model();
 
     let result = metadata.clone().insert(db).await?;