81 lines
2.6 KiB
Rust
81 lines
2.6 KiB
Rust
use bzod::config::Config;
|
|
use bzod::db::Db;
|
|
use std::fs;
|
|
use std::path::PathBuf;
|
|
use std::sync::Arc;
|
|
use tokio::sync::Barrier;
|
|
|
|
fn create_temp_config(temp_dir: PathBuf) -> Config {
|
|
let mut config = Config::load_with_cli(Some(&temp_dir)).unwrap();
|
|
config.backup_dir = temp_dir.clone();
|
|
config.base_url = Some("http://bzo.in".to_string());
|
|
config
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn test_concurrent_slug_creation() {
|
|
let temp_dir =
|
|
std::env::temp_dir().join(format!("bzod_test_concurrency_{}", uuid::Uuid::new_v4()));
|
|
fs::create_dir_all(&temp_dir).unwrap();
|
|
let config = create_temp_config(temp_dir.clone());
|
|
|
|
let db = Db::init(&config).expect("Failed to init Db");
|
|
|
|
let barrier = Arc::new(Barrier::new(2));
|
|
let db_path = temp_dir.join("slugs/global_urls.db");
|
|
|
|
let b1 = barrier.clone();
|
|
let path1 = db_path.clone();
|
|
let task1 = tokio::spawn(async move {
|
|
let conn = rusqlite::Connection::open(&path1).unwrap();
|
|
b1.wait().await;
|
|
let now = chrono::Utc::now().to_rfc3339();
|
|
conn.execute(
|
|
"INSERT INTO global_urls (slug, owner_tenant_id, target_id, created_at, updated_at, status, retired_at)
|
|
VALUES ('!conc-slug', 't-tenant10', 'url10', ?1, ?2, 'active', NULL);",
|
|
rusqlite::params![now, now],
|
|
)
|
|
});
|
|
|
|
let b2 = barrier.clone();
|
|
let path2 = db_path.clone();
|
|
let task2 = tokio::spawn(async move {
|
|
let conn = rusqlite::Connection::open(&path2).unwrap();
|
|
b2.wait().await;
|
|
let now = chrono::Utc::now().to_rfc3339();
|
|
conn.execute(
|
|
"INSERT INTO global_urls (slug, owner_tenant_id, target_id, created_at, updated_at, status, retired_at)
|
|
VALUES ('!conc-slug', 't-tenant20', 'url20', ?1, ?2, 'active', NULL);",
|
|
rusqlite::params![now, now],
|
|
)
|
|
});
|
|
|
|
let res1 = task1.await.unwrap();
|
|
let res2 = task2.await.unwrap();
|
|
|
|
// Exactly one should succeed, and one should fail (due to UNIQUE constraint)
|
|
match (res1, res2) {
|
|
(Ok(_), Err(_)) => {}
|
|
(Err(_), Ok(_)) => {}
|
|
(r1, r2) => panic!(
|
|
"Concurrent creation outcome invalid. Res1: {:?}, Res2: {:?}",
|
|
r1, r2
|
|
),
|
|
}
|
|
|
|
// Verify exactly 1 record exists in global_urls
|
|
{
|
|
let urls_conn = db.global_urls.lock().unwrap();
|
|
let count: i64 = urls_conn
|
|
.query_row(
|
|
"SELECT COUNT(*) FROM global_urls WHERE slug = '!conc-slug';",
|
|
[],
|
|
|row| row.get(0),
|
|
)
|
|
.unwrap();
|
|
assert_eq!(count, 1);
|
|
}
|
|
|
|
let _ = fs::remove_dir_all(&temp_dir);
|
|
}
|