34 lines
918 B
JavaScript
34 lines
918 B
JavaScript
// Licensed to Elasticsearch B.V under one or more agreements.
|
|
// Elasticsearch B.V licenses this file to you under the Apache 2.0 License.
|
|
// See the LICENSE file in the project root for more information
|
|
|
|
'use strict'
|
|
|
|
const { promisify } = require('util')
|
|
const sleep = promisify(setTimeout)
|
|
const buildServer = require('./buildServer')
|
|
const buildCluster = require('./buildCluster')
|
|
const connection = require('./MockConnection')
|
|
|
|
async function waitCluster (client, waitForStatus = 'green', timeout = '50s', times = 0) {
|
|
if (!client) {
|
|
throw new Error('waitCluster helper: missing client instance')
|
|
}
|
|
try {
|
|
await client.cluster.health({ waitForStatus, timeout })
|
|
} catch (err) {
|
|
if (++times < 10) {
|
|
await sleep(5000)
|
|
return waitCluster(client, waitForStatus, timeout, times)
|
|
}
|
|
throw err
|
|
}
|
|
}
|
|
|
|
module.exports = {
|
|
buildServer,
|
|
buildCluster,
|
|
connection,
|
|
waitCluster
|
|
}
|