ACCUMULO-1376 Add a unit test that ensure that the user is prompted w/o the force option when dropping a user
Project: http://git-wip-us.apache.org/repos/asf/accumulo/repo Commit: http://git-wip-us.apache.org/repos/asf/accumulo/commit/3bb70a69 Tree: http://git-wip-us.apache.org/repos/asf/accumulo/tree/3bb70a69 Diff: http://git-wip-us.apache.org/repos/asf/accumulo/diff/3bb70a69 Branch: refs/heads/ACCUMULO-378 Commit: 3bb70a694103a7290cdebcd4dbbed0935a0b5a56 Parents: f3d7391 Author: Josh Elser <els...@apache.org> Authored: Thu Jun 12 11:39:15 2014 -0400 Committer: Josh Elser <els...@apache.org> Committed: Thu Jun 12 11:39:15 2014 -0400 ---------------------------------------------------------------------- .../shell/command/DropUserCommandTest.java | 83 ++++++++++++++++++++ 1 file changed, 83 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/accumulo/blob/3bb70a69/shell/src/test/java/org/apache/accumulo/shell/command/DropUserCommandTest.java ---------------------------------------------------------------------- diff --git a/shell/src/test/java/org/apache/accumulo/shell/command/DropUserCommandTest.java b/shell/src/test/java/org/apache/accumulo/shell/command/DropUserCommandTest.java new file mode 100644 index 0000000..a204075 --- /dev/null +++ b/shell/src/test/java/org/apache/accumulo/shell/command/DropUserCommandTest.java @@ -0,0 +1,83 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.accumulo.shell.command; + +import jline.console.ConsoleReader; + +import org.apache.accumulo.core.client.Connector; +import org.apache.accumulo.core.client.admin.SecurityOperations; +import org.apache.accumulo.shell.Shell; +import org.apache.accumulo.shell.commands.DropUserCommand; +import org.apache.commons.cli.CommandLine; +import org.easymock.EasyMock; +import org.junit.Before; +import org.junit.Test; + +/** + * + */ +public class DropUserCommandTest { + + private DropUserCommand cmd; + + @Before + public void setup() { + cmd = new DropUserCommand(); + + // Initialize that internal state + cmd.getOptions(); + } + + @Test + public void dropUserWithoutForcePrompts() throws Exception { + Connector conn = EasyMock.createMock(Connector.class); + CommandLine cli = EasyMock.createMock(CommandLine.class); + Shell shellState = EasyMock.createMock(Shell.class); + ConsoleReader reader = EasyMock.createMock(ConsoleReader.class); + SecurityOperations secOps = EasyMock.createMock(SecurityOperations.class); + + EasyMock.expect(shellState.getConnector()).andReturn(conn); + + // The user we want to remove + EasyMock.expect(cli.getArgs()).andReturn(new String[] {"user"}); + + // We're the root user + EasyMock.expect(conn.whoami()).andReturn("root"); + + // Force option was not provided + EasyMock.expect(cli.hasOption("f")).andReturn(false); + EasyMock.expect(shellState.getReader()).andReturn(reader); + reader.flush(); + EasyMock.expectLastCall().once(); + + // Fake a "yes" response + EasyMock.expect(shellState.getReader()).andReturn(reader); + EasyMock.expect(reader.readLine(EasyMock.anyObject(String.class))).andReturn("yes"); + EasyMock.expect(shellState.getConnector()).andReturn(conn); + + EasyMock.expect(conn.securityOperations()).andReturn(secOps); + secOps.dropLocalUser("user"); + EasyMock.expectLastCall(); + + EasyMock.replay(conn, cli, shellState, reader, secOps); + + cmd.execute("dropuser foo -f", cli, shellState); + + EasyMock.verify(conn, cli, shellState, reader, secOps); + } + +}
