IntColumnTest.java

package edu.washington.escience.myria.column;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.nio.BufferOverflowException;

import org.junit.Test;

import edu.washington.escience.myria.Type;
import edu.washington.escience.myria.column.builder.IntColumnBuilder;
import edu.washington.escience.myria.proto.DataProto.ColumnMessage;
import edu.washington.escience.myria.storage.TupleBatch;
import edu.washington.escience.myria.storage.TupleUtils;

public class IntColumnTest {

  @Test
  public void testProto() {
    final IntColumnBuilder original = new IntColumnBuilder();
    original.appendInt(1).appendInt(2).appendInt(5).appendInt(11);
    final ColumnMessage serialized = original.build().serializeToProto();
    assertEquals(ColumnMessage.Type.INT, serialized.getType());
    final IntColumn deserialized = IntColumnBuilder.buildFromProtobuf(serialized, original.size());
    assertEquals(original.build().toString(), deserialized.toString());
  }

  @Test
  public void testIntProtoColumn() {
    final IntColumnBuilder original = new IntColumnBuilder();
    original.appendInt(1).appendInt(2).appendInt(5).appendInt(11).appendInt(17);
    final ColumnMessage serialized = original.build().serializeToProto();
    assertEquals(ColumnMessage.Type.INT, serialized.getType());
    final IntProtoColumn deserialized = new IntProtoColumn(serialized.getIntColumn());
    assertTrue(original.build().toString().equals(deserialized.toString()));
  }

  @Test
  public void testFull() {
    final IntColumnBuilder builder = new IntColumnBuilder();
    for (int i = 0; i < TupleUtils.getBatchSize(Type.INT_TYPE); i++) {
      builder.appendInt(i);
    }
    builder.build();
  }

  @Test(expected = BufferOverflowException.class)
  public void testOverflow() {
    final IntColumnBuilder builder = new IntColumnBuilder();
    for (int i = 0; i < TupleUtils.getBatchSize(Type.INT_TYPE); i++) {
      builder.appendInt(i);
    }
    builder.appendInt(TupleUtils.getBatchSize(Type.INT_TYPE));
    builder.build();
  }
}