OpenSearch/x-pack/plugin/ml/qa/native-multi-node-tests/build.gradle

import org.elasticsearch.gradle.LoggedExec

apply plugin: 'elasticsearch.standalone-rest-test'
apply plugin: 'elasticsearch.rest-test'

dependencies {
    // "org.elasticsearch.plugin:x-pack-core:${version}" doesn't work with idea because the testArtifacts are also here
    testCompile project(path: xpackModule('core'), configuration: 'default')
    testCompile project(path: xpackModule('core'), configuration: 'testArtifacts')
    testCompile project(path: xpackModule('ml'), configuration: 'runtime')
    testCompile project(path: xpackModule('ml'), configuration: 'testArtifacts')
}

integTestRunner {
    /*
     * We have to disable setting the number of available processors as tests in the same JVM randomize processors and will step on each
     * other if we allow them to set the number of available processors as it's set-once in Netty.
     */
    systemProperty 'es.set.netty.runtime.available.processors', 'false'
}

// location for keys and certificates
File keystoreDir = new File(project.buildDir, 'keystore')
File nodeKey = file("$keystoreDir/testnode.pem")
File nodeCert = file("$keystoreDir/testnode.crt")
// Add key and certs to test classpath: it expects it there
task copyKeyCerts(type: Copy) {
    from(project(':x-pack:plugin:core').file('src/test/resources/org/elasticsearch/xpack/security/transport/ssl/certs/simple/')) {
        include 'testnode.crt', 'testnode.pem'
    }
    into keystoreDir
}
// Add keys and cets to test classpath: it expects it there
sourceSets.test.resources.srcDir(keystoreDir)
processTestResources.dependsOn(copyKeyCerts)

integTestCluster {
    dependsOn copyKeyCerts
    setting 'xpack.security.enabled', 'true'
    setting 'xpack.ml.enabled', 'true'
    setting 'logger.org.elasticsearch.xpack.ml.datafeed', 'TRACE'
    setting 'xpack.monitoring.enabled', 'false'
    setting 'xpack.security.authc.token.enabled', 'true'
    setting 'xpack.security.transport.ssl.enabled', 'true'
    setting 'xpack.security.transport.ssl.key', nodeKey.name
    setting 'xpack.security.transport.ssl.certificate', nodeCert.name
    setting 'xpack.security.transport.ssl.verification_mode', 'certificate'
    setting 'xpack.security.audit.enabled', 'true'
    setting 'xpack.license.self_generated.type', 'trial'
    setting 'xpack.ml.min_disk_space_off_heap', '200mb'

    keystoreSetting 'bootstrap.password', 'x-pack-test-password'
    keystoreSetting 'xpack.security.transport.ssl.secure_key_passphrase', 'testnode'

    numNodes = 3

    setupCommand 'setupDummyUser',
            'bin/elasticsearch-users', 'useradd', 'x_pack_rest_user', '-p', 'x-pack-test-password', '-r', 'superuser'

    extraConfigFile nodeKey.name, nodeKey
    extraConfigFile nodeCert.name, nodeCert

    waitCondition = { node, ant ->
        File tmpFile = new File(node.cwd, 'wait.success')
        ant.get(src: "http://${node.httpUri()}/_cluster/health?wait_for_nodes=>=${numNodes}&wait_for_status=yellow",
                dest: tmpFile.toString(),
                username: 'x_pack_rest_user',
                password: 'x-pack-test-password',
                ignoreerrors: true,
                retries: 10)
        return tmpFile.exists()
    }
}