138 lines
3.1 KiB
JavaScript
138 lines
3.1 KiB
JavaScript
'use strict'
|
|
|
|
const { test } = require('tap')
|
|
const { URL } = require('url')
|
|
const { buildCluster } = require('../utils')
|
|
const { Client, Connection, events, errors } = require('../../index')
|
|
|
|
test('Should update the connection pool', t => {
|
|
t.plan(8)
|
|
|
|
buildCluster(({ nodes, shutdown }) => {
|
|
const client = new Client({
|
|
node: nodes[Object.keys(nodes)[0]].url
|
|
})
|
|
t.strictEqual(client.connectionPool.connections.size, 1)
|
|
|
|
// run the sniffer
|
|
client.transport.sniff((err, hosts) => {
|
|
t.error(err)
|
|
t.strictEqual(hosts.length, 4)
|
|
|
|
const ids = Object.keys(nodes)
|
|
for (var i = 0; i < hosts.length; i++) {
|
|
const id = ids[i]
|
|
t.deepEqual(hosts[i], {
|
|
url: new URL(nodes[id].url),
|
|
id: id,
|
|
roles: {
|
|
master: true,
|
|
data: true,
|
|
ingest: true
|
|
},
|
|
ssl: null,
|
|
agent: null
|
|
})
|
|
}
|
|
|
|
t.strictEqual(client.connectionPool.connections.size, 4)
|
|
shutdown()
|
|
})
|
|
})
|
|
})
|
|
|
|
test('Sniff interval', t => {
|
|
t.plan(8)
|
|
|
|
buildCluster(({ nodes, shutdown, kill }) => {
|
|
const client = new Client({
|
|
node: nodes[Object.keys(nodes)[0]].url,
|
|
sniffInterval: 50
|
|
})
|
|
|
|
// this event will be triggered by api calls
|
|
client.on(events.SNIFF, (err, hosts) => {
|
|
t.error(err)
|
|
t.strictEqual(
|
|
client.connectionPool.connections.size,
|
|
hosts.length
|
|
)
|
|
})
|
|
|
|
t.strictEqual(client.connectionPool.connections.size, 1)
|
|
setTimeout(() => client.info(t.error), 60)
|
|
|
|
setTimeout(() => {
|
|
// let's kill a node
|
|
kill('node1')
|
|
client.info(t.error)
|
|
}, 150)
|
|
|
|
setTimeout(() => {
|
|
t.strictEqual(client.connectionPool.connections.size, 3)
|
|
shutdown()
|
|
}, 200)
|
|
})
|
|
})
|
|
|
|
test('Should not close living connections', t => {
|
|
t.plan(3)
|
|
|
|
buildCluster(({ nodes, shutdown, kill }) => {
|
|
class MyConnection extends Connection {
|
|
close () {
|
|
t.fail('Should not be called')
|
|
}
|
|
}
|
|
|
|
const client = new Client({
|
|
node: {
|
|
url: new URL(nodes[Object.keys(nodes)[0]].url),
|
|
id: 'node1'
|
|
},
|
|
Connection: MyConnection
|
|
})
|
|
|
|
t.strictEqual(client.connectionPool.connections.size, 1)
|
|
client.transport.sniff((err, hosts) => {
|
|
t.error(err)
|
|
t.strictEqual(
|
|
client.connectionPool.connections.size,
|
|
hosts.length
|
|
)
|
|
shutdown()
|
|
})
|
|
})
|
|
})
|
|
|
|
test('Sniff on connection fault', t => {
|
|
t.plan(4)
|
|
|
|
buildCluster(({ nodes, shutdown }) => {
|
|
const client = new Client({
|
|
nodes: [
|
|
// TODO: this url may cause a flaky test
|
|
'http://localhost:9200',
|
|
nodes[Object.keys(nodes)[0]].url
|
|
],
|
|
maxRetries: 0,
|
|
sniffOnConnectionFault: true
|
|
})
|
|
t.strictEqual(client.connectionPool.connections.size, 2)
|
|
|
|
// this event will be triggered by the connection fault
|
|
client.on(events.SNIFF, (err, hosts) => {
|
|
t.error(err)
|
|
t.strictEqual(
|
|
client.connectionPool.connections.size,
|
|
hosts.length
|
|
)
|
|
shutdown()
|
|
})
|
|
|
|
client.info((err, result) => {
|
|
t.ok(err instanceof errors.ConnectionError)
|
|
})
|
|
})
|
|
})
|