accord-core/src/test/java/accord/local/CommandStoreTest.java

package accord.local;

import accord.Utils;
import accord.api.KeyRange;
import accord.impl.IntKey;
import accord.impl.TopologyUtils;
import accord.topology.KeyRanges;
import accord.topology.Shard;
import accord.topology.Topology;
import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;

import java.util.List;

import static accord.impl.IntKey.key;

public class CommandStoreTest
{
    @Test
    void topologyChangeTest()
    {
        List<Node.Id> ids = Utils.ids(5);
        KeyRanges ranges = TopologyUtils.initialRanges(5, 500);
        Topology topology = TopologyUtils.initialTopology(ids, ranges, 3);
        Topology local = topology.forNode(ids.get(0));

        KeyRanges shards = CommandStores.shardRanges(local.ranges(), 10).get(0);
        Assertions.assertEquals(ranges(r(0, 10), r(300, 310), r(400, 410)), shards);

        CommandStore commandStore = new CommandStore.Synchronized(0, ids.get(0), null, null, null);
        commandStore.updateTopology(topology, shards.union(r(350, 360)), KeyRanges.EMPTY);
        commandStore.commandsForKey(key(355));
        commandStore.commandsForKey(key(356));
        commandStore.commandsForKey(key(357));

        Assertions.assertTrue(commandStore.hasCommandsForKey(key(355)));
        Assertions.assertTrue(commandStore.hasCommandsForKey(key(356)));
        Assertions.assertTrue(commandStore.hasCommandsForKey(key(357)));

        commandStore.updateTopology(topology, KeyRanges.EMPTY, ranges(r(355, 360)));

        Assertions.assertTrue(commandStore.hasCommandsForKey(key(355)));
        Assertions.assertFalse(commandStore.hasCommandsForKey(key(356)));
        Assertions.assertFalse(commandStore.hasCommandsForKey(key(357)));
    }

    private static Shard[] shards(Topology topology, int... indexes)
    {
        Shard[] shards = new Shard[indexes.length];

        for (int i=0; i<indexes.length; i++)
            shards[i] = topology.get(indexes[i]);
        return shards;
    }

    private static void assertMapping(KeyRanges ranges, Shard[] shards, CommandStore.RangeMapping mapping)
    {
        Assertions.assertEquals(ranges, mapping.ranges);
        Assertions.assertArrayEquals(shards, mapping.shards);
    }

    private static KeyRanges ranges(KeyRange... ranges)
    {
        return new KeyRanges(ranges);
    }

    private static KeyRange<IntKey> r(int start, int end)
    {
        return IntKey.range(start, end);
    }

    @Test
    void mapRangesTest()
    {
        List<Node.Id> ids = Utils.ids(5);
        KeyRanges ranges = TopologyUtils.initialRanges(5, 500);
        Topology topology = TopologyUtils.initialTopology(ids, ranges, 3);
        Topology local = topology.forNode(ids.get(0));

        KeyRanges shards = CommandStores.shardRanges(local.ranges(), 10).get(0);
        Assertions.assertEquals(ranges(r(0, 10), r(300, 310), r(400, 410)), shards);

        assertMapping(shards, shards(local, 0, 1, 2),
                      CommandStore.mapRanges(shards, local));
        assertMapping(ranges(r(0, 10), r(300, 310), r(390, 400), r(400, 410)), shards(local, 0, 1, 1, 2),
                      CommandStore.mapRanges(ranges(r(0, 10), r(300, 310), r(390, 410)), local));

        assertMapping(ranges(r(0, 10), r(300, 310), r(350, 360), r(400, 410)), shards(local, 0, 1, 1, 2),
                      CommandStore.mapRanges(ranges(r(0, 10), r(300, 310), r(350, 360), r(400, 410)), local));

    }
}