+ add zset tests
This commit is contained in:
Costin Leau
@@ -0,0 +1,390 @@
|
||||
/*
|
||||
* Copyright 2010 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.springframework.data.keyvalue.redis.util;
|
||||
|
||||
import static org.junit.Assert.*;
|
||||
import static org.junit.matchers.JUnitMatchers.*;
|
||||
|
||||
import java.util.Iterator;
|
||||
import java.util.NoSuchElementException;
|
||||
import java.util.Set;
|
||||
|
||||
import org.junit.Before;
|
||||
import org.junit.Test;
|
||||
import org.springframework.data.keyvalue.redis.core.BoundZSetOperations;
|
||||
import org.springframework.data.keyvalue.redis.core.RedisTemplate;
|
||||
|
||||
/**
|
||||
* Integration test for Redis ZSet.
|
||||
*
|
||||
* @author Costin Leau
|
||||
*/
|
||||
public abstract class AbstractRedisZSetTest<T> extends AbstractRedisCollectionTests<T> {
|
||||
|
||||
protected RedisZSet<T> zSet;
|
||||
|
||||
/**
|
||||
* Constructs a new <code>AbstractRedisZSetTest</code> instance.
|
||||
*
|
||||
* @param factory
|
||||
* @param template
|
||||
*/
|
||||
public AbstractRedisZSetTest(ObjectFactory<T> factory, RedisTemplate template) {
|
||||
super(factory, template);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@Before
|
||||
public void setUp() throws Exception {
|
||||
super.setUp();
|
||||
zSet = (RedisZSet<T>) collection;
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testAddWithScore() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
|
||||
zSet.add(t1, 3);
|
||||
zSet.add(t2, 4);
|
||||
zSet.add(t3, 5);
|
||||
|
||||
Iterator<T> iterator = zSet.iterator();
|
||||
assertEquals(t1, iterator.next());
|
||||
assertEquals(t2, iterator.next());
|
||||
assertEquals(t3, iterator.next());
|
||||
assertFalse(iterator.hasNext());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testAdd() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
|
||||
zSet.add(t1);
|
||||
zSet.add(t2);
|
||||
zSet.add(t3);
|
||||
|
||||
Double d = new Double("1");
|
||||
|
||||
assertEquals(d, zSet.score(t1));
|
||||
assertEquals(d, zSet.score(t2));
|
||||
assertEquals(d, zSet.score(t3));
|
||||
}
|
||||
|
||||
|
||||
@Test
|
||||
public void testFirst() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
|
||||
zSet.add(t1, 3);
|
||||
zSet.add(t2, 4);
|
||||
zSet.add(t3, 5);
|
||||
|
||||
assertEquals(3, zSet.size());
|
||||
assertEquals(t1, zSet.first());
|
||||
}
|
||||
|
||||
@Test(expected = NoSuchElementException.class)
|
||||
public void testFirstException() throws Exception {
|
||||
zSet.first();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testLast() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
|
||||
zSet.add(t1, 3);
|
||||
zSet.add(t2, 4);
|
||||
zSet.add(t3, 5);
|
||||
|
||||
assertEquals(3, zSet.size());
|
||||
assertEquals(t3, zSet.last());
|
||||
}
|
||||
|
||||
@Test(expected = NoSuchElementException.class)
|
||||
public void testLastException() throws Exception {
|
||||
zSet.last();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRank() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
|
||||
zSet.add(t1, 3);
|
||||
zSet.add(t2, 4);
|
||||
zSet.add(t3, 5);
|
||||
|
||||
assertEquals(Integer.valueOf(0), zSet.rank(t1));
|
||||
assertEquals(Integer.valueOf(1), zSet.rank(t2));
|
||||
assertEquals(Integer.valueOf(2), zSet.rank(t3));
|
||||
assertNull(zSet.rank(getT()));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testReverseRank() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
|
||||
zSet.add(t1, 3);
|
||||
zSet.add(t2, 4);
|
||||
zSet.add(t3, 5);
|
||||
|
||||
assertEquals(Integer.valueOf(0), zSet.reverseRank(t3));
|
||||
assertEquals(Integer.valueOf(1), zSet.reverseRank(t2));
|
||||
assertEquals(Integer.valueOf(2), zSet.reverseRank(t1));
|
||||
assertNull(zSet.rank(getT()));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testScore() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
|
||||
zSet.add(t1, 3);
|
||||
zSet.add(t2, 4);
|
||||
zSet.add(t3, 5);
|
||||
|
||||
assertNull(zSet.score(getT()));
|
||||
assertEquals(Double.valueOf(3), zSet.score(t1));
|
||||
assertEquals(Double.valueOf(4), zSet.score(t2));
|
||||
assertEquals(Double.valueOf(5), zSet.score(t3));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testDefaultScore() {
|
||||
assertEquals(1, zSet.getDefaultScore(), 0);
|
||||
}
|
||||
|
||||
private RedisZSet<T> createZSetFor(String key) {
|
||||
return new DefaultRedisZSet<T>((BoundZSetOperations<String, T>) zSet.getOperations().forZSet(key));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIntersectAndStore() {
|
||||
RedisZSet<T> interSet1 = createZSetFor("test:zset:inter1");
|
||||
RedisZSet<T> interSet2 = createZSetFor("test:zset:inter");
|
||||
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
T t4 = getT();
|
||||
|
||||
zSet.add(t1, 1);
|
||||
zSet.add(t2, 2);
|
||||
zSet.add(t3, 3);
|
||||
|
||||
interSet1.add(t2, 2);
|
||||
interSet1.add(t4, 3);
|
||||
interSet2.add(t2, 2);
|
||||
interSet2.add(t3, 3);
|
||||
|
||||
String resultName = "test:zset:inter:result:1";
|
||||
RedisZSet<T> inter = zSet.intersectAndStore(resultName, interSet1, interSet2);
|
||||
|
||||
assertEquals(1, inter.size());
|
||||
assertThat(inter, hasItem(t2));
|
||||
assertEquals(Double.valueOf(6), inter.score(t2));
|
||||
assertEquals(resultName, inter.getKey());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRange() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
|
||||
zSet.add(t1, 1);
|
||||
zSet.add(t2, 2);
|
||||
zSet.add(t3, 3);
|
||||
|
||||
Set<T> range = zSet.range(1, 2);
|
||||
assertEquals(2, range.size());
|
||||
Iterator<T> iterator = range.iterator();
|
||||
assertEquals(t2, iterator.next());
|
||||
assertEquals(t3, iterator.next());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testReverseRange() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
|
||||
zSet.add(t1, 1);
|
||||
zSet.add(t2, 2);
|
||||
zSet.add(t3, 3);
|
||||
|
||||
Set<T> range = zSet.reverseRange(1, 2);
|
||||
assertEquals(2, range.size());
|
||||
Iterator<T> iterator = range.iterator();
|
||||
assertEquals(t2, iterator.next());
|
||||
assertEquals(t1, iterator.next());
|
||||
}
|
||||
|
||||
public void testRangeByScore() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
|
||||
zSet.add(t1, 1);
|
||||
zSet.add(t2, 2);
|
||||
zSet.add(t3, 3);
|
||||
|
||||
Set<T> range = zSet.rangeByScore(1.5, 3.5);
|
||||
assertEquals(2, range.size());
|
||||
assertThat(range, hasItems(t2, t3));
|
||||
|
||||
Iterator<T> iterator = range.iterator();
|
||||
assertEquals(t2, iterator.next());
|
||||
assertEquals(t3, iterator.next());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRemove() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
T t4 = getT();
|
||||
|
||||
zSet.add(t1, 1);
|
||||
zSet.add(t2, 2);
|
||||
zSet.add(t3, 3);
|
||||
zSet.add(t4, 4);
|
||||
|
||||
zSet.remove(1, 2);
|
||||
|
||||
assertEquals(2, zSet.size());
|
||||
Iterator<T> iterator = zSet.iterator();
|
||||
assertEquals(t1, iterator.next());
|
||||
assertEquals(t4, iterator.next());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testRemoveByScore() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
T t4 = getT();
|
||||
|
||||
zSet.add(t1, 1);
|
||||
zSet.add(t2, 2);
|
||||
zSet.add(t3, 3);
|
||||
zSet.add(t4, 4);
|
||||
|
||||
zSet.removeByScore(1.5, 2.5);
|
||||
|
||||
assertEquals(3, zSet.size());
|
||||
Iterator<T> iterator = zSet.iterator();
|
||||
assertEquals(t1, iterator.next());
|
||||
assertEquals(t3, iterator.next());
|
||||
assertEquals(t4, iterator.next());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testUnionAndStore() {
|
||||
RedisZSet<T> unionSet1 = createZSetFor("test:zset:union1");
|
||||
RedisZSet<T> unionSet2 = createZSetFor("test:zset:union2");
|
||||
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
T t4 = getT();
|
||||
|
||||
zSet.add(t1, 1);
|
||||
zSet.add(t2, 2);
|
||||
|
||||
unionSet1.add(t2, 2);
|
||||
unionSet1.add(t4, 5);
|
||||
unionSet2.add(t3, 6);
|
||||
|
||||
String resultName = "test:zset:union:result:1";
|
||||
RedisZSet<T> union = zSet.unionAndStore(resultName, unionSet1, unionSet2);
|
||||
assertEquals(4, union.size());
|
||||
assertThat(union, hasItems(t1, t2, t3, t4));
|
||||
assertEquals(resultName, union.getKey());
|
||||
|
||||
assertEquals(Double.valueOf(1), union.score(t1));
|
||||
assertEquals(Double.valueOf(4), union.score(t2));
|
||||
assertEquals(Double.valueOf(6), union.score(t3));
|
||||
assertEquals(Double.valueOf(5), union.score(t4));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIterator() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
T t4 = getT();
|
||||
|
||||
zSet.add(t1, 1);
|
||||
zSet.add(t2, 2);
|
||||
zSet.add(t3, 3);
|
||||
zSet.add(t4, 4);
|
||||
|
||||
Iterator<T> iterator = collection.iterator();
|
||||
|
||||
assertEquals(t1, iterator.next());
|
||||
assertEquals(t2, iterator.next());
|
||||
assertEquals(t3, iterator.next());
|
||||
assertEquals(t4, iterator.next());
|
||||
assertFalse(iterator.hasNext());
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@Test
|
||||
public void testToArray() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
T t4 = getT();
|
||||
|
||||
zSet.add(t1, 1);
|
||||
zSet.add(t2, 2);
|
||||
zSet.add(t3, 3);
|
||||
zSet.add(t4, 4);
|
||||
|
||||
Object[] array = collection.toArray();
|
||||
assertArrayEquals(new Object[] { t1, t2, t3, t4 }, array);
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
@Test
|
||||
public void testToArrayWithGenerics() {
|
||||
T t1 = getT();
|
||||
T t2 = getT();
|
||||
T t3 = getT();
|
||||
T t4 = getT();
|
||||
|
||||
zSet.add(t1, 1);
|
||||
zSet.add(t2, 2);
|
||||
zSet.add(t3, 3);
|
||||
zSet.add(t4, 4);
|
||||
|
||||
Object[] array = collection.toArray(new Object[zSet.size()]);
|
||||
assertArrayEquals(new Object[] { t1, t2, t3, t4 }, array);
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,47 @@
|
||||
/*
|
||||
* Copyright 2010 the original author or authors.
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.springframework.data.keyvalue.redis.util;
|
||||
|
||||
import org.springframework.data.keyvalue.redis.core.RedisTemplate;
|
||||
|
||||
/**
|
||||
* Parameterized instance of Redis sorted set tests.
|
||||
*
|
||||
* @author Costin Leau
|
||||
*/
|
||||
public class RedisZSetTest extends AbstractRedisZSetTest<Object> {
|
||||
|
||||
/**
|
||||
* Constructs a new <code>RedisZSetTest</code> instance.
|
||||
*
|
||||
* @param factory
|
||||
* @param template
|
||||
*/
|
||||
public RedisZSetTest(ObjectFactory<Object> factory, RedisTemplate template) {
|
||||
super(factory, template);
|
||||
}
|
||||
|
||||
@Override
|
||||
RedisStore<Object> copyStore(RedisStore<Object> store) {
|
||||
return new DefaultRedisZSet(store.getKey().toString(), store.getOperations());
|
||||
}
|
||||
|
||||
@Override
|
||||
AbstractRedisCollection<Object> createCollection() {
|
||||
String redisName = getClass().getName();
|
||||
return new DefaultRedisZSet(redisName, template);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user