@@ -746,30 +746,30 @@ mod tests {
746
746
#[ tokio:: test]
747
747
async fn test_put_kv_happy_path ( ) {
748
748
// 3 servers 1, 2, 3
749
- // Targeted key has affinity [2, 3 , 1].
749
+ // Targeted key has affinity [3, 2 , 1].
750
750
//
751
751
// Put on 2 and 3 is successful
752
- // Get succeeds on 2 .
752
+ // Get succeeds on 3 .
753
753
let mock_search_service_1 = MockSearchService :: new ( ) ;
754
754
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 (
756
762
|put_req : quickwit_proto:: search:: PutKvRequest | {
757
763
assert_eq ! ( put_req. key, b"my_key" ) ;
758
764
assert_eq ! ( put_req. payload, b"my_payload" ) ;
759
765
} ,
760
766
) ;
761
- mock_search_service_2 . expect_get_kv ( ) . once ( ) . returning (
767
+ mock_search_service_3 . expect_get_kv ( ) . once ( ) . returning (
762
768
|get_req : quickwit_proto:: search:: GetKvRequest | {
763
769
assert_eq ! ( get_req. key, b"my_key" ) ;
764
770
Some ( b"my_payload" . to_vec ( ) )
765
771
} ,
766
772
) ;
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 | { } ) ;
773
773
let searcher_pool = searcher_pool_for_test ( [
774
774
( "127.0.0.1:1001" , mock_search_service_1) ,
775
775
( "127.0.0.1:1002" , mock_search_service_2) ,
@@ -791,11 +791,11 @@ mod tests {
791
791
#[ tokio:: test]
792
792
async fn test_put_kv_failing_get ( ) {
793
793
// 3 servers 1, 2, 3
794
- // Targeted key has affinity [2, 3 , 1].
794
+ // Targeted key has affinity [3, 2 , 1].
795
795
//
796
796
// 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 .
799
799
let mock_search_service_1 = MockSearchService :: new ( ) ;
800
800
let mut mock_search_service_2 = MockSearchService :: new ( ) ;
801
801
mock_search_service_2. expect_put_kv ( ) . once ( ) . returning (
@@ -807,7 +807,7 @@ mod tests {
807
807
mock_search_service_2. expect_get_kv ( ) . once ( ) . returning (
808
808
|get_req : quickwit_proto:: search:: GetKvRequest | {
809
809
assert_eq ! ( get_req. key, b"my_key" ) ;
810
- None
810
+ Some ( b"my_payload" . to_vec ( ) )
811
811
} ,
812
812
) ;
813
813
let mut mock_search_service_3 = MockSearchService :: new ( ) ;
@@ -820,7 +820,7 @@ mod tests {
820
820
mock_search_service_3. expect_get_kv ( ) . once ( ) . returning (
821
821
|get_req : quickwit_proto:: search:: GetKvRequest | {
822
822
assert_eq ! ( get_req. key, b"my_key" ) ;
823
- Some ( b"my_payload" . to_vec ( ) )
823
+ None
824
824
} ,
825
825
) ;
826
826
let searcher_pool = searcher_pool_for_test ( [
0 commit comments