/** This tests offlining a entityGroup */
@Test(timeout = 120000)
public void testOfflineEntityGroup() throws Exception {
String table = "testOfflineEntityGroup";
try {
EntityGroupInfo egInfo = createTableAndGetOneEntityGroup(table);
EntityGroupStates entityGroupStates =
TEST_UTIL.getWaspCluster().getMaster().getAssignmentManager().getEntityGroupStates();
ServerName serverName = entityGroupStates.getFServerOfEntityGroup(egInfo);
TEST_UTIL.assertEntityGroupOnServer(egInfo, serverName, 200);
admin.offline(egInfo.getEntityGroupName());
long timeoutTime = System.currentTimeMillis() + 800;
while (true) {
List<EntityGroupInfo> entityGroups =
entityGroupStates.getEntityGroupsOfTable(Bytes.toBytes(table));
if (!entityGroups.contains(egInfo)) break;
long now = System.currentTimeMillis();
if (now > timeoutTime) {
fail("Failed to offline the entityGroup in time");
break;
}
Thread.sleep(10);
}
EntityGroupState entityGroupState = entityGroupStates.getEntityGroupState(egInfo);
assertTrue(entityGroupState.isOffline());
} finally {
TEST_UTIL.deleteTable(Bytes.toBytes(table));
}
}
/** This tests moving a entityGroup */
@Test(timeout = 120000)
public void testMoveEntityGroup() throws Exception {
String table = "testMoveEntityGroup";
try {
EntityGroupInfo egInfo = createTableAndGetOneEntityGroup(table);
EntityGroupStates entityGroupStates =
TEST_UTIL.getWaspCluster().getMaster().getAssignmentManager().getEntityGroupStates();
ServerName serverName = entityGroupStates.getFServerOfEntityGroup(egInfo);
ServerName destServerName = null;
for (int i = 0; i < 3; i++) {
FServer destServer = TEST_UTIL.getWaspCluster().getFServer(i);
if (!destServer.getServerName().equals(serverName)) {
destServerName = destServer.getServerName();
break;
}
}
assertTrue(destServerName != null && !destServerName.equals(serverName));
TEST_UTIL
.getWaspAdmin()
.move(egInfo.getEncodedNameAsBytes(), Bytes.toBytes(destServerName.getServerName()));
long timeoutTime = System.currentTimeMillis() + 5000;
while (true) {
ServerName sn = entityGroupStates.getFServerOfEntityGroup(egInfo);
if (sn != null && sn.equals(destServerName)) {
TEST_UTIL.assertEntityGroupOnServer(egInfo, sn, 2000);
break;
}
long now = System.currentTimeMillis();
if (now > timeoutTime) {
fail("Failed to move the entityGroup in time");
}
entityGroupStates.waitForUpdate(50);
}
} finally {
TEST_UTIL.deleteTable(Bytes.toBytes(table));
}
}
private MiniWaspCluster createEntityGroups(String tableName)
throws InterruptedException, ZooKeeperConnectionException, IOException, KeeperException {
MiniWaspCluster cluster = TEST_UTIL.getWaspCluster();
log("Waiting for active/ready master");
cluster.waitForActiveAndReadyMaster();
zkw = new ZooKeeperWatcher(TEST_UTIL.getConfiguration(), "testOpenedEntityGroupHandler", null);
// Create a table with entityGroups
byte[] table = Bytes.toBytes(tableName);
byte[] family = Bytes.toBytes("family");
TEST_UTIL.createTable(table);
// wait till the entityGroups are online
log("Waiting for no more RIT");
ZKAssign.blockUntilNoRIT(zkw);
return cluster;
}
/** This tests entityGroup assignment */
@Test(timeout = 120000)
public void testAssignEntityGroup() throws Exception {
String table = "testAssignEntityGroup";
try {
FTable desc = FMetaTestUtil.makeTable(table);
admin.createTable(desc);
EntityGroupInfo egInfo =
new EntityGroupInfo(
Bytes.toBytes(desc.getTableName()), Bytes.toBytes("A"), Bytes.toBytes("Z"));
FMetaEditor.addEntityGroupToMeta(conf, egInfo);
FMaster master = TEST_UTIL.getWaspCluster().getMaster();
master.assignEntityGroup(egInfo);
master.getAssignmentManager().waitForAssignment(egInfo);
ServerName serverName =
master.getAssignmentManager().getEntityGroupStates().getFServerOfEntityGroup(egInfo);
TEST_UTIL.assertEntityGroupOnServer(egInfo, serverName, 200);
} finally {
TEST_UTIL.deleteTable(Bytes.toBytes(table));
}
}
EntityGroupInfo createTableAndGetOneEntityGroup(final String tableName)
throws IOException, InterruptedException {
FTable desc = FMetaTestUtil.makeTable(tableName);
admin.createTable(desc, Bytes.toBytes("A"), Bytes.toBytes("Z"), 5);
// wait till the table is assigned
FMaster master = TEST_UTIL.getWaspCluster().getMaster();
long timeoutTime = System.currentTimeMillis() + 100;
while (true) {
List<EntityGroupInfo> entityGroups =
master
.getAssignmentManager()
.getEntityGroupStates()
.getEntityGroupsOfTable(Bytes.toBytes(tableName));
if (entityGroups.size() > 3) {
return entityGroups.get(2);
}
long now = System.currentTimeMillis();
if (now > timeoutTime) {
fail("Could not find an online entityGroup");
}
Thread.sleep(10);
}
}
