TableSlotMapTest.java

package com.taobao.tddl.rule.virtualnode;

import java.util.Map;

import org.junit.Assert;
import org.junit.Test;

import com.google.common.collect.Maps;
import com.taobao.tddl.common.exception.TddlException;

public class TableSlotMapTest {

    @Test
    public void testSimple() {
        TableSlotMap slot = new TableSlotMap();
        Map<String, String> map = Maps.newHashMap();
        map.put("0", "0,1-256");
        map.put("1", "257-512");
        map.put("2", "513");
        map.put("3", "514-718");
        map.put("4", "719-1024");
        map.put("5", "1025");
        slot.setTableSlotMap(map);
        slot.init();

        Assert.assertEquals("0", slot.getValue("0"));
        Assert.assertEquals("1", slot.getValue("257"));
        Assert.assertEquals("2", slot.getValue("513"));
        Assert.assertEquals("3", slot.getValue("514"));
        Assert.assertEquals("4", slot.getValue("719"));
        Assert.assertEquals("5", slot.getValue("1025"));
        slot.init();
    }

    @Test
    public void testPartition() throws TddlException {
        TableSlotMap slot = new TableSlotMap();
        PartitionFunction keyFunc = new PartitionFunction();
        keyFunc.setFirstValue(-1);
        keyFunc.setPartitionCount("1,1,1,1,1,1");
        keyFunc.setPartitionLength("1,1,1,1,1,1");

        PartitionFunction valueFunc = new PartitionFunction();
        valueFunc.setFirstValue(0);
        valueFunc.setPartitionCount("1,1,1,1,1,1");
        valueFunc.setPartitionLength("257,256,1,205,306,1");

        slot.setKeyPartitionFunction(keyFunc);
        slot.setValuePartitionFunction(valueFunc);
        slot.init();

        Assert.assertEquals("0", slot.getValue("0"));
        Assert.assertEquals("1", slot.getValue("257"));
        Assert.assertEquals("2", slot.getValue("513"));
        Assert.assertEquals("3", slot.getValue("514"));
        Assert.assertEquals("4", slot.getValue("719"));
        Assert.assertEquals("5", slot.getValue("1025"));
        slot.destory();
    }
}