mirror of
https://github.com/launchbadge/sqlx
synced 2024-09-20 14:21:57 +00:00
style: rustfmt
This commit is contained in:
parent
7c4c185698
commit
9abdd7e408
2 changed files with 17 additions and 8 deletions
|
@ -17,7 +17,6 @@ where
|
||||||
Ok(DB::Connection::connect(&env::var("DATABASE_URL")?).await?)
|
Ok(DB::Connection::connect(&env::var("DATABASE_URL")?).await?)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// Make a new pool
|
// Make a new pool
|
||||||
// Ensure [dotenv] and [env_logger] have been setup
|
// Ensure [dotenv] and [env_logger] have been setup
|
||||||
pub async fn pool<DB>() -> anyhow::Result<Pool<DB>>
|
pub async fn pool<DB>() -> anyhow::Result<Pool<DB>>
|
||||||
|
|
|
@ -1,7 +1,7 @@
|
||||||
use futures::TryStreamExt;
|
use futures::TryStreamExt;
|
||||||
use sqlx::postgres::PgRow;
|
use sqlx::postgres::PgRow;
|
||||||
use sqlx::postgres::{PgDatabaseError, PgErrorPosition, PgSeverity};
|
use sqlx::postgres::{PgDatabaseError, PgErrorPosition, PgSeverity};
|
||||||
use sqlx::{postgres::Postgres, Connection, Executor, Row, PgPool};
|
use sqlx::{postgres::Postgres, Connection, Executor, PgPool, Row};
|
||||||
use sqlx_test::new;
|
use sqlx_test::new;
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
|
|
||||||
|
@ -150,11 +150,16 @@ async fn it_can_fail_and_recover() -> anyhow::Result<()> {
|
||||||
|
|
||||||
for i in 0..10 {
|
for i in 0..10 {
|
||||||
// make a query that will fail
|
// make a query that will fail
|
||||||
let res = conn.execute("INSERT INTO not_found (column) VALUES (10)").await;
|
let res = conn
|
||||||
|
.execute("INSERT INTO not_found (column) VALUES (10)")
|
||||||
|
.await;
|
||||||
assert!(res.is_err());
|
assert!(res.is_err());
|
||||||
|
|
||||||
// now try and use the connection
|
// now try and use the connection
|
||||||
let val: i32 = conn.fetch_one(&*format!("SELECT {}::int4", i)).await?.get(0);
|
let val: i32 = conn
|
||||||
|
.fetch_one(&*format!("SELECT {}::int4", i))
|
||||||
|
.await?
|
||||||
|
.get(0);
|
||||||
assert_eq!(val, i);
|
assert_eq!(val, i);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -167,11 +172,16 @@ async fn it_can_fail_and_recover_with_pool() -> anyhow::Result<()> {
|
||||||
|
|
||||||
for i in 0..10 {
|
for i in 0..10 {
|
||||||
// make a query that will fail
|
// make a query that will fail
|
||||||
let res = pool.execute("INSERT INTO not_found (column) VALUES (10)").await;
|
let res = pool
|
||||||
|
.execute("INSERT INTO not_found (column) VALUES (10)")
|
||||||
|
.await;
|
||||||
assert!(res.is_err());
|
assert!(res.is_err());
|
||||||
|
|
||||||
// now try and use the connection
|
// now try and use the connection
|
||||||
let val: i32 = pool.fetch_one(&*format!("SELECT {}::int4", i)).await?.get(0);
|
let val: i32 = pool
|
||||||
|
.fetch_one(&*format!("SELECT {}::int4", i))
|
||||||
|
.await?
|
||||||
|
.get(0);
|
||||||
assert_eq!(val, i);
|
assert_eq!(val, i);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue