Examples with Environment org.opensearch.env.Environment used on opensource projects

Example 76 with Environment

use of org.opensearch.env.Environment in project OpenSearch by opensearch-project.

the class ListPluginsCommandTests method listPlugins.

static MockTerminal listPlugins(Path home, String[] args) throws Exception {
    String[] argsAndHome = new String[args.length + 1];
    System.arraycopy(args, 0, argsAndHome, 0, args.length);
    argsAndHome[args.length] = "-Epath.home=" + home;
    MockTerminal terminal = new MockTerminal();
    int status = new ListPluginsCommand() {

        @Override
        protected Environment createEnv(Map<String, String> settings) throws UserException {
            Settings.Builder builder = Settings.builder().put("path.home", home);
            settings.forEach((k, v) -> builder.put(k, v));
            final Settings realSettings = builder.build();
            return new Environment(realSettings, home.resolve("config"));
        }

        @Override
        protected boolean addShutdownHook() {
            return false;
        }
    }.main(argsAndHome, terminal);
    assertEquals(ExitCodes.OK, status);
    return terminal;
}

Example 77 with Environment

use of org.opensearch.env.Environment in project OpenSearch by opensearch-project.

the class InstallPluginCommandTests method testSpaceInUrl.

public void testSpaceInUrl() throws Exception {
    Tuple<Path, Environment> env = createEnv(fs, temp);
    Path pluginDir = createPluginDir(temp);
    String pluginZip = createPluginUrl("fake", pluginDir);
    Path pluginZipWithSpaces = createTempFile("foo bar", ".zip");
    try (InputStream in = FileSystemUtils.openFileURLStream(new URL(pluginZip))) {
        Files.copy(in, pluginZipWithSpaces, StandardCopyOption.REPLACE_EXISTING);
    }
    installPlugin(pluginZipWithSpaces.toUri().toURL().toString(), env.v1());
    assertPlugin("fake", pluginDir, env.v2());
}

Example 78 with Environment

use of org.opensearch.env.Environment in project OpenSearch by opensearch-project.

the class InstallPluginCommandTests method testUnknownPlugin.

public void testUnknownPlugin() throws Exception {
    Tuple<Path, Environment> env = createEnv(fs, temp);
    UserException e = expectThrows(UserException.class, () -> installPlugin("foo", env.v1()));
    assertTrue(e.getMessage(), e.getMessage().contains("Unknown plugin foo"));
}

Example 79 with Environment

use of org.opensearch.env.Environment in project OpenSearch by opensearch-project.

the class AvailableIndexFoldersBenchmark method setup.

@Setup
public void setup() throws IOException {
    Path path = Files.createTempDirectory("test");
    String[] paths = new String[] { path.toString() };
    nodePath = new NodeEnvironment.NodePath(path);
    LogConfigurator.setNodeName("test");
    Settings settings = Settings.builder().put(Environment.PATH_HOME_SETTING.getKey(), path).putList(Environment.PATH_DATA_SETTING.getKey(), paths).build();
    nodeEnv = new NodeEnvironment(settings, new Environment(settings, null));
    Files.createDirectories(nodePath.indicesPath);
    excludedDirs = new HashSet<>();
    int numIndices = 5000;
    for (int i = 0; i < numIndices; i++) {
        String dirName = "dir" + i;
        Files.createDirectory(nodePath.indicesPath.resolve(dirName));
        excludedDirs.add(dirName);
    }
    if (nodeEnv.availableIndexFoldersForPath(nodePath).size() != numIndices) {
        throw new IllegalStateException("bad size");
    }
    if (nodeEnv.availableIndexFoldersForPath(nodePath, excludedDirs::contains).size() != 0) {
        throw new IllegalStateException("bad size");
    }
}

Example 80 with Environment

use of org.opensearch.env.Environment in project OpenSearch by opensearch-project.

the class InstallPluginCommandTests method testMissingDescriptor.

public void testMissingDescriptor() throws Exception {
    Tuple<Path, Environment> env = createEnv(fs, temp);
    Path pluginDir = createPluginDir(temp);
    Files.createFile(pluginDir.resolve("fake.yml"));
    String pluginZip = writeZip(pluginDir, null).toUri().toURL().toString();
    NoSuchFileException e = expectThrows(NoSuchFileException.class, () -> installPlugin(pluginZip, env.v1()));
    assertTrue(e.getMessage(), e.getMessage().contains("plugin-descriptor.properties"));
    assertInstallCleaned(env.v2());
}