TestZkPartitionProperties.java

package com.liveramp.hank.coordinator.zk;

import com.liveramp.hank.test.ZkTestCase;
import com.liveramp.hank.zookeeper.ZkPath;
import org.junit.Test;

import static org.junit.Assert.assertEquals;

public class TestZkPartitionProperties extends ZkTestCase {
  @Test
  public void testCreate() throws Exception {
    ZkPartitionProperties pi = ZkPartitionProperties.create(getZk(), getRoot(), 1, 15000, 550);
    assertEquals(1, pi.getPartitionNumber());
    assertEquals(15000, pi.getNumBytes());
    assertEquals(550, pi.getNumRecords());

    // should not throw an exception
    pi = ZkPartitionProperties.create(getZk(), getRoot(), 1, 15000, 550);
    assertEquals(1, pi.getPartitionNumber());
    assertEquals(15000, pi.getNumBytes());
    assertEquals(550, pi.getNumRecords());
  }

  @Test
  public void testLoad() throws Exception {
    ZkPartitionProperties.create(getZk(), getRoot(), 1, 15000, 550);

    ZkPartitionProperties pi = new ZkPartitionProperties(getZk(), ZkPath.append(getRoot(), "part-1"));
    assertEquals(1, pi.getPartitionNumber());
    assertEquals(15000, pi.getNumBytes());
    assertEquals(550, pi.getNumRecords());
  }
}