|
@@ -126,7 +126,8 @@ public class FileBasedUnicastHostsProviderTests extends ESTestCase {
|
|
|
final Settings settings = Settings.builder()
|
|
|
.put(Environment.PATH_HOME_SETTING.getKey(), createTempDir())
|
|
|
.build();
|
|
|
- final FileBasedUnicastHostsProvider provider = new FileBasedUnicastHostsProvider(settings, transportService, executorService);
|
|
|
+ final Environment environment = new Environment(settings);
|
|
|
+ final FileBasedUnicastHostsProvider provider = new FileBasedUnicastHostsProvider(environment, transportService, executorService);
|
|
|
final List<DiscoveryNode> nodes = provider.buildDynamicNodes();
|
|
|
assertEquals(0, nodes.size());
|
|
|
}
|
|
@@ -152,13 +153,20 @@ public class FileBasedUnicastHostsProviderTests extends ESTestCase {
|
|
|
final Settings settings = Settings.builder()
|
|
|
.put(Environment.PATH_HOME_SETTING.getKey(), homeDir)
|
|
|
.build();
|
|
|
- final Path configDir = homeDir.resolve("config").resolve("discovery-file");
|
|
|
- Files.createDirectories(configDir);
|
|
|
- final Path unicastHostsPath = configDir.resolve(UNICAST_HOSTS_FILE);
|
|
|
+ final Path configPath;
|
|
|
+ if (randomBoolean()) {
|
|
|
+ configPath = homeDir.resolve("config");
|
|
|
+ } else {
|
|
|
+ configPath = createTempDir();
|
|
|
+ }
|
|
|
+ final Path discoveryFilePath = configPath.resolve("discovery-file");
|
|
|
+ Files.createDirectories(discoveryFilePath);
|
|
|
+ final Path unicastHostsPath = discoveryFilePath.resolve(UNICAST_HOSTS_FILE);
|
|
|
try (BufferedWriter writer = Files.newBufferedWriter(unicastHostsPath)) {
|
|
|
writer.write(String.join("\n", hostEntries));
|
|
|
}
|
|
|
|
|
|
- return new FileBasedUnicastHostsProvider(settings, transportService, executorService).buildDynamicNodes();
|
|
|
+ return new FileBasedUnicastHostsProvider(
|
|
|
+ new Environment(settings, configPath), transportService, executorService).buildDynamicNodes();
|
|
|
}
|
|
|
}
|