Mock a Jedis transaction with mockito

TransactionMockFuckTest.java

import redis.clients.jedis.Client;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.Transaction;

import org.testng.annotations.*;
import static org.mockito.Mockito.*;

public class TransactionMockFuckTest {
        /**
         * Let's say we want to test the following method
         * public void run(Jedis jedis){
         *       Transaction t = jedis.multi();
         *       t.set("mykey", "myvalue");
         *       t.expire("mykey", 100);
         *       t.exec();
         * }
         */
        @Test
        public void testJedisTransaction(){
                String expectedKey = "mykey";
                String expectedValue = "myvalue";
        
                Jedis mockJedis = mock(Jedis.class);
                Client mockClient = mock(Client.class);
                Transaction spyTrans = spy(new Transaction(mockClient));
                when(mockJedis.multi()).thenReturn(spyTrans);
        
                doNothing().when(mockClient).set(anyString(), anyString());
        
                MyClassToTest mc = new MyClassToTest();
                mc.run(mockJedis);
        
                verify(mockClient, times(1)).set(expectedKey, expectedValue);
                verify(mockClient, times(1)).expire(expectedKey, 100);
        }
}