Skip to content

Commit

Permalink
fmt
Browse files Browse the repository at this point in the history
  • Loading branch information
varshith257 authored Oct 20, 2024
1 parent f837b6b commit 1dcb165
Showing 1 changed file with 0 additions and 37 deletions.
37 changes: 0 additions & 37 deletions extension/tests/integration_tests.rs
Original file line number Diff line number Diff line change
Expand Up @@ -860,40 +860,3 @@ async fn test_cohere() {
.unwrap();
assert_eq!(search_results.len(), 3);
}

#[ignore]
#[tokio::test]
async fn test_drop_table_triggers_job_deletion() {
let conn = common::init_database().await;
let mut rng = rand::thread_rng();
let test_num = rng.gen_range(1..100000);
let test_table_name = format!("drop_test_table_{}", test_num);
let job_name = format!("job_{}", test_num);

common::init_test_table(&test_table_name, &conn).await;

let insert_job_query = format!(
"INSERT INTO vectorize.job (name, index_dist_type, transformer, search_alg, params, last_completion)
VALUES ('{test_table_name}', 'pgv_hsnw_cosine', 'sentence-transformers/all-MiniLM-L6-v2', 'search_algorithm', '{{}}', NOW());"
);
sqlx::query(&insert_job_query)
.execute(&conn)
.await
.expect("failed to insert job");

let rowcount_before = common::row_count("vectorize.job", &conn).await;
assert!(rowcount_before >= 1);

let drop_table_query = format!("DROP TABLE {test_table_name};");
sqlx::query(&drop_table_query)
.execute(&conn)
.await
.expect("failed to drop table");

let rowcount_after = common::row_count("vectorize.job", &conn).await;
assert_eq!(
rowcount_after,
rowcount_before - 1,
"Job was not deleted after table drop"
);
}

0 comments on commit 1dcb165

Please sign in to comment.