Fix users tool tests

This commit fixes the users tool command tests which were broken because
of a guard added that es.path.conf is set. We do not want to set this
system property in tests so instead we override createEnv where the
problematic guard exists.

Original commit: elastic/x-pack-elasticsearch@78b757695b
This commit is contained in:
Jason Tedor 2017-07-28 17:32:39 +09:00
parent d8277942ac
commit e77db8faf4
2 changed files with 80 additions and 6 deletions

View File

@ -47,11 +47,31 @@ public class UsersTool extends MultiCommand {
UsersTool() { UsersTool() {
super("Manages elasticsearch native users"); super("Manages elasticsearch native users");
subcommands.put("useradd", new AddUserCommand()); subcommands.put("useradd", newAddUserCommand());
subcommands.put("userdel", new DeleteUserCommand()); subcommands.put("userdel", newDeleteUserCommand());
subcommands.put("passwd", new PasswordCommand()); subcommands.put("passwd", newPasswordCommand());
subcommands.put("roles", new RolesCommand()); subcommands.put("roles", newRolesCommand());
subcommands.put("list", new ListCommand()); subcommands.put("list", newListCommand());
}
protected AddUserCommand newAddUserCommand() {
return new AddUserCommand();
}
protected DeleteUserCommand newDeleteUserCommand() {
return new DeleteUserCommand();
}
protected PasswordCommand newPasswordCommand() {
return new PasswordCommand();
}
protected RolesCommand newRolesCommand() {
return new RolesCommand();
}
protected ListCommand newListCommand() {
return new ListCommand();
} }
static class AddUserCommand extends EnvironmentAwareCommand { static class AddUserCommand extends EnvironmentAwareCommand {

View File

@ -10,7 +10,9 @@ import com.google.common.jimfs.Jimfs;
import org.apache.lucene.util.IOUtils; import org.apache.lucene.util.IOUtils;
import org.elasticsearch.cli.Command; import org.elasticsearch.cli.Command;
import org.elasticsearch.cli.CommandTestCase; import org.elasticsearch.cli.CommandTestCase;
import org.elasticsearch.cli.EnvironmentAwareCommand;
import org.elasticsearch.cli.ExitCodes; import org.elasticsearch.cli.ExitCodes;
import org.elasticsearch.cli.Terminal;
import org.elasticsearch.cli.UserException; import org.elasticsearch.cli.UserException;
import org.elasticsearch.common.Strings; import org.elasticsearch.common.Strings;
import org.elasticsearch.common.io.PathUtilsForTesting; import org.elasticsearch.common.io.PathUtilsForTesting;
@ -19,6 +21,7 @@ import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.env.Environment; import org.elasticsearch.env.Environment;
import org.elasticsearch.test.SecuritySettingsSource; import org.elasticsearch.test.SecuritySettingsSource;
import org.elasticsearch.xpack.XPackSettings; import org.elasticsearch.xpack.XPackSettings;
import org.elasticsearch.xpack.security.authc.UserTokenTests;
import org.elasticsearch.xpack.security.authc.support.Hasher; import org.elasticsearch.xpack.security.authc.support.Hasher;
import org.elasticsearch.xpack.security.authz.store.ReservedRolesStore; import org.elasticsearch.xpack.security.authz.store.ReservedRolesStore;
import org.elasticsearch.xpack.XPackPlugin; import org.elasticsearch.xpack.XPackPlugin;
@ -36,6 +39,7 @@ import java.nio.file.Path;
import java.util.Arrays; import java.util.Arrays;
import java.util.Collections; import java.util.Collections;
import java.util.List; import java.util.List;
import java.util.Map;
import java.util.Objects; import java.util.Objects;
public class UsersToolTests extends CommandTestCase { public class UsersToolTests extends CommandTestCase {
@ -102,7 +106,57 @@ public class UsersToolTests extends CommandTestCase {
@Override @Override
protected Command newCommand() { protected Command newCommand() {
return new UsersTool(); return new UsersTool() {
@Override
protected AddUserCommand newAddUserCommand() {
return new AddUserCommand() {
@Override
protected Environment createEnv(Terminal terminal, Map<String, String> settings) throws UserException {
return new Environment(UsersToolTests.this.settings, confDir.getParent());
}
};
}
@Override
protected DeleteUserCommand newDeleteUserCommand() {
return new DeleteUserCommand() {
@Override
protected Environment createEnv(Terminal terminal, Map<String, String> settings) throws UserException {
return new Environment(UsersToolTests.this.settings, confDir.getParent());
}
};
}
@Override
protected PasswordCommand newPasswordCommand() {
return new PasswordCommand() {
@Override
protected Environment createEnv(Terminal terminal, Map<String, String> settings) throws UserException {
return new Environment(UsersToolTests.this.settings, confDir.getParent());
}
};
}
@Override
protected RolesCommand newRolesCommand() {
return new RolesCommand() {
@Override
protected Environment createEnv(Terminal terminal, Map<String, String> settings) throws UserException {
return new Environment(UsersToolTests.this.settings, confDir.getParent());
}
};
}
@Override
protected ListCommand newListCommand() {
return new ListCommand() {
@Override
protected Environment createEnv(Terminal terminal, Map<String, String> settings) throws UserException {
return new Environment(UsersToolTests.this.settings, confDir.getParent());
}
};
}
};
} }
/** checks the user exists with the given password */ /** checks the user exists with the given password */