Skip to content

Commit 56cd409

Browse files
committed
[cleanup] Fix TransactionCommandsTest mocked test
1 parent f475177 commit 56cd409

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/test/java/redis/clients/jedis/commands/jedis/TransactionCommandsTest.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,11 +30,13 @@
3030

3131
import redis.clients.jedis.Jedis;
3232
import redis.clients.jedis.Protocol;
33+
import redis.clients.jedis.PushConsumer;
3334
import redis.clients.jedis.RedisProtocol;
3435
import redis.clients.jedis.Response;
3536
import redis.clients.jedis.Transaction;
3637
import redis.clients.jedis.exceptions.JedisConnectionException;
3738
import redis.clients.jedis.exceptions.JedisDataException;
39+
import redis.clients.jedis.util.RedisInputStream;
3840
import redis.clients.jedis.util.SafeEncoder;
3941

4042
@ParameterizedClass
@@ -176,7 +178,7 @@ public void discardFail() {
176178
trans.set("b", "b");
177179

178180
try (MockedStatic<Protocol> protocol = Mockito.mockStatic(Protocol.class)) {
179-
protocol.when(() -> Protocol.read(any())).thenThrow(JedisConnectionException.class);
181+
protocol.when(() -> Protocol.read(any(RedisInputStream.class), any(PushConsumer.class))).thenThrow(JedisConnectionException.class);
180182

181183
trans.discard();
182184
fail("Should get mocked JedisConnectionException.");
@@ -196,7 +198,7 @@ public void execFail() {
196198
trans.set("b", "b");
197199

198200
try (MockedStatic<Protocol> protocol = Mockito.mockStatic(Protocol.class)) {
199-
protocol.when(() -> Protocol.read(any())).thenThrow(JedisConnectionException.class);
201+
protocol.when(() -> Protocol.read(any(RedisInputStream.class), any(PushConsumer.class))).thenThrow(JedisConnectionException.class);
200202

201203
trans.exec();
202204
fail("Should get mocked JedisConnectionException.");

0 commit comments

Comments
 (0)