SortingCommandsTest.java

package redis.clients.jedis.commands.jedis;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static redis.clients.jedis.util.AssertUtil.assertByteArrayListEquals;

import java.util.ArrayList;
import java.util.List;

import io.redis.test.annotations.EnabledOnCommand;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.params.ParameterizedClass;
import org.junit.jupiter.params.provider.MethodSource;

import redis.clients.jedis.RedisProtocol;
import redis.clients.jedis.params.SortingParams;

@ParameterizedClass
@MethodSource("redis.clients.jedis.commands.CommandsTestsParameters#respVersions")
public class SortingCommandsTest extends JedisCommandsTestBase {
  final byte[] bfoo = { 0x01, 0x02, 0x03, 0x04 };
  final byte[] bfoodest = { 0x01, 0x02, 0x03, 0x04, 0x05 };
  final byte[] bbar1 = { 0x05, 0x06, 0x07, 0x08, '1' };
  final byte[] bbar2 = { 0x05, 0x06, 0x07, 0x08, '2' };
  final byte[] bbar3 = { 0x05, 0x06, 0x07, 0x08, '3' };
  final byte[] bbar10 = { 0x05, 0x06, 0x07, 0x08, '1', '0' };
  final byte[] bbarstar = { 0x05, 0x06, 0x07, 0x08, '*' };
  final byte[] bcar1 = { 0x0A, 0x0B, 0x0C, 0x0D, '1' };
  final byte[] bcar2 = { 0x0A, 0x0B, 0x0C, 0x0D, '2' };
  final byte[] bcar10 = { 0x0A, 0x0B, 0x0C, 0x0D, '1', '0' };
  final byte[] bcarstar = { 0x0A, 0x0B, 0x0C, 0x0D, '*' };
  final byte[] b1 = { '1' };
  final byte[] b2 = { '2' };
  final byte[] b3 = { '3' };
  final byte[] b10 = { '1', '0' };

  public SortingCommandsTest(RedisProtocol protocol) {
    super(protocol);
  }

  @Test
  public void sort() {
    jedis.lpush("foo", "3");
    jedis.lpush("foo", "2");
    jedis.lpush("foo", "1");

    List<String> result = jedis.sort("foo");

    List<String> expected = new ArrayList<String>();
    expected.add("1");
    expected.add("2");
    expected.add("3");

    assertEquals(expected, result);

    // Binary
    jedis.lpush(bfoo, b3);
    jedis.lpush(bfoo, b2);
    jedis.lpush(bfoo, b1);

    List<byte[]> bresult = jedis.sort(bfoo);

    List<byte[]> bexpected = new ArrayList<byte[]>();
    bexpected.add(b1);
    bexpected.add(b2);
    bexpected.add(b3);

    assertByteArrayListEquals(bexpected, bresult);
  }

  @Test
  public void sortBy() {
    jedis.lpush("foo", "2");
    jedis.lpush("foo", "3");
    jedis.lpush("foo", "1");

    jedis.set("bar1", "3");
    jedis.set("bar2", "2");
    jedis.set("bar3", "1");

    SortingParams sp = new SortingParams();
    sp.by("bar*");

    List<String> result = jedis.sort("foo", sp);

    List<String> expected = new ArrayList<String>();
    expected.add("3");
    expected.add("2");
    expected.add("1");

    assertEquals(expected, result);
    
    // Sort to dest key
    long resultCount = jedis.sort("foo", sp, "foodest");
    assertEquals(3L, resultCount);

    result = jedis.lpop("foodest", 5);
    assertEquals(expected, result);
    
    // Binary
    jedis.lpush(bfoo, b2);
    jedis.lpush(bfoo, b3);
    jedis.lpush(bfoo, b1);

    jedis.set(bbar1, b3);
    jedis.set(bbar2, b2);
    jedis.set(bbar3, b1);

    SortingParams bsp = new SortingParams();
    bsp.by(bbarstar);

    List<byte[]> bresult = jedis.sort(bfoo, bsp);

    List<byte[]> bexpected = new ArrayList<byte[]>();
    bexpected.add(b3);
    bexpected.add(b2);
    bexpected.add(b1);

    assertByteArrayListEquals(bexpected, bresult);
    
    // Sort to dest key
    resultCount = jedis.sort(bfoo, sp, bfoodest);
    assertEquals(3L, resultCount);

    bresult = jedis.lpop(bfoodest, 5);
    assertByteArrayListEquals(bexpected, bresult);
  }

  @Test
  public void sortDesc() {
    jedis.lpush("foo", "3");
    jedis.lpush("foo", "2");
    jedis.lpush("foo", "1");

    SortingParams sp = new SortingParams();
    sp.desc();

    List<String> result = jedis.sort("foo", sp);

    List<String> expected = new ArrayList<String>();
    expected.add("3");
    expected.add("2");
    expected.add("1");

    assertEquals(expected, result);

    // Binary
    jedis.lpush(bfoo, b3);
    jedis.lpush(bfoo, b2);
    jedis.lpush(bfoo, b1);

    SortingParams bsp = new SortingParams();
    bsp.desc();

    List<byte[]> bresult = jedis.sort(bfoo, bsp);

    List<byte[]> bexpected = new ArrayList<byte[]>();
    bexpected.add(b3);
    bexpected.add(b2);
    bexpected.add(b1);

    assertByteArrayListEquals(bexpected, bresult);
  }

