Java DelegationTokenFetcher.main方法代码示例

import org.apache.hadoop.hdfs.tools.DelegationTokenFetcher; //导入方法依赖的package包/类
/**
 * Call fetch token using http server 
 */
@Test
public void expectedTokenIsRetrievedFromHttp() throws Exception {
  bootstrap = startHttpServer(httpPort, testToken, serviceUrl);
  DelegationTokenFetcher.main(new String[] { "-webservice=" + serviceUrl,
      tokenFile });
  Path p = new Path(fileSys.getWorkingDirectory(), tokenFile);
  Credentials creds = Credentials.readTokenStorageFile(p, conf);
  Iterator<Token<?>> itr = creds.getAllTokens().iterator();
  assertTrue("token not exist error", itr.hasNext());
  Token<?> fetchedToken = itr.next();
  Assert.assertArrayEquals("token wrong identifier error",
      testToken.getIdentifier(), fetchedToken.getIdentifier());
  Assert.assertArrayEquals("token wrong password error",
      testToken.getPassword(), fetchedToken.getPassword());
  if (assertionError != null)
    throw assertionError;
}
 

import org.apache.hadoop.hdfs.tools.DelegationTokenFetcher; //导入方法依赖的package包/类
/**
 * try to fetch token without http server with IOException
 */
@Test
public void testTokenFetchFail() throws Exception {
  try {
    DelegationTokenFetcher.main(new String[] { "-webservice=" + serviceUrl,
        tokenFile });
    fail("Token fetcher shouldn't start in absense of NN");
  } catch (IOException ex) {
  }
}
 

import org.apache.hadoop.hdfs.tools.DelegationTokenFetcher; //导入方法依赖的package包/类
private static void expectDelegationTokenFetcherExit(String[] args) {
  try {
    DelegationTokenFetcher.main(args);
    fail("should call exit");
  } catch (ExitException e) {
    ExitUtil.resetFirstExitException();
  } catch (Exception ex) {
    fail("expectDelegationTokenFetcherExit ex error " + ex);
  }
}
 

import org.apache.hadoop.hdfs.tools.DelegationTokenFetcher; //导入方法依赖的package包/类
/**
 * Verify that when the DelegationTokenFetcher runs, it talks to the Namenode,
 * pulls out the correct user's token and successfully serializes it to disk.
 */
@Test
public void expectedTokenIsRetrievedFromDFS() throws Exception {
  final byte[] ident = new byte[]{1,2,3,4};
  final byte[] pw = new byte[]{42};
  final Text service = new Text(uri.toString());
  final String user = 
      UserGroupInformation.getCurrentUser().getShortUserName();

  // Create a token for the fetcher to fetch, wire NN to return it when asked
  // for this particular user.
  Token<DelegationTokenIdentifier> t = 
    new Token<DelegationTokenIdentifier>(ident, pw, KIND, service);
  when(dfs.getDelegationToken(eq(user))).thenReturn(t);
  when(dfs.renewDelegationToken(eq(t))).thenReturn(1000L);
  FakeRenewer.reset();
  
  DelegationTokenFetcher.main(new String[]{"-fs", uri.toString(), 
                                           "file.dta"});
  FileSystem fs = FileSystem.getLocal(conf);
  Path p = new Path(fs.getWorkingDirectory(), "file.dta");
  Credentials creds = Credentials.readTokenStorageFile(p , conf);
  Iterator<Token<?>> itr = creds.getAllTokens().iterator();
  // make sure we got back exactly the 1 token we expected
  assertTrue(itr.hasNext());
  assertEquals(t, itr.next());
  assertTrue(!itr.hasNext());

  DelegationTokenFetcher.main(new String[]{"--renew", "file.dta"});
  assertEquals(t, FakeRenewer.lastRenewed);
  FakeRenewer.reset();

  DelegationTokenFetcher.main(new String[]{"--cancel", "file.dta"});
  assertEquals(t, FakeRenewer.lastCanceled);
}
 

import org.apache.hadoop.hdfs.tools.DelegationTokenFetcher; //导入方法依赖的package包/类
/**
 * Verify that when the DelegationTokenFetcher runs, it talks to the Namenode,
 * pulls out the correct user's token and successfully serializes it to disk.
 */
@Test
public void expectedTokenIsRetrievedFromDFS() throws Exception {
  final byte[] ident = new DelegationTokenIdentifier(new Text("owner"),
      new Text("renewer"), new Text("realuser")).getBytes();
  final byte[] pw = new byte[] { 42 };
  final Text kind = new Text("MY-KIND");
  final Text service = new Text(uri.toString());

  // Create a token for the fetcher to fetch, wire NN to return it when asked
  // for this particular user.
  Token<DelegationTokenIdentifier> t = new Token<DelegationTokenIdentifier>(
      ident, pw, kind, service);
  when(dfs.getDelegationToken((String) null)).thenReturn(t);
  when(dfs.renewDelegationToken(eq(t))).thenReturn(1000L);
  when(dfs.getUri()).thenReturn(uri);

  FileSystem fileSys = FileSystem.getLocal(conf);
  try {
    DelegationTokenFetcher.main(new String[] { "-fs", uri.toString(),
        tokenFile });
    Path p = new Path(fileSys.getWorkingDirectory(), tokenFile);
    Credentials creds = Credentials.readTokenStorageFile(p, conf);
    Iterator<Token<?>> itr = creds.getAllTokens().iterator();
    // make sure we got back exactly the 1 token we expected
    assertTrue(itr.hasNext());
    assertEquals(t, itr.next());
    assertTrue(!itr.hasNext());

    DelegationTokenFetcher.main(new String[] { "-fs", uri.toString(),
        "--print", tokenFile });
    DelegationTokenFetcher.main(new String[] { "-fs", uri.toString(),
        "--renew", tokenFile });
    DelegationTokenFetcher.main(new String[] { "-fs", uri.toString(),
        "--cancel", tokenFile });
    verify(dfs).renewDelegationToken(eq(t));
    verify(dfs).cancelDelegationToken(eq(t));
  } finally {
    fileSys.delete(new Path(tokenFile), true);
  }
}