96 lines
2.6 KiB
Java
96 lines
2.6 KiB
Java
package redis.clients.jedis.tests.commands;
|
|
|
|
import java.util.Arrays;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
import java.util.Set;
|
|
|
|
import org.junit.After;
|
|
import org.junit.Before;
|
|
import org.junit.ComparisonFailure;
|
|
|
|
import redis.clients.jedis.Jedis;
|
|
import redis.clients.jedis.tests.HostAndPortUtil;
|
|
import redis.clients.jedis.tests.JedisTestBase;
|
|
import redis.clients.jedis.tests.HostAndPortUtil.HostAndPort;
|
|
|
|
public abstract class JedisCommandTestBase extends JedisTestBase {
|
|
protected static HostAndPort hnp = HostAndPortUtil.getRedisServers().get(0);
|
|
|
|
protected Jedis jedis;
|
|
|
|
public JedisCommandTestBase() {
|
|
super();
|
|
}
|
|
|
|
@Before
|
|
public void setUp() throws Exception {
|
|
jedis = new Jedis(hnp.host, hnp.port, 500);
|
|
jedis.connect();
|
|
jedis.auth("foobared");
|
|
jedis.configSet("timeout", "300");
|
|
jedis.flushAll();
|
|
}
|
|
|
|
@After
|
|
public void tearDown() {
|
|
jedis.disconnect();
|
|
}
|
|
|
|
protected Jedis createJedis() {
|
|
Jedis j = new Jedis(hnp.host, hnp.port);
|
|
j.connect();
|
|
j.auth("foobared");
|
|
j.flushAll();
|
|
return j;
|
|
}
|
|
|
|
protected void assertEquals(List<byte[]> expected, List<byte[]> actual) {
|
|
assertEquals(expected.size(), actual.size());
|
|
for (int n = 0; n < expected.size(); n++) {
|
|
assertArrayEquals(expected.get(n), actual.get(n));
|
|
}
|
|
}
|
|
|
|
protected void assertEquals(Set<byte[]> expected, Set<byte[]> actual) {
|
|
assertEquals(expected.size(), actual.size());
|
|
Iterator<byte[]> e = expected.iterator();
|
|
while (e.hasNext()) {
|
|
byte[] next = e.next();
|
|
boolean contained = false;
|
|
for (byte[] element : expected) {
|
|
if (Arrays.equals(next, element)) {
|
|
contained = true;
|
|
}
|
|
}
|
|
if (!contained) {
|
|
throw new ComparisonFailure("element is missing", next
|
|
.toString(), actual.toString());
|
|
}
|
|
}
|
|
}
|
|
|
|
protected boolean arrayContains(List<byte[]> array, byte[] expected) {
|
|
for (byte[] a : array) {
|
|
try {
|
|
assertArrayEquals(a, expected);
|
|
return true;
|
|
} catch (AssertionError e) {
|
|
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
protected boolean setContains(Set<byte[]> set, byte[] expected) {
|
|
for (byte[] a : set) {
|
|
try {
|
|
assertArrayEquals(a, expected);
|
|
return true;
|
|
} catch (AssertionError e) {
|
|
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
} |