コード例 #1
0
 /** 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));
   }
 }
コード例 #2
0
  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);
    }
  }
コード例 #3
0
  @Test
  public void testShouldNotCompeleteOpenedEntityGroupSuccessfullyIfVersionMismatches()
      throws Exception {
    EntityGroup entityGroup = null;
    try {
      int testIndex = 0;
      TEST_UTIL.getHBaseTestingUtility().startMiniZKCluster();
      TEST_UTIL
          .getConfiguration()
          .set(
              FConstants.ZOOKEEPER_QUORUM,
              TEST_UTIL.getConfiguration().get(HConstants.ZOOKEEPER_QUORUM));
      TEST_UTIL
          .getConfiguration()
          .set(
              FConstants.ZOOKEEPER_CLIENT_PORT,
              TEST_UTIL.getConfiguration().get(HConstants.ZOOKEEPER_CLIENT_PORT));
      final Server server = new MockServer(TEST_UTIL);
      FTable htd =
          FMetaTestUtil.makeTable(
              "testShouldNotCompeleteOpenedEntityGroupSuccessfullyIfVersionMismatches");
      EntityGroupInfo egi =
          new EntityGroupInfo(
              Bytes.toBytes(htd.getTableName()),
              Bytes.toBytes(testIndex),
              Bytes.toBytes(testIndex + 1));
      entityGroup = EntityGroup.createEntityGroup(egi, TEST_UTIL.getConfiguration(), htd, null);
      assertNotNull(entityGroup);
      AssignmentManager am = Mockito.mock(AssignmentManager.class);
      EntityGroupStates rsm = Mockito.mock(EntityGroupStates.class);
      Mockito.doReturn(rsm).when(am).getEntityGroupStates();
      when(rsm.isEntityGroupInTransition(egi)).thenReturn(false);
      when(rsm.getEntityGroupState(egi))
          .thenReturn(
              new EntityGroupState(
                  entityGroup.getEntityGroupInfo(),
                  EntityGroupState.State.OPEN,
                  System.currentTimeMillis(),
                  server.getServerName()));
      // create a node with OPENED state
      zkw =
          WaspTestingUtility.createAndForceNodeToOpenedState(
              TEST_UTIL, entityGroup, server.getServerName());
      when(am.getZKTable()).thenReturn(new ZKTable(zkw));
      Stat stat = new Stat();
      String nodeName =
          ZKAssign.getNodeName(zkw, entityGroup.getEntityGroupInfo().getEncodedName());
      ZKUtil.getDataAndWatch(zkw, nodeName, stat);

      // use the version for the OpenedEntityGroupHandler
      OpenedEntityGroupHandler handler =
          new OpenedEntityGroupHandler(
              server,
              am,
              entityGroup.getEntityGroupInfo(),
              server.getServerName(),
              stat.getVersion());
      // Once again overwrite the same znode so that the version changes.
      ZKAssign.transitionNode(
          zkw,
          entityGroup.getEntityGroupInfo(),
          server.getServerName(),
          EventType.FSERVER_ZK_ENTITYGROUP_OPENED,
          EventType.FSERVER_ZK_ENTITYGROUP_OPENED,
          stat.getVersion());

      // Should not invoke assignmentmanager.entityGroupOnline. If it is
      // invoked as per current mocking it will throw null pointer exception.
      boolean expectedException = false;
      try {
        handler.process();
      } catch (Exception e) {
        expectedException = true;
      }
      assertFalse("The process method should not throw any exception.", expectedException);
      List<String> znodes = ZKUtil.listChildrenAndWatchForNewChildren(zkw, zkw.assignmentZNode);
      String entityGroupName = znodes.get(0);
      assertEquals(
          "The entityGroup should not be opened successfully.",
          entityGroupName,
          entityGroup.getEntityGroupInfo().getEncodedName());
    } finally {
      EntityGroup.closeEntityGroup(entityGroup);
      TEST_UTIL.getHBaseTestingUtility().shutdownMiniZKCluster();
    }
  }