  @Test
  public void sortLimit() {
    for (int n = 10; n > 0; n--) {
      jedis.lpush("foo", String.valueOf(n));
    }

    SortingParams sp = new SortingParams();
    sp.limit(0, 3);

    List<String> result = jedis.sort("foo", sp);

    List<String> expected = new ArrayList<String>();
    expected.add("1");
    expected.add("2");
    expected.add("3");

    assertEquals(expected, result);

    // Binary
    jedis.rpush(bfoo, new byte[] { (byte) '4' });
    jedis.rpush(bfoo, new byte[] { (byte) '3' });
    jedis.rpush(bfoo, new byte[] { (byte) '2' });
    jedis.rpush(bfoo, new byte[] { (byte) '1' });

    SortingParams bsp = new SortingParams();
    bsp.limit(0, 3);

    List<byte[]> bresult = jedis.sort(bfoo, bsp);

    List<byte[]> bexpected = new ArrayList<byte[]>();
    bexpected.add(b1);
    bexpected.add(b2);
    bexpected.add(b3);

    assertByteArrayListEquals(bexpected, bresult);
  }

  @Test
  public void sortAlpha() {
    jedis.lpush("foo", "1");
    jedis.lpush("foo", "2");
    jedis.lpush("foo", "10");

    SortingParams sp = new SortingParams();
    sp.alpha();

    List<String> result = jedis.sort("foo", sp);

    List<String> expected = new ArrayList<String>();
    expected.add("1");
    expected.add("10");
    expected.add("2");

    assertEquals(expected, result);

    // Binary
    jedis.lpush(bfoo, b1);
    jedis.lpush(bfoo, b2);
    jedis.lpush(bfoo, b10);

    SortingParams bsp = new SortingParams();
    bsp.alpha();

    List<byte[]> bresult = jedis.sort(bfoo, bsp);

    List<byte[]> bexpected = new ArrayList<byte[]>();
    bexpected.add(b1);
    bexpected.add(b10);
    bexpected.add(b2);

    assertByteArrayListEquals(bexpected, bresult);
  }

  @Test
  public void sortGet() {
    jedis.lpush("foo", "1");
    jedis.lpush("foo", "2");
    jedis.lpush("foo", "10");

    jedis.set("bar1", "bar1");
    jedis.set("bar2", "bar2");
    jedis.set("bar10", "bar10");

    jedis.set("car1", "car1");
    jedis.set("car2", "car2");
    jedis.set("car10", "car10");

    SortingParams sp = new SortingParams();
    sp.get("car*", "bar*");

    List<String> result = jedis.sort("foo", sp);

    List<String> expected = new ArrayList<String>();
    expected.add("car1");
    expected.add("bar1");
    expected.add("car2");
    expected.add("bar2");
    expected.add("car10");
    expected.add("bar10");

    assertEquals(expected, result);

    // Binary
    jedis.lpush(bfoo, b1);
    jedis.lpush(bfoo, b2);
    jedis.lpush(bfoo, b10);

    jedis.set(bbar1, bbar1);
    jedis.set(bbar2, bbar2);
    jedis.set(bbar10, bbar10);

    jedis.set(bcar1, bcar1);
    jedis.set(bcar2, bcar2);
    jedis.set(bcar10, bcar10);

    SortingParams bsp = new SortingParams();
    bsp.get(bcarstar, bbarstar);

    List<byte[]> bresult = jedis.sort(bfoo, bsp);

    List<byte[]> bexpected = new ArrayList<byte[]>();
    bexpected.add(bcar1);
    bexpected.add(bbar1);
    bexpected.add(bcar2);
    bexpected.add(bbar2);
    bexpected.add(bcar10);
    bexpected.add(bbar10);

    assertByteArrayListEquals(bexpected, bresult);
  }

  @Test
  public void sortStore() {
    jedis.lpush("foo", "1");
    jedis.lpush("foo", "2");
    jedis.lpush("foo", "10");

    long result = jedis.sort("foo", "result");

    List<String> expected = new ArrayList<String>();
    expected.add("1");
    expected.add("2");
    expected.add("10");

    assertEquals(3, result);
    assertEquals(expected, jedis.lrange("result", 0, 1000));

    // Binary
    jedis.lpush(bfoo, b1);
    jedis.lpush(bfoo, b2);
    jedis.lpush(bfoo, b10);

    byte[] bkresult = new byte[] { 0X09, 0x0A, 0x0B, 0x0C };
    long bresult = jedis.sort(bfoo, bkresult);

    List<byte[]> bexpected = new ArrayList<byte[]>();
    bexpected.add(b1);
    bexpected.add(b2);
    bexpected.add(b10);

    assertEquals(3, bresult);
    assertByteArrayListEquals(bexpected, jedis.lrange(bkresult, 0, 1000));
  }

  @Test
  @EnabledOnCommand("SORT_RO")
  public void sort_ro() {
    jedis.rpush("foo", "1", "3", "2");

    List<String> result = jedis.sortReadonly("foo", new SortingParams().desc());
    List<String> expected = new ArrayList<>();
    expected.add("3");
    expected.add("2");
    expected.add("1");

    assertEquals(expected, result);

    // Binary
    jedis.rpush(bfoo, b3, b1, b2);

    List<byte[]> bresult = jedis.sortReadonly(bfoo, new SortingParams().alpha());
    List<byte[]> bexpected = new ArrayList<>();
    bexpected.add(b1);
    bexpected.add(b2);
    bexpected.add(b3);

    assertByteArrayListEquals(bexpected, bresult);
  }
}