use of de.flapdoodle.embed.process.io.IStreamProcessor in project gora by apache.
the class GoraMongodbAuthenticationTestDriver method runScriptAndWait.
private void runScriptAndWait(String scriptText, String token, String[] failures, String dbName, String username, String password) throws IOException {
IStreamProcessor mongoOutput;
if (!isEmpty(token)) {
mongoOutput = new LogWatchStreamProcessor(token, (failures != null) ? new HashSet<>(asList(failures)) : Collections.emptySet(), namedConsole("[mongo shell output]"));
} else {
mongoOutput = new NamedOutputStreamProcessor("[mongo shell output]", console());
}
IRuntimeConfig runtimeConfig = new RuntimeConfigBuilder().defaults(Command.Mongo).processOutput(new ProcessOutput(mongoOutput, namedConsole("[mongo shell error]"), console())).build();
MongoShellStarter starter = MongoShellStarter.getInstance(runtimeConfig);
final File scriptFile = writeTmpScriptFile(scriptText);
final MongoShellConfigBuilder builder = new MongoShellConfigBuilder();
if (!isEmpty(dbName)) {
builder.dbName(dbName);
}
if (!isEmpty(username)) {
builder.username(username);
}
if (!isEmpty(password)) {
builder.password(password);
}
starter.prepare(builder.scriptName(scriptFile.getAbsolutePath()).version(mongodConfig.version()).net(mongodConfig.net()).build()).start();
if (mongoOutput instanceof LogWatchStreamProcessor) {
((LogWatchStreamProcessor) mongoOutput).waitForResult(INIT_TIMEOUT_MS);
}
}
Aggregations