Fix tests

Author: weiznich

src/lib.rs

@@ -20,7 +20,6 @@ pub use self::run_query_dsl::*;
 pub use self::stmt_cache::StmtCache;
 pub use self::transaction_manager::{AnsiTransactionManager, TransactionManager};
 
-
 #[async_trait::async_trait]
 pub trait SimpleAsyncConnection {
     async fn batch_execute(&mut self, query: &str) -> QueryResult<()>;

src/mysql/mod.rs

@@ -146,7 +146,8 @@ impl AsyncMysqlConnection {
         ];
 
         for stmt in setup_statements {
-            diesel::sql_query(stmt).execute(&mut conn)
+            diesel::sql_query(stmt)
+                .execute(&mut conn)
                 .await
                 .map_err(ConnectionError::CouldntSetupConfiguration)?;
         }

src/pg/mod.rs

@@ -207,8 +207,12 @@ impl AsyncPgConnection {
     async fn set_config_options(&mut self) -> QueryResult<()> {
         use crate::run_query_dsl::RunQueryDsl;
 
-        diesel::sql_query("SET TIME ZONE 'UTC'").execute(self).await?;
-        diesel::sql_query("SET CLIENT_ENCODING TO 'UTF8'").execute(self).await?;
+        diesel::sql_query("SET TIME ZONE 'UTC'")
+            .execute(self)
+            .await?;
+        diesel::sql_query("SET CLIENT_ENCODING TO 'UTF8'")
+            .execute(self)
+            .await?;
         Ok(())
     }
 
@@ -232,9 +236,7 @@ impl AsyncPgConnection {
             let res = query.collect_binds(&mut bind_collector, self, &diesel::pg::Pg);
 
             if !self.next_lookup.is_empty() {
-                for (schema, lookup_type_name) in
-                    std::mem::take(&mut self.next_lookup)
-                {
+                for (schema, lookup_type_name) in std::mem::take(&mut self.next_lookup) {
                     // as this is an async call and we don't want to infect the whole diesel serialization
                     // api with async we just error out in the `PgMetadataLookup` implementation below if we encounter
                     // a type that is not cached yet

tests/lib.rs

@@ -77,9 +77,36 @@ async fn test_basic_insert_and_load() -> QueryResult<()> {
     Ok(())
 }
 
+#[cfg(feature = "mysql")]
+async fn setup(connection: &mut TestConnection) -> TestConnection {
+    diesel::sql_query(
+        "CREATE TABLE IF NOT EXISTS users (
+                id INTEGER PRIMARY KEY AUTO_INCREMENT,
+                name TEXT NOT NULL
+            ) CHARACTER SET utf8mb4",
+    )
+    .execute(&mut connection)
+    .await
+    .unwrap();
+}
+
+#[cfg(feature = "postgres")]
+async fn setup(connection: &mut TestConnection) -> TestConnection {
+    diesel::sql_query(
+        "CREATE TABLE IF NOT EXISTS users (
+                id SERIAL PRIMARY KEY,
+                name VARCHAR NOT NULL
+            )",
+    )
+    .execute(&mut connection)
+    .await
+    .unwrap();
+}
+
 async fn connection() -> TestConnection {
     let db_url = std::env::var("DATABASE_URL").unwrap();
     let mut conn = TestConnection::establish(&db_url).await.unwrap();
     conn.begin_test_transaction().await.unwrap();
+    setup(&mut conn);
     conn
 }