Skip to content

Commit cbf35fa

Browse files
fix failing test after rustc and serde update (#5564)
1 parent 297aabe commit cbf35fa

File tree

4 files changed

+22
-21
lines changed

4 files changed

+22
-21
lines changed

quickwit/quickwit-common/src/rendezvous_hasher.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ mod tests {
6464
let mut socket_set3 = vec![socket1, socket4];
6565
sort_by_rendez_vous_hash(&mut socket_set3, "key");
6666

67-
assert_eq!(socket_set1, &[socket1, socket2, socket3, socket4]);
67+
assert_eq!(socket_set1, &[socket1, socket3, socket2, socket4]);
6868
assert_eq!(socket_set2, &[socket1, socket2, socket4]);
6969
assert_eq!(socket_set3, &[socket1, socket4]);
7070
}

quickwit/quickwit-search/src/cluster_client.rs

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -746,30 +746,30 @@ mod tests {
746746
#[tokio::test]
747747
async fn test_put_kv_happy_path() {
748748
// 3 servers 1, 2, 3
749-
// Targeted key has affinity [2, 3, 1].
749+
// Targeted key has affinity [3, 2, 1].
750750
//
751751
// Put on 2 and 3 is successful
752-
// Get succeeds on 2.
752+
// Get succeeds on 3.
753753
let mock_search_service_1 = MockSearchService::new();
754754
let mut mock_search_service_2 = MockSearchService::new();
755-
mock_search_service_2.expect_put_kv().once().returning(
755+
// Due to the buffered call it is possible for the
756+
// put request to 2 to be emitted too.
757+
mock_search_service_2
758+
.expect_put_kv()
759+
.returning(|_put_req: quickwit_proto::search::PutKvRequest| {});
760+
let mut mock_search_service_3 = MockSearchService::new();
761+
mock_search_service_3.expect_put_kv().once().returning(
756762
|put_req: quickwit_proto::search::PutKvRequest| {
757763
assert_eq!(put_req.key, b"my_key");
758764
assert_eq!(put_req.payload, b"my_payload");
759765
},
760766
);
761-
mock_search_service_2.expect_get_kv().once().returning(
767+
mock_search_service_3.expect_get_kv().once().returning(
762768
|get_req: quickwit_proto::search::GetKvRequest| {
763769
assert_eq!(get_req.key, b"my_key");
764770
Some(b"my_payload".to_vec())
765771
},
766772
);
767-
let mut mock_search_service_3 = MockSearchService::new();
768-
// Due to the buffered call it is possible for the
769-
// put request to 3 to be emitted too.
770-
mock_search_service_3
771-
.expect_put_kv()
772-
.returning(|_put_req: quickwit_proto::search::PutKvRequest| {});
773773
let searcher_pool = searcher_pool_for_test([
774774
("127.0.0.1:1001", mock_search_service_1),
775775
("127.0.0.1:1002", mock_search_service_2),
@@ -791,11 +791,11 @@ mod tests {
791791
#[tokio::test]
792792
async fn test_put_kv_failing_get() {
793793
// 3 servers 1, 2, 3
794-
// Targeted key has affinity [2, 3, 1].
794+
// Targeted key has affinity [3, 2, 1].
795795
//
796796
// Put on 2 and 3 is successful
797-
// Get fails on 2.
798-
// Get succeeds on 3.
797+
// Get fails on 3.
798+
// Get succeeds on 2.
799799
let mock_search_service_1 = MockSearchService::new();
800800
let mut mock_search_service_2 = MockSearchService::new();
801801
mock_search_service_2.expect_put_kv().once().returning(
@@ -807,7 +807,7 @@ mod tests {
807807
mock_search_service_2.expect_get_kv().once().returning(
808808
|get_req: quickwit_proto::search::GetKvRequest| {
809809
assert_eq!(get_req.key, b"my_key");
810-
None
810+
Some(b"my_payload".to_vec())
811811
},
812812
);
813813
let mut mock_search_service_3 = MockSearchService::new();
@@ -820,7 +820,7 @@ mod tests {
820820
mock_search_service_3.expect_get_kv().once().returning(
821821
|get_req: quickwit_proto::search::GetKvRequest| {
822822
assert_eq!(get_req.key, b"my_key");
823-
Some(b"my_payload".to_vec())
823+
None
824824
},
825825
);
826826
let searcher_pool = searcher_pool_for_test([

quickwit/quickwit-search/src/search_job_placer.rs

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -427,21 +427,22 @@ mod tests {
427427

428428
let expected_searcher_addr_1: SocketAddr = ([127, 0, 0, 1], 1001).into();
429429
let expected_searcher_addr_2: SocketAddr = ([127, 0, 0, 1], 1002).into();
430+
// on a small number of splits, we may be unbalanced
430431
let expected_assigned_jobs = vec![
431432
(
432433
expected_searcher_addr_1,
433434
vec![
434-
SearchJob::for_test("split5", 5),
435-
SearchJob::for_test("split4", 4),
436435
SearchJob::for_test("split3", 3),
436+
SearchJob::for_test("split2", 2),
437+
SearchJob::for_test("split1", 1),
437438
],
438439
),
439440
(
440441
expected_searcher_addr_2,
441442
vec![
442443
SearchJob::for_test("split6", 6),
443-
SearchJob::for_test("split2", 2),
444-
SearchJob::for_test("split1", 1),
444+
SearchJob::for_test("split5", 5),
445+
SearchJob::for_test("split4", 4),
445446
],
446447
),
447448
];

quickwit/quickwit-serve/src/elasticsearch_api/model/bulk_query_params.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ mod tests {
114114
serde_qs::from_str::<ElasticBulkOptions>("refresh=wait")
115115
.unwrap_err()
116116
.to_string(),
117-
"unknown variant `wait`, expected one of `false`, `true`, `wait_for`"
117+
"unknown variant `wait`, expected one of `false`, ``, `true`, `wait_for`"
118118
);
119119
}
120120
}

0 commit comments

Comments
 (0)