grpc
/
grpc-node
mirror of https://github.com/grpc/grpc-node.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Connectivity test: ensure all calls end before ending the test

This commit is contained in:
murgatroid99 2019-06-24 12:56:27 -07:00
parent 2e13f46141
commit eb3c479e36
1 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
test/api/connectivity_test.js
View File

@ -76,6 +76,13 @@ describe('Reconnection', function() {
server2.forceShutdown(); server2.forceShutdown();
}); });
it('Should end with either OK or UNAVAILABLE when querying a server that is shutting down', function(done) { it('Should end with either OK or UNAVAILABLE when querying a server that is shutting down', function(done) {
let pendingCalls = 0;
let testDone = false;
function maybeDone() {
if (testDone && pendingCalls == 0) {
done();
}
};
client.unary({}, (err, data) => { client.unary({}, (err, data) => {
assert.ifError(err); assert.ifError(err);
server1.tryShutdown(() => { server1.tryShutdown(() => {
@ -84,14 +91,18 @@ describe('Reconnection', function() {
client.unary({}, (err, data) => { client.unary({}, (err, data) => {
assert.ifError(err); assert.ifError(err);
clearInterval(callInterval); clearInterval(callInterval);
done(); testDone = true;
maybeDone();
}); });
}); });
let callInterval = setInterval(() => { let callInterval = setInterval(() => {
pendingCalls += 1;
client.unary({}, (err, data) => { client.unary({}, (err, data) => {
pendingCalls -= 1;
if (err) { if (err) {
assert.strictEqual(err.code, clientGrpc.status.UNAVAILABLE); assert.strictEqual(err.code, clientGrpc.status.UNAVAILABLE);
} }
maybeDone();
}); });
}, 0); }, 0);
}); });