@@ -42,79 +42,26 @@ async fn client_create_stream_error_test() {
42
42
43
43
#[ tokio:: test( flavor = "multi_thread" ) ]
44
44
async fn client_create_and_delete_super_stream_test ( ) {
45
- let super_stream_name = "test-super-stream" ;
46
-
47
- let client = Client :: connect ( ClientOptions :: default ( ) ) . await . unwrap ( ) ;
48
-
49
- let partitions: Vec < String > = [
50
- "test-super-stream-0" ,
51
- "test-super-stream-1" ,
52
- "test-super-stream-2" ,
53
- ]
54
- . iter ( )
55
- . map ( |& x| x. into ( ) )
56
- . collect ( ) ;
57
-
58
- let binding_keys: Vec < String > = [ "0" , "1" , "2" ] . iter ( ) . map ( |& x| x. into ( ) ) . collect ( ) ;
59
-
60
- let response = client
61
- . create_super_stream ( & super_stream_name, partitions, binding_keys, HashMap :: new ( ) )
62
- . await
63
- . unwrap ( ) ;
64
-
65
- assert_eq ! ( & ResponseCode :: Ok , response. code( ) ) ;
66
-
67
- let response = client
68
- . delete_super_stream ( & super_stream_name)
69
- . await
70
- . unwrap ( ) ;
71
-
72
- assert_eq ! ( & ResponseCode :: Ok , response. code( ) ) ;
73
-
74
- let _ = client. close ( ) . await ;
45
+ let test = TestClient :: create_super_stream ( ) . await ;
75
46
}
47
+
76
48
#[ tokio:: test( flavor = "multi_thread" ) ]
77
49
async fn client_create_super_stream_error_test ( ) {
78
- let super_stream_name = "test-super-stream-error" ;
79
-
80
- let client = Client :: connect ( ClientOptions :: default ( ) ) . await . unwrap ( ) ;
81
-
82
- let partitions: Vec < String > = [
83
- "test-super-stream-error-0" ,
84
- "test-super-stream-error-1" ,
85
- "test-super-stream-error-2" ,
86
- ]
87
- . iter ( )
88
- . map ( |& x| x. into ( ) )
89
- . collect ( ) ;
90
-
50
+ let test = TestClient :: create_super_stream ( ) . await ;
91
51
let binding_keys: Vec < String > = [ "0" , "1" , "2" ] . iter ( ) . map ( |& x| x. into ( ) ) . collect ( ) ;
92
52
93
- let response = client
53
+ let response = test
54
+ . client
94
55
. create_super_stream (
95
- & super_stream_name ,
96
- partitions. clone ( ) ,
97
- binding_keys. clone ( ) ,
56
+ & test . super_stream ,
57
+ test . partitions . clone ( ) ,
58
+ binding_keys,
98
59
HashMap :: new ( ) ,
99
60
)
100
61
. await
101
62
. unwrap ( ) ;
102
63
103
- assert_eq ! ( & ResponseCode :: Ok , response. code( ) ) ;
104
-
105
- let response = client
106
- . create_super_stream ( & super_stream_name, partitions, binding_keys, HashMap :: new ( ) )
107
- . await
108
- . unwrap ( ) ;
109
-
110
64
assert_eq ! ( & ResponseCode :: StreamAlreadyExists , response. code( ) ) ;
111
-
112
- let response = client
113
- . delete_super_stream ( & super_stream_name)
114
- . await
115
- . unwrap ( ) ;
116
-
117
- assert_eq ! ( & ResponseCode :: Ok , response. code( ) ) ;
118
65
}
119
66
async fn client_delete_stream_test ( ) {
120
67
let test = TestClient :: create ( ) . await ;
@@ -463,86 +410,40 @@ async fn client_exchange_command_versions() {
463
410
464
411
#[ tokio:: test( flavor = "multi_thread" ) ]
465
412
async fn client_test_partitions_test ( ) {
466
- let super_stream_name = "test-super-stream-partitions" ;
467
-
468
- let client = Client :: connect ( ClientOptions :: default ( ) ) . await . unwrap ( ) ;
469
-
470
- let partitions: Vec < String > = [
471
- "test-super-stream-partitions-0" ,
472
- "test-super-stream-partitions-1" ,
473
- "test-super-stream-partitions-2" ,
474
- ]
475
- . iter ( )
476
- . map ( |& x| x. into ( ) )
477
- . collect ( ) ;
478
-
479
- let binding_keys: Vec < String > = [ "0" , "1" , "2" ] . iter ( ) . map ( |& x| x. into ( ) ) . collect ( ) ;
480
-
481
- let response = client
482
- . create_super_stream ( & super_stream_name, partitions, binding_keys, HashMap :: new ( ) )
483
- . await
484
- . unwrap ( ) ;
485
-
486
- assert_eq ! ( & ResponseCode :: Ok , response. code( ) ) ;
487
-
488
- let response = client
489
- . partitions ( super_stream_name. to_string ( ) )
490
- . await
491
- . unwrap ( ) ;
492
-
493
- assert_eq ! ( response. streams. get( 0 ) . unwrap( ) , "test-super-stream-partitions-0" ) ;
494
- assert_eq ! ( response. streams. get( 1 ) . unwrap( ) , "test-super-stream-partitions-1" ) ;
495
- assert_eq ! ( response. streams. get( 2 ) . unwrap( ) , "test-super-stream-partitions-2" ) ;
413
+ let test = TestClient :: create_super_stream ( ) . await ;
496
414
497
- let response = client
498
- . delete_super_stream ( & super_stream_name)
415
+ let response = test
416
+ . client
417
+ . partitions ( test. super_stream . to_string ( ) )
499
418
. await
500
419
. unwrap ( ) ;
501
420
502
- assert_eq ! ( & ResponseCode :: Ok , response. code( ) ) ;
503
-
504
- let _ = client. close ( ) . await ;
421
+ assert_eq ! (
422
+ response. streams. get( 0 ) . unwrap( ) ,
423
+ test. partitions. get( 0 ) . unwrap( )
424
+ ) ;
425
+ assert_eq ! (
426
+ response. streams. get( 1 ) . unwrap( ) ,
427
+ test. partitions. get( 1 ) . unwrap( )
428
+ ) ;
429
+ assert_eq ! (
430
+ response. streams. get( 2 ) . unwrap( ) ,
431
+ test. partitions. get( 2 ) . unwrap( )
432
+ ) ;
505
433
}
506
434
507
435
#[ tokio:: test( flavor = "multi_thread" ) ]
508
436
async fn client_test_route_test ( ) {
509
- let super_stream_name = "test-super-stream-route" ;
510
-
511
- let client = Client :: connect ( ClientOptions :: default ( ) ) . await . unwrap ( ) ;
512
-
513
- let partitions: Vec < String > = [
514
- "test-super-stream-route-0" ,
515
- "test-super-stream-route-1" ,
516
- "test-super-stream-route-2" ,
517
- ]
518
- . iter ( )
519
- . map ( |& x| x. into ( ) )
520
- . collect ( ) ;
521
-
522
- let binding_keys: Vec < String > = [ "0" , "1" , "2" ] . iter ( ) . map ( |& x| x. into ( ) ) . collect ( ) ;
523
-
524
- let response = client
525
- . create_super_stream ( & super_stream_name, partitions, binding_keys, HashMap :: new ( ) )
526
- . await
527
- . unwrap ( ) ;
528
-
529
- assert_eq ! ( & ResponseCode :: Ok , response. code( ) ) ;
530
-
531
- let response = client
532
- . route ( "0" . to_string ( ) , super_stream_name. to_string ( ) )
437
+ let test = TestClient :: create_super_stream ( ) . await ;
438
+ let response = test
439
+ . client
440
+ . route ( "0" . to_string ( ) , test. super_stream . to_string ( ) )
533
441
. await
534
442
. unwrap ( ) ;
535
443
536
444
assert_eq ! ( response. streams. len( ) , 1 ) ;
537
- assert_eq ! ( response. streams. get( 0 ) . unwrap( ) , "test-super-stream-route-0" ) ;
538
-
539
-
540
- let response = client
541
- . delete_super_stream ( & super_stream_name)
542
- . await
543
- . unwrap ( ) ;
544
-
545
- assert_eq ! ( & ResponseCode :: Ok , response. code( ) ) ;
546
-
547
- let _ = client. close ( ) . await ;
548
- }
445
+ assert_eq ! (
446
+ response. streams. get( 0 ) . unwrap( ) ,
447
+ test. partitions. get( 0 ) . unwrap( )
448
+ ) ;
449
+ }
0 commit comments