use sqlx::{PgPool, Postgres, Transaction};

#[sqlx_database_tester::test(
	pool(variable = "migrated_pool", migrations = "./other_dir_migrations"),
	pool(variable = "default_migrated_pool"),
	pool(variable = "empty_pool", skip_migrations)
)]
async fn test_migrations() {
	let migrated_fox = sqlx::query("SELECT * FROM fox").fetch_all(&migrated_pool).await;
	let default_migrated_bat =
		sqlx::query("SELECT * FROM bat").fetch_all(&default_migrated_pool).await;
	let nonexistent_fox = sqlx::query("SELECT * FROM fox").fetch_all(&empty_pool).await;
	assert!(migrated_fox.is_ok());
	assert!(default_migrated_bat.is_ok());
	assert!(nonexistent_fox.is_err());
}

#[sqlx_database_tester::test(pool(variable = "pool", transaction_variable = "transaction"))]
async fn test_transaction() -> std::io::Result<()> {
	run_transaction_query(&mut transaction).await;
	Ok(())
}

#[sqlx_database_tester::test(pool(variable = "empty_pool", skip_migrations))]
#[should_panic(expected = "test")]
#[allow(unreachable_code)]
async fn test_panic() {
	panic!("test");
}

#[sqlx_database_tester::test(pool(variable = "pool"))]
#[should_panic(expected = "test")]
#[allow(unreachable_code, clippy::unwrap_used)]
async fn test_own_transaction_panic() {
	let _transaction = pool.begin().await.unwrap();
	panic!("test");
}

#[sqlx_database_tester::test(pool(variable = "pool", transaction_variable = "_transaction"))]
#[should_panic(expected = "test")]
#[allow(unreachable_code)]
async fn test_transaction_panic() {
	panic!("test");
}

#[sqlx_database_tester::test(pool(variable = "empty_pool", skip_migrations))]
async fn test_case_returning() -> std::io::Result<()> {
	Ok(())
}

#[sqlx_database_tester::test(pool(variable = "empty_pool", skip_migrations))]
async fn test_pool_ownership_passed() {
	take_pool(empty_pool);
}

fn take_pool(_: PgPool) {}

async fn run_transaction_query(transaction: &mut Transaction<'_, Postgres>) {
	let bat = sqlx::query("SELECT * FROM bat").fetch_all(transaction).await;
	assert!(bat.is_ok());
}

#[cfg(feature = "sqlx-log")]
#[sqlx_database_tester::test(level = "Trace", pool(variable = "pool"))]
async fn log_set() {
	take_pool(pool);
}