Skip to content

Commit a0ad970

Browse files
committed
Refactor tests to expect()
1 parent 3499a13 commit a0ad970

File tree

1 file changed

+30
-26
lines changed

1 file changed

+30
-26
lines changed

typesense/tests/client/documents_test.rs

Lines changed: 30 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -36,11 +36,11 @@ async fn run_test_schemaless_document_lifecycle() {
3636
..Default::default()
3737
};
3838

39-
let create_collection_result = client.collections().create(schema).await;
40-
assert!(
41-
create_collection_result.is_ok(),
42-
"Failed to create collection"
43-
);
39+
let _create_collection_response = client
40+
.collections()
41+
.create(schema)
42+
.await
43+
.expect("Failed to create collection");
4444

4545
let book_1_id = &new_id("document_1");
4646
let book_1 = json!({"id": book_1_id, "title": "The Hitchhiker's Guide to the Galaxy","author": "John","publication_year": 1979});
@@ -66,30 +66,32 @@ async fn run_test_schemaless_document_lifecycle() {
6666
);
6767

6868
// --- Retrieve a single document (via `document(id).retrieve()`) ---
69-
let retrieve_res = client
69+
let retrieved_book = client
7070
.collection_schemaless(&collection_name)
7171
.document(book_1_id)
7272
.retrieve()
73-
.await;
74-
assert!(retrieve_res.is_ok(), "Failed to retrieve document 1");
75-
assert_eq!(retrieve_res.unwrap(), book_1);
73+
.await
74+
.expect("Failed to retrieve document 1");
75+
assert_eq!(retrieved_book, book_1);
7676

7777
// --- Search for documents ---
7878
let search_params = SearchParameters::builder()
7979
.q("the")
8080
.query_by("title")
8181
.build();
82-
let search_res = documents_client.search(search_params).await;
83-
assert!(search_res.is_ok(), "Search failed");
84-
assert_eq!(search_res.unwrap().found, Some(2));
82+
let search_res = documents_client
83+
.search(search_params)
84+
.await
85+
.expect("Search failed");
86+
assert_eq!(search_res.found, Some(2));
8587

8688
// --- Delete a single document ---
87-
let delete_res = client
89+
let _deleted_book = client
8890
.collection_schemaless(&collection_name)
8991
.document(book_1_id)
9092
.delete()
91-
.await;
92-
assert!(delete_res.is_ok(), "Failed to delete document 1");
93+
.await
94+
.expect("Failed to delete document 1");
9395

9496
// --- Verify single deletion ---
9597
let retrieve_after_delete_res = client
@@ -107,10 +109,10 @@ async fn run_test_schemaless_document_lifecycle() {
107109
filter_by: Some("author:John".to_owned()),
108110
..Default::default()
109111
};
110-
let export_res = documents_client.export_jsonl(export_params).await;
111-
112-
assert!(export_res.is_ok(), "Export failed");
113-
let exported_jsonl = export_res.unwrap();
112+
let exported_jsonl = documents_client
113+
.export_jsonl(export_params)
114+
.await
115+
.expect("Export failed");
114116

115117
// Verify the exported content is a JSONL string with 2 lines.
116118
let lines: Vec<&str> = exported_jsonl.trim().split('\n').collect();
@@ -123,10 +125,12 @@ async fn run_test_schemaless_document_lifecycle() {
123125
filter_by: "publication_year:>1960".to_owned(),
124126
..Default::default()
125127
};
126-
let bulk_delete_res = documents_client.delete(delete_params).await;
127-
assert!(bulk_delete_res.is_ok(), "Bulk delete failed");
128+
let bulk_delete_response = documents_client
129+
.delete(delete_params)
130+
.await
131+
.expect("Bulk delete failed");
128132
// Only "The Hitchhiker's Guide to the Galaxy" (1979) should be deleted
129-
assert_eq!(bulk_delete_res.unwrap().num_deleted, 1);
133+
assert_eq!(bulk_delete_response.num_deleted, 1);
130134
}
131135

132136
// --- TESTS FOR GENERIC FEATURES ---
@@ -272,7 +276,7 @@ async fn run_test_generic_document_lifecycle() {
272276
let bulk_update_params = UpdateDocumentsParameters {
273277
filter_by: Some("publication_year:>1965".to_owned()),
274278
};
275-
let bulk_update_res = typed_collection
279+
let bulk_update_response = typed_collection
276280
.documents()
277281
.update(
278282
&BookPartial {
@@ -281,10 +285,10 @@ async fn run_test_generic_document_lifecycle() {
281285
},
282286
bulk_update_params,
283287
)
284-
.await;
285-
assert!(bulk_update_res.is_ok(), "Bulk update failed");
288+
.await
289+
.expect("Bulk update failed");
286290
// Should update book 1 (1966)
287-
assert_eq!(bulk_update_res.unwrap().num_updated, 1);
291+
assert_eq!(bulk_update_response.num_updated, 1);
288292

289293
// --- 8. Delete a single document, receiving the typed struct back ---
290294
let deleted_book = typed_collection

0 commit comments

Comments
 (0)