29
29
import java .time .Instant ;
30
30
import java .util .Arrays ;
31
31
import java .util .Collections ;
32
+ import java .util .List ;
32
33
33
34
import org .junit .Rule ;
34
35
import org .junit .Test ;
@@ -54,12 +55,12 @@ public void existsShouldReturnTrueForExistingKeys() {
54
55
55
56
nativeCommands .set (KEY_1 , VALUE_1 );
56
57
57
- assertThat (connection .keyCommands ().exists (KEY_1_BBUFFER ). block (), is ( true ));
58
+ StepVerifier . create (connection .keyCommands ().exists (KEY_1_BBUFFER )). expectNext ( true ). verifyComplete ( );
58
59
}
59
60
60
61
@ Test // DATAREDIS-525
61
62
public void existsShouldReturnFalseForNonExistingKeys () {
62
- assertThat (connection .keyCommands ().exists (KEY_1_BBUFFER ). block (), is ( false ));
63
+ StepVerifier . create (connection .keyCommands ().exists (KEY_1_BBUFFER )). expectNext ( false ). verifyComplete ( );
63
64
}
64
65
65
66
@ Test // DATAREDIS-525
@@ -69,9 +70,9 @@ public void typeShouldReturnTypeCorrectly() {
69
70
nativeCommands .sadd (KEY_2 , VALUE_2 );
70
71
nativeCommands .hset (KEY_3 , KEY_1 , VALUE_1 );
71
72
72
- assertThat (connection .keyCommands ().type (KEY_1_BBUFFER ). block (), is ( DataType .STRING ));
73
- assertThat (connection .keyCommands ().type (KEY_2_BBUFFER ). block (), is ( DataType .SET ));
74
- assertThat (connection .keyCommands ().type (KEY_3_BBUFFER ). block (), is ( DataType .HASH ));
73
+ StepVerifier . create (connection .keyCommands ().type (KEY_1_BBUFFER )). expectNext ( DataType .STRING ). verifyComplete ( );
74
+ StepVerifier . create (connection .keyCommands ().type (KEY_2_BBUFFER )). expectNext ( DataType .SET ). verifyComplete ( );
75
+ StepVerifier . create (connection .keyCommands ().type (KEY_3_BBUFFER )). expectNext ( DataType .HASH ). verifyComplete ( );
75
76
}
76
77
77
78
@ Test // DATAREDIS-525
@@ -85,8 +86,12 @@ public void keysShouldReturnCorrectly() {
85
86
nativeCommands .set (VALUE_2 , KEY_2 );
86
87
nativeCommands .set (VALUE_3 , KEY_3 );
87
88
88
- assertThat (connection .keyCommands ().keys (ByteBuffer .wrap ("*" .getBytes ())).block (), hasSize (6 ));
89
- assertThat (connection .keyCommands ().keys (ByteBuffer .wrap ("key*" .getBytes ())).block (), hasSize (3 ));
89
+ StepVerifier .create (connection .keyCommands ().keys (ByteBuffer .wrap ("*" .getBytes ())).flatMapIterable (it -> it )) //
90
+ .expectNextCount (6 ) //
91
+ .verifyComplete ();
92
+
93
+ StepVerifier .create (connection .keyCommands ().keys (ByteBuffer .wrap ("key*" .getBytes ())).flatMapIterable (it -> it )) //
94
+ .expectNextCount (3 ).verifyComplete ();
90
95
}
91
96
92
97
@ Test // DATAREDIS-525
@@ -96,35 +101,39 @@ public void randomKeyShouldReturnAnyKey() {
96
101
nativeCommands .set (KEY_2 , VALUE_2 );
97
102
nativeCommands .set (KEY_3 , VALUE_3 );
98
103
99
- assertThat (connection .keyCommands ().randomKey (). block (), is ( notNullValue ()) );
104
+ StepVerifier . create (connection .keyCommands ().randomKey ()). expectNextCount ( 1 ). verifyComplete ( );
100
105
}
101
106
102
107
@ Test // DATAREDIS-525
103
108
public void randomKeyShouldReturnNullWhenNoKeyExists () {
104
- assertThat (connection .keyCommands ().randomKey (). block (), is ( nullValue ()) );
109
+ StepVerifier . create (connection .keyCommands ().randomKey ()). verifyComplete ( );
105
110
}
106
111
107
112
@ Test // DATAREDIS-525
108
113
public void renameShouldAlterKeyNameCorrectly () {
109
114
110
115
nativeCommands .set (KEY_1 , VALUE_2 );
111
116
112
- assertThat (connection .keyCommands ().rename (KEY_1_BBUFFER , KEY_2_BBUFFER ).block (), is (true ));
117
+ StepVerifier .create (connection .keyCommands ().rename (KEY_1_BBUFFER , KEY_2_BBUFFER )).expectNext (true )
118
+ .verifyComplete ();
113
119
assertThat (nativeCommands .exists (KEY_2 ), is (1L ));
114
120
assertThat (nativeCommands .exists (KEY_1 ), is (0L ));
115
121
}
116
122
117
- @ Test ( expected = RedisSystemException . class ) // DATAREDIS-525
123
+ @ Test // DATAREDIS-525
118
124
public void renameShouldThrowErrorWhenKeyDoesNotExist () {
119
- assertThat (connection .keyCommands ().rename (KEY_1_BBUFFER , KEY_2_BBUFFER ).block (), is (true ));
125
+
126
+ StepVerifier .create (connection .keyCommands ().rename (KEY_1_BBUFFER , KEY_2_BBUFFER ))
127
+ .expectError (RedisSystemException .class ).verify ();
120
128
}
121
129
122
130
@ Test // DATAREDIS-525
123
131
public void renameNXShouldAlterKeyNameCorrectly () {
124
132
125
133
nativeCommands .set (KEY_1 , VALUE_2 );
126
134
127
- assertThat (connection .keyCommands ().rename (KEY_1_BBUFFER , KEY_2_BBUFFER ).block (), is (true ));
135
+ StepVerifier .create (connection .keyCommands ().renameNX (KEY_1_BBUFFER , KEY_2_BBUFFER )).expectNext (true )
136
+ .verifyComplete ();
128
137
129
138
assertThat (nativeCommands .exists (KEY_2 ), is (1L ));
130
139
assertThat (nativeCommands .exists (KEY_1 ), is (0L ));
@@ -136,16 +145,16 @@ public void renameNXShouldNotAlterExistingKeyName() {
136
145
nativeCommands .set (KEY_1 , VALUE_2 );
137
146
nativeCommands .set (KEY_2 , VALUE_2 );
138
147
139
- assertThat (connection .keyCommands ().renameNX (KEY_1_BBUFFER , KEY_2_BBUFFER ).block (), is (false ));
148
+ StepVerifier .create (connection .keyCommands ().renameNX (KEY_1_BBUFFER , KEY_2_BBUFFER )).expectNext (false )
149
+ .verifyComplete ();
140
150
}
141
151
142
152
@ Test // DATAREDIS-525
143
153
public void shouldDeleteKeyCorrectly () {
144
154
145
155
nativeCommands .set (KEY_1 , VALUE_1 );
146
156
147
- Mono <Long > result = connection .keyCommands ().del (KEY_1_BBUFFER );
148
- assertThat (result .block (), is (1L ));
157
+ StepVerifier .create (connection .keyCommands ().del (KEY_1_BBUFFER )).expectNext (1L ).verifyComplete ();
149
158
}
150
159
151
160
@ Test // DATAREDIS-525
@@ -168,7 +177,7 @@ public void shouldDeleteKeysInBatchCorrectly() {
168
177
169
178
Mono <Long > result = connection .keyCommands ().mDel (Arrays .asList (KEY_1_BBUFFER , KEY_2_BBUFFER ));
170
179
171
- assertThat (result . block (), is ( 2L ));
180
+ StepVerifier . create (result ). expectNext ( 2L ). verifyComplete ( );
172
181
}
173
182
174
183
@ Test // DATAREDIS-525
@@ -177,9 +186,11 @@ public void shouldDeleteKeysInMultipleBatchesCorrectly() {
177
186
nativeCommands .set (KEY_1 , VALUE_1 );
178
187
nativeCommands .set (KEY_2 , VALUE_2 );
179
188
189
+ Flux <List <ByteBuffer >> input = Flux .just (Arrays .asList (KEY_1_BBUFFER , KEY_2_BBUFFER ),
190
+ Collections .singletonList (KEY_1_BBUFFER ));
191
+
180
192
Flux <Long > result = connection .keyCommands ()
181
- .mDel (
182
- Flux .fromIterable (Arrays .asList (Arrays .asList (KEY_1_BBUFFER , KEY_2_BBUFFER ), Arrays .asList (KEY_1_BBUFFER ))))
193
+ .mDel (input )
183
194
.map (NumericResponse ::getOutput );
184
195
185
196
StepVerifier .create (result ).expectNextCount (2 ).verifyComplete ();
@@ -216,7 +227,7 @@ public void shouldUnlinkKeysInBatchCorrectly() {
216
227
217
228
Mono <Long > result = connection .keyCommands ().mUnlink (Arrays .asList (KEY_1_BBUFFER , KEY_2_BBUFFER ));
218
229
219
- assertThat (result . block (), is ( 2L ));
230
+ StepVerifier . create (result ). expectNext ( 2L ). verifyComplete ( );
220
231
}
221
232
222
233
@ Test // DATAREDIS-693
@@ -226,9 +237,11 @@ public void shouldUnlinkKeysInMultipleBatchesCorrectly() {
226
237
nativeCommands .set (KEY_1 , VALUE_1 );
227
238
nativeCommands .set (KEY_2 , VALUE_2 );
228
239
240
+ Flux <List <ByteBuffer >> input = Flux .just (Arrays .asList (KEY_1_BBUFFER , KEY_2_BBUFFER ),
241
+ Collections .singletonList (KEY_1_BBUFFER ));
242
+
229
243
Flux <Long > result = connection .keyCommands ()
230
- .mUnlink (Flux .fromIterable (
231
- Arrays .asList (Arrays .asList (KEY_1_BBUFFER , KEY_2_BBUFFER ), Collections .singletonList (KEY_1_BBUFFER ))))
244
+ .mUnlink (input )
232
245
.map (NumericResponse ::getOutput );
233
246
234
247
StepVerifier .create (result ).expectNextCount (2 ).verifyComplete ();
@@ -357,7 +370,7 @@ public void touchReturnsNrOfKeysTouched() {
357
370
@ Test // DATAREDIS-694
358
371
public void touchReturnsZeroIfNoKeysTouched () {
359
372
360
- StepVerifier .create (connection .keyCommands ().touch (Arrays . asList (KEY_1_BBUFFER ))) //
373
+ StepVerifier .create (connection .keyCommands ().touch (Collections . singletonList (KEY_1_BBUFFER ))) //
361
374
.expectNext (0L ) //
362
375
.verifyComplete ();
363
376
}
0 commit comments