How to use

TestRegionSizeCalculator

in

org.apache.hadoop.hbase.mapreduce

/**
 * When size of region in megabytes is larger than largest possible integer there could be
 * error caused by lost of precision.
 * */
@Test
public void testLargeRegion() throws Exception {
 RegionLocator regionLocator = mockRegionLocator("largeRegion");
 Admin admin = mockAdmin(
   mockRegion("largeRegion", Integer.MAX_VALUE)
 );
 RegionSizeCalculator calculator = new RegionSizeCalculator(regionLocator, admin);
 assertEquals(((long) Integer.MAX_VALUE) * megabyte, calculator.getRegionSize("largeRegion".getBytes()));
}

@Test
public void testSimpleTestCase() throws Exception {
 RegionLocator regionLocator = mockRegionLocator("region1", "region2", "region3");
 Admin admin = mockAdmin(
   mockRegion("region1", 123),
   mockRegion("region3", 1232),
   mockRegion("region2",  54321)
 );
 RegionSizeCalculator calculator = new RegionSizeCalculator(regionLocator, admin);
 assertEquals(123 * megabyte, calculator.getRegionSize("region1".getBytes()));
 assertEquals(54321 * megabyte, calculator.getRegionSize("region2".getBytes()));
 assertEquals(1232 * megabyte, calculator.getRegionSize("region3".getBytes()));
 // if regionCalculator does not know about a region, it should return 0
 assertEquals(0 * megabyte, calculator.getRegionSize("otherTableRegion".getBytes()));
 assertEquals(3, calculator.getRegionSizeMap().size());
}

/** When calculator is disabled, it should return 0 for each request.*/
@Test
public void testDisabled() throws Exception {
 String regionName = "cz.goout:/index.html";
 RegionLocator table = mockRegionLocator(regionName);
 Admin admin = mockAdmin(
   mockRegion(regionName, 999)
 );
 //first request on enabled calculator
 RegionSizeCalculator calculator = new RegionSizeCalculator(table, admin);
 assertEquals(999 * megabyte, calculator.getRegionSize(regionName.getBytes()));
 //then disabled calculator.
 configuration.setBoolean(RegionSizeCalculator.ENABLE_REGIONSIZECALCULATOR, false);
 RegionSizeCalculator disabledCalculator = new RegionSizeCalculator(table, admin);
 assertEquals(0 * megabyte, disabledCalculator.getRegionSize(regionName.getBytes()));
 assertEquals(0, disabledCalculator.getRegionSizeMap().size());
}

/**
 * When size of region in megabytes is larger than largest possible integer there could be
 * error caused by lost of precision.
 * */
@Test
public void testLargeRegion() throws Exception {
 RegionLocator regionLocator = mockRegionLocator("largeRegion");
 Admin admin = mockAdmin(
   mockRegion("largeRegion", Integer.MAX_VALUE)
 );
 RegionSizeCalculator calculator = new RegionSizeCalculator(regionLocator, admin);
 assertEquals(((long) Integer.MAX_VALUE) * megabyte, calculator.getRegionSize("largeRegion".getBytes()));
}

/**
 * When size of region in megabytes is larger than largest possible integer there could be
 * error caused by lost of precision.
 * */
@Test
public void testLargeRegion() throws Exception {
 RegionLocator regionLocator = mockRegionLocator("largeRegion");
 Admin admin = mockAdmin(
   mockRegion("largeRegion", Integer.MAX_VALUE)
 );
 RegionSizeCalculator calculator = new RegionSizeCalculator(regionLocator, admin);
 assertEquals(((long) Integer.MAX_VALUE) * megabyte, calculator.getRegionSize("largeRegion".getBytes()));
}

@Test
public void testSimpleTestCase() throws Exception {
 RegionLocator regionLocator = mockRegionLocator("region1", "region2", "region3");
 Admin admin = mockAdmin(
   mockRegion("region1", 123),
   mockRegion("region3", 1232),
   mockRegion("region2",  54321)
 );
 RegionSizeCalculator calculator = new RegionSizeCalculator(regionLocator, admin);
 assertEquals(123 * megabyte, calculator.getRegionSize("region1".getBytes()));
 assertEquals(54321 * megabyte, calculator.getRegionSize("region2".getBytes()));
 assertEquals(1232 * megabyte, calculator.getRegionSize("region3".getBytes()));
 // if regionCalculator does not know about a region, it should return 0
 assertEquals(0 * megabyte, calculator.getRegionSize("otherTableRegion".getBytes()));
 assertEquals(3, calculator.getRegionSizeMap().size());
}

@Test
public void testSimpleTestCase() throws Exception {
 RegionLocator regionLocator = mockRegionLocator("region1", "region2", "region3");
 Admin admin = mockAdmin(
   mockRegion("region1", 123),
   mockRegion("region3", 1232),
   mockRegion("region2",  54321)
 );
 RegionSizeCalculator calculator = new RegionSizeCalculator(regionLocator, admin);
 assertEquals(123 * megabyte, calculator.getRegionSize("region1".getBytes()));
 assertEquals(54321 * megabyte, calculator.getRegionSize("region2".getBytes()));
 assertEquals(1232 * megabyte, calculator.getRegionSize("region3".getBytes()));
 // if regionCalculator does not know about a region, it should return 0
 assertEquals(0 * megabyte, calculator.getRegionSize("otherTableRegion".getBytes()));
 assertEquals(3, calculator.getRegionSizeMap().size());
}

/** When calculator is disabled, it should return 0 for each request.*/
@Test
public void testDisabled() throws Exception {
 String regionName = "cz.goout:/index.html";
 RegionLocator table = mockRegionLocator(regionName);
 Admin admin = mockAdmin(
   mockRegion(regionName, 999)
 );
 //first request on enabled calculator
 RegionSizeCalculator calculator = new RegionSizeCalculator(table, admin);
 assertEquals(999 * megabyte, calculator.getRegionSize(regionName.getBytes()));
 //then disabled calculator.
 configuration.setBoolean(RegionSizeCalculator.ENABLE_REGIONSIZECALCULATOR, false);
 RegionSizeCalculator disabledCalculator = new RegionSizeCalculator(table, admin);
 assertEquals(0 * megabyte, disabledCalculator.getRegionSize(regionName.getBytes()));
 assertEquals(0, disabledCalculator.getRegionSizeMap().size());
}

/** When calculator is disabled, it should return 0 for each request.*/
@Test
public void testDisabled() throws Exception {
 String regionName = "cz.goout:/index.html";
 RegionLocator table = mockRegionLocator(regionName);
 Admin admin = mockAdmin(
   mockRegion(regionName, 999)
 );
 //first request on enabled calculator
 RegionSizeCalculator calculator = new RegionSizeCalculator(table, admin);
 assertEquals(999 * megabyte, calculator.getRegionSize(regionName.getBytes()));
 //then disabled calculator.
 configuration.setBoolean(RegionSizeCalculator.ENABLE_REGIONSIZECALCULATOR, false);
 RegionSizeCalculator disabledCalculator = new RegionSizeCalculator(table, admin);
 assertEquals(0 * megabyte, disabledCalculator.getRegionSize(regionName.getBytes()));
 assertEquals(0, disabledCalculator.getRegionSizeMap().size());
}