How to use

ZooKeeperMain$MyCommandOptions

in

org.apache.zookeeper

  @Test
  public void testSetAclRecursive() throws Exception {
    final ZooKeeper zk = createClient();
    final byte[] EMPTY = new byte[0];

    zk.setData("/", EMPTY, -1);
    zk.create("/a", EMPTY, Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
    zk.create("/a/b", EMPTY, Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
    zk.create("/a/b/c", EMPTY, Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
    zk.create("/a/d", EMPTY, Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
    zk.create("/e", EMPTY, Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);

    ZooKeeperMain zkMain = new ZooKeeperMain(zk);
    String setAclCommand = "setAcl -R /a world:anyone:r";
    zkMain.cl.parseCommand(setAclCommand);
    Assert.assertFalse(zkMain.processZKCmd(zkMain.cl));

    Assert.assertEquals(Ids.READ_ACL_UNSAFE, zk.getACL("/a", new Stat()));
    Assert.assertEquals(Ids.READ_ACL_UNSAFE, zk.getACL("/a/b", new Stat()));
    Assert.assertEquals(Ids.READ_ACL_UNSAFE, zk.getACL("/a/b/c", new Stat()));
    Assert.assertEquals(Ids.READ_ACL_UNSAFE, zk.getACL("/a/d", new Stat()));
    // /e is unset, its acl should remain the same.
    Assert.assertEquals(Ids.OPEN_ACL_UNSAFE, zk.getACL("/e", new Stat()));
  }
}

protected boolean processZKCmd(MyCommandOptions co) throws CliException, IOException, InterruptedException {
  String[] args = co.getArgArray();
  String cmd = co.getCommand();
  if (args.length < 1) {
    usage();
      throw new MalformedCommandException("Command index out of range");
    cl.parseCommand(history.get(i));
    if (cl.getCommand().equals("redo")) {
      throw new MalformedCommandException("No redoing redos");

protected void connectToZK(String newHost) throws InterruptedException, IOException {
  if (zk != null && zk.getState().isAlive()) {
    zk.close();
  }
  host = newHost;
  boolean readOnly = cl.getOption("readonly") != null;
  zk = new ZooKeeper(host,
       Integer.parseInt(cl.getOption("timeout")),
       new MyWatcher(), readOnly);
}

@Test
public void testParseWithMultipleQuotes() throws Exception {
  final ZooKeeper zk = createClient();
  ZooKeeperMain zkMain = new ZooKeeperMain(zk);
  String cmdstring = "create /node '' ''";
  zkMain.cl.parseCommand(cmdstring);
  Assert.assertEquals("expected 5 arguments", zkMain.cl.getNumArguments(), 4);
  Assert.assertEquals("create is not taken as first argument", zkMain.cl.getCmdArgument(0), "create");
  Assert.assertEquals("/node is not taken as second argument", zkMain.cl.getCmdArgument(1), "/node");
  Assert.assertEquals("empty string is not taken as third argument", zkMain.cl.getCmdArgument(2), "");
  Assert.assertEquals("empty string is not taken as fourth argument", zkMain.cl.getCmdArgument(3), "");
}

@Test
public void testSetData() throws Exception {
  final ZooKeeper zk = createClient();
  ZooKeeperMain zkMain = new ZooKeeperMain(zk);
  String cmdstring1 = "create -e /node4 data";
  String cmdstring2 = "set /node4 " + "data";
  String cmdstring3 = "delete /node4";
  Stat stat = new Stat();
  int version = 0;
  zkMain.cl.parseCommand(cmdstring1);
  Assert.assertTrue(zkMain.processZKCmd(zkMain.cl));
  stat = zk.exists("/node4", true);
  version = stat.getVersion();
  zkMain.cl.parseCommand(cmdstring2);
  Assert.assertFalse(zkMain.processZKCmd(zkMain.cl));
  stat = zk.exists("/node4", true);
  Assert.assertEquals(version + 1, stat.getVersion());
  zkMain.cl.parseCommand(cmdstring3);
  Assert.assertFalse(zkMain.processZKCmd(zkMain.cl));
}

private void testInvalidCommand(String cmdString, int exitCode) throws Exception {
  final ZooKeeper zk = createClient();
  ZooKeeperMain zkMain = new ZooKeeperMain(zk);
  zkMain.cl.parseCommand(cmdString);
  // Verify that the exit code is set properly
  zkMain.processCmd(zkMain.cl);
  Assert.assertEquals(exitCode, zkMain.exitCode);
  // Verify that the correct exception is thrown
  try {
   zkMain.processZKCmd(zkMain.cl);
   Assert.fail();
  } catch (CliException e) {
   return;
  }
  Assert.fail("invalid command should throw CliException");
}

  public ZooKeeperMain(String args[]) throws IOException, InterruptedException {
    cl.parseOptions(args);
    System.out.println("Connecting to " + cl.getOption("server"));
    connectToZK(cl.getOption("server"));
    //zk = new ZooKeeper(cl.getOption("server"),
//                Integer.parseInt(cl.getOption("timeout")), new MyWatcher());
  }

public void executeLine(String line)
throws InterruptedException, IOException, KeeperException {
 if (!line.equals("")) {
  cl.parseCommand(line);
  addToHistory(commandCount,line);
  processCmd(cl);
  commandCount++;
 }
}

/**
 * Test verifies deletion of NodeDataChanged watches
 */
@Test(timeout = 30000)
public void testRemoveNodeDataChangedWatches() throws Exception {
  LOG.info("Adding data watcher using getData()");
  List<EventType> expectedEvents = new ArrayList<Watcher.Event.EventType>();
  expectedEvents.add(EventType.DataWatchRemoved);
  MyWatcher myWatcher = new MyWatcher("/testnode1", expectedEvents, 1);
  zk.create("/testnode1", "data".getBytes(), Ids.OPEN_ACL_UNSAFE,
      CreateMode.PERSISTENT);
  zk.getData("/testnode1", myWatcher, null);
  String cmdstring = "removewatches /testnode1 -d";
  LOG.info("Remove watchers using shell command : {}", cmdstring);
  zkMain.cl.parseCommand(cmdstring);
  Assert.assertTrue("Removewatches cmd fails to remove data watches",
      zkMain.processZKCmd(zkMain.cl));
  LOG.info("Waiting for the DataWatchRemoved event");
  myWatcher.matches();
  // verifying that other path data watches are removed
  Assert.assertEquals(
      "Data watches are not removed : " + zk.getDataWatches(), 0, zk
          .getDataWatches().size());
}

zkMain.cl.parseCommand(cmdstring);
Assert.assertTrue("Removewatches cmd fails to remove child watches",
    zkMain.processZKCmd(zkMain.cl));

zkMain.cl.parseCommand(cmdstring);
Assert.assertTrue("Doesn't create node without data", zkMain
    .processZKCmd(zkMain.cl));
zkMain.cl.parseCommand(cmdstring);
Assert.assertTrue("Doesn't create node without data", zkMain
    .processZKCmd(zkMain.cl));
zkMain.cl.parseCommand(cmdstring);
Assert.assertTrue("Doesn't create node without data", zkMain
    .processZKCmd(zkMain.cl));
zkMain.cl.parseCommand(cmdstring);
try {
  Assert.assertTrue("Created node with wrong option", zkMain

@Test
public void testParseWithMixedQuotes() throws Exception {
  final ZooKeeper zk = createClient();
  ZooKeeperMain zkMain = new ZooKeeperMain(zk);
  for (String[] quoteChars : new String[][] {{"'", "\""}, {"\"", "'"}}) {
    String outerQuotes = quoteChars[0];
    String innerQuotes = quoteChars[1];
    String cmdstring = String.format("create /node %1$s%2$squoted data%2$s%1$s", outerQuotes, innerQuotes);
    zkMain.cl.parseCommand(cmdstring);
    Assert.assertEquals("quotes combine arguments", zkMain.cl.getNumArguments(), 3);
    Assert.assertEquals("create is not taken as first argument", zkMain.cl.getCmdArgument(0), "create");
    Assert.assertEquals("/node is not taken as second argument", zkMain.cl.getCmdArgument(1), "/node");
    Assert.assertEquals("quoted data is not taken as third argument", zkMain.cl.getCmdArgument(2), innerQuotes + "quoted data" + innerQuotes);
  }
}

zkMain.cl.parseCommand(cmdstring1);
Assert.assertTrue(
    "Removewatches cmd fails to remove pre-create watches",
zkMain.cl.parseCommand(cmdstring2);
Assert.assertTrue("Removewatches cmd fails to remove data watches",
    zkMain.processZKCmd(zkMain.cl));

@Test
public void testInvalidStatCommand() throws Exception {
  final ZooKeeper zk = createClient();
  ZooKeeperMain zkMain = new ZooKeeperMain(zk);
  // node doesn't exists
  String cmdstring1 = "stat /node123";
  zkMain.cl.parseCommand(cmdstring1);
  try {
    Assert.assertFalse(zkMain.processZKCmd(zkMain.cl));
    Assert.fail("Path doesn't exists so, command should fail.");
  } catch (CliWrapperException e) {
    Assert.assertEquals(KeeperException.Code.NONODE, ((KeeperException)e.getCause()).code());
  }
}

zkMain.cl.parseCommand(cmdstring0);
Assert.assertFalse(zkMain.processZKCmd(zkMain.cl));
Assert.assertEquals(null, zk.exists("/a/b/v", null));
zkMain.cl.parseCommand(cmdstring1);
Assert.assertFalse(zkMain.processZKCmd(zkMain.cl));
Assert.assertNull(zk.exists("/a", null));

@Test
public void testStatWhenPathDoesNotExist() throws IOException,
    InterruptedException, MalformedCommandException {
  final ZooKeeper zk = createClient();
  ZooKeeperMain main = new ZooKeeperMain(zk);
  String cmdstring = "stat /invalidPath";
  main.cl.parseCommand(cmdstring);
  try {
    main.processZKCmd(main.cl);
    Assert.fail("As Node does not exist, command should fail by throwing No Node Exception.");
  } catch (CliException e) {
    Assert.assertEquals("Node does not exist: /invalidPath", e.getMessage());
  }
}

zkMain.cl.parseCommand(cmdstring);
Assert.assertTrue("Removewatches cmd fails to remove child watches",
    zkMain.processZKCmd(zkMain.cl));

@Test
public void testStatCommand() throws Exception {
  final ZooKeeper zk = createClient();
  ZooKeeperMain zkMain = new ZooKeeperMain(zk);
  String cmdstring1 = "create -e /node3 data";
  String cmdstring2 = "stat /node3";
  String cmdstring3 = "delete /node3";
  zkMain.cl.parseCommand(cmdstring1);
  Assert.assertTrue(zkMain.processZKCmd(zkMain.cl));
  zkMain.cl.parseCommand(cmdstring2);
  Assert.assertFalse(zkMain.processZKCmd(zkMain.cl));
  zkMain.cl.parseCommand(cmdstring3);
  Assert.assertFalse(zkMain.processZKCmd(zkMain.cl));
}

void run() throws CliException, IOException, InterruptedException {
  if (cl.getCommand() == null) {
    System.out.println("Welcome to ZooKeeper!");

@Test
public void testDelete() throws Exception {
  final ZooKeeper zk = createClient();
  ZooKeeperMain zkMain = new ZooKeeperMain(zk);
  String cmdstring1 = "create -e /node2 data";
  String cmdstring2 = "delete /node2";
  String cmdstring3 = "ls /node2";
  zkMain.cl.parseCommand(cmdstring1);
  Assert.assertTrue(zkMain.processZKCmd(zkMain.cl));
  zkMain.cl.parseCommand(cmdstring2);
  Assert.assertFalse(zkMain.processZKCmd(zkMain.cl));
  zkMain.cl.parseCommand(cmdstring3);
  Assert.assertFalse("", zkMain.processCmd(zkMain.cl));
}