EnumWithPartitionTest.java

package org.yamcs.yarch;

import static org.junit.Assert.*;

import java.nio.ByteBuffer;
import java.util.Arrays;
import java.util.List;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import org.junit.runners.Parameterized.Parameter;
import org.junit.runners.Parameterized.Parameters;
@RunWith(Parameterized.class)
public class EnumWithPartitionTest extends YarchTestCase {
    int n=10;
    
    @Parameter
    public String partitionStorage; 
    @Parameters
    public static Iterable<String> data() {
        return Arrays.asList("IN_KEY", "COLUMN_FAMILY");
    }
    
    private void populate(String tblname) throws Exception {
        ydb.execute("create table "+tblname+"(gentime timestamp, packetName enum, packet binary, primary key(gentime,packetName)) partition by time_and_value(gentime, packetName)  partition_storage="+partitionStorage);
        ydb.execute("create stream "+tblname+"_in(gentime timestamp, packetName enum, packet binary)");
        ydb.execute("insert into "+tblname+" select * from "+tblname+"_in");

        Stream s = ydb.getStream(tblname+"_in");
        TupleDefinition td=new TupleDefinition();
        td.addColumn(new ColumnDefinition("gentime", DataType.TIMESTAMP));
        td.addColumn(new ColumnDefinition("packetName", DataType.ENUM));
        td.addColumn(new ColumnDefinition("packet", DataType.BINARY));

        for (int i=0;i<n;i++) {
            ByteBuffer bb=ByteBuffer.allocate(2000);
            while(bb.remaining()>0) bb.putInt(i);
            Tuple t=new Tuple(td, new Object[]{
                    1000L*i, "pn"+(i%10), bb.array()
            });
            s.emitTuple(t);
        }
        execute("close stream "+tblname+"_in");
    }
    
    @Test
    public void test1() throws Exception {
        populate("testenum");
        ydb.execute("create stream testenum_out as select * from testenum");
        final List<Tuple> tuples= fetchAll("testenum_out");
    
        for(int i=0;i<n;i++) {
            Tuple t=tuples.get(i);
            assertEquals(i*1000l, (long)(Long)t.getColumn(0));
            assertEquals("pn"+(i%10), (String)t.getColumn(1));
        }
    }
    
    
    @Test
    public void test2() throws Exception {
        populate("testenum2");
        ydb.execute("create stream testenum2_out as select * from testenum2 where packetName in ('pn1', 'invalid')");
        List<Tuple> tuples= fetchAll("testenum2_out");
        int i = 1;
        for(Tuple t:tuples) {
            assertEquals(i*1000l, (long)(Long)t.getColumn(0));
            assertEquals("pn"+(i%10), (String)t.getColumn(1));
            i+=10;
        }
        assertEquals(n+1, i);
        
        ydb.execute("create stream testenum2_out2 as select * from testenum2 where packetName = 'pn1'");
        tuples = fetchAll("testenum2_out2");
        i = 1;
        for(Tuple t:tuples) {
            assertEquals(i*1000l, (long)(Long)t.getColumn(0));
            assertEquals("pn"+(i%10), (String)t.getColumn(1));
            i+=10;
        }
        assertEquals(n+1, i);
    }
    
    @Test
    public void test3() throws Exception {
        populate("testenum3");
        ydb.execute("create stream testenum3_out as select * from testenum3 where packetName in ('invalid')");
        List<Tuple> tuples= fetchAll("testenum3_out");
        assertTrue(tuples.isEmpty());
        
        ydb.execute("create stream testenum3_out2 as select * from testenum3 where packetName = 'invalid'");
        tuples= fetchAll("testenum3_out2");
        assertTrue(tuples.isEmpty());
    }
    
    @Test
    public void test4() throws Exception {
        ydb.execute("create table testenum4(gentime timestamp, packetName enum, packet binary, primary key(gentime,packetName)) partition by time_and_value(gentime, packetName)  partition_storage="+partitionStorage);
        ydb.execute("create stream testenum4_out as select * from testenum4 where packetName in ('invalid')");
        List<Tuple> tuples= fetchAll("testenum4_out");
        assertTrue(tuples.isEmpty());
        
        ydb.execute("create stream testenum4_out2 as select * from testenum4 where packetName = 'invalid'");
        tuples= fetchAll("testenum4_out2");
        assertTrue(tuples.isEmpty());
    }
}