mirror of
https://github.com/launchbadge/sqlx
synced 2024-11-10 06:24:16 +00:00
1d8eb2add4
* fix: fixture macro attribute * remove extra new line * add extra new line * feat: add test for slqx::test macro * feat: update test for sqlx::test macro * remove old macro test * feat: add postgres and sqlite test * rust format * cargo fmt * fix fixtures execution order in test
107 lines
3 KiB
Rust
107 lines
3 KiB
Rust
// The no-arg variant is covered by other tests already.
|
|
|
|
use sqlx::{Row, SqlitePool};
|
|
|
|
const MIGRATOR: sqlx::migrate::Migrator = sqlx::migrate!("tests/sqlite/migrations");
|
|
|
|
#[sqlx::test]
|
|
async fn it_gets_a_pool(pool: SqlitePool) -> sqlx::Result<()> {
|
|
let mut conn = pool.acquire().await?;
|
|
|
|
// https://www.sqlite.org/pragma.html#pragma_database_list
|
|
let db = sqlx::query("PRAGMA database_list")
|
|
.fetch_one(&mut *conn)
|
|
.await?;
|
|
|
|
let db_name = db.get::<String, _>(2);
|
|
|
|
assert!(
|
|
db_name.ends_with("target/sqlx/test-dbs/sqlite_test_attr/it_gets_a_pool.sqlite"),
|
|
"db_name: {:?}",
|
|
db_name
|
|
);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
// This should apply migrations and then `fixtures/users.sql`
|
|
#[sqlx::test(migrations = "tests/sqlite/migrations", fixtures("users"))]
|
|
async fn it_gets_users(pool: SqlitePool) -> sqlx::Result<()> {
|
|
let usernames: Vec<String> =
|
|
sqlx::query_scalar(r#"SELECT username FROM "user" ORDER BY username"#)
|
|
.fetch_all(&pool)
|
|
.await?;
|
|
|
|
assert_eq!(usernames, ["alice", "bob"]);
|
|
|
|
let post_exists: bool = sqlx::query_scalar("SELECT exists(SELECT 1 FROM post)")
|
|
.fetch_one(&pool)
|
|
.await?;
|
|
|
|
assert!(!post_exists);
|
|
|
|
let comment_exists: bool = sqlx::query_scalar("SELECT exists(SELECT 1 FROM comment)")
|
|
.fetch_one(&pool)
|
|
.await?;
|
|
|
|
assert!(!comment_exists);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx::test(migrations = "tests/sqlite/migrations", fixtures("users", "posts"))]
|
|
async fn it_gets_posts(pool: SqlitePool) -> sqlx::Result<()> {
|
|
let post_contents: Vec<String> =
|
|
sqlx::query_scalar("SELECT content FROM post ORDER BY created_at")
|
|
.fetch_all(&pool)
|
|
.await?;
|
|
|
|
assert_eq!(
|
|
post_contents,
|
|
[
|
|
"This new computer is lightning-fast!",
|
|
"@alice is a haxxor :("
|
|
]
|
|
);
|
|
|
|
let comment_exists: bool = sqlx::query_scalar("SELECT exists(SELECT 1 FROM comment)")
|
|
.fetch_one(&pool)
|
|
.await?;
|
|
|
|
assert!(!comment_exists);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
// Try `migrator`
|
|
#[sqlx::test(migrator = "MIGRATOR", fixtures("users", "posts", "comments"))]
|
|
async fn it_gets_comments(pool: SqlitePool) -> sqlx::Result<()> {
|
|
let post_1_comments: Vec<String> =
|
|
sqlx::query_scalar("SELECT content FROM comment WHERE post_id = ? ORDER BY created_at")
|
|
.bind(&1)
|
|
.fetch_all(&pool)
|
|
.await?;
|
|
|
|
assert_eq!(
|
|
post_1_comments,
|
|
["lol bet ur still bad, 1v1 me", "you're on!"]
|
|
);
|
|
|
|
let post_2_comments: Vec<String> =
|
|
sqlx::query_scalar("SELECT content FROM comment WHERE post_id = ? ORDER BY created_at")
|
|
.bind(&2)
|
|
.fetch_all(&pool)
|
|
.await?;
|
|
|
|
assert_eq!(post_2_comments, ["lol you're just mad you lost :P"]);
|
|
|
|
Ok(())
|
|
}
|
|
|
|
#[sqlx::test(
|
|
migrations = "tests/sqlite/migrations",
|
|
fixtures(path = "./fixtures", scripts("users", "posts"))
|
|
)]
|
|
async fn this_should_compile(_pool: SqlitePool) -> sqlx::Result<()> {
|
|
Ok(())
|
|
}
|