mirror of https://github.com/grpc/grpc-node.git
104 lines
4.4 KiB
TypeScript
104 lines
4.4 KiB
TypeScript
/*
|
|
* Copyright 2023 gRPC authors.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*
|
|
*/
|
|
|
|
import { Backend, createBackends } from "./backend";
|
|
import { XdsTestClient } from "./client";
|
|
import { FakeEdsCluster, FakeRouteGroup, FakeServerRoute } from "./framework";
|
|
import { ControlPlaneServer } from "./xds-server";
|
|
|
|
import { register } from "../src";
|
|
import assert = require("assert");
|
|
import { connectivityState } from "@grpc/grpc-js";
|
|
|
|
register();
|
|
|
|
describe('core xDS functionality', () => {
|
|
let xdsServer: ControlPlaneServer;
|
|
let client: XdsTestClient;
|
|
beforeEach(done => {
|
|
xdsServer = new ControlPlaneServer();
|
|
xdsServer.startServer(error => {
|
|
done(error);
|
|
});
|
|
});
|
|
afterEach(() => {
|
|
client?.close();
|
|
xdsServer?.shutdownServer();
|
|
})
|
|
it('should route requests to the single backend', async () => {
|
|
const [backend] = await createBackends(1);
|
|
const serverRoute = new FakeServerRoute(backend.getPort(), 'serverRoute');
|
|
xdsServer.setRdsResource(serverRoute.getRouteConfiguration());
|
|
xdsServer.setLdsResource(serverRoute.getListener());
|
|
xdsServer.addResponseListener((typeUrl, responseState) => {
|
|
if (responseState.state === 'NACKED') {
|
|
client?.stopCalls();
|
|
assert.fail(`Client NACKED ${typeUrl} resource with message ${responseState.errorMessage}`);
|
|
}
|
|
});
|
|
const cluster = new FakeEdsCluster('cluster1', 'endpoint1', [{backends: [backend], locality:{region: 'region1'}}]);
|
|
const routeGroup = new FakeRouteGroup('listener1', 'route1', [{cluster: cluster}]);
|
|
await routeGroup.startAllBackends(xdsServer);
|
|
xdsServer.setEdsResource(cluster.getEndpointConfig());
|
|
xdsServer.setCdsResource(cluster.getClusterConfig());
|
|
xdsServer.setRdsResource(routeGroup.getRouteConfiguration());
|
|
xdsServer.setLdsResource(routeGroup.getListener());
|
|
client = XdsTestClient.createFromServer('listener1', xdsServer);
|
|
client.startCalls(100);
|
|
await routeGroup.waitForAllBackendsToReceiveTraffic();
|
|
client.stopCalls();
|
|
});
|
|
it('should be able to enter and exit idle', function(done) {
|
|
this.timeout(5000);
|
|
createBackends(1).then(([backend]) => {
|
|
const serverRoute = new FakeServerRoute(backend.getPort(), 'serverRoute');
|
|
xdsServer.setRdsResource(serverRoute.getRouteConfiguration());
|
|
xdsServer.setLdsResource(serverRoute.getListener());
|
|
xdsServer.addResponseListener((typeUrl, responseState) => {
|
|
if (responseState.state === 'NACKED') {
|
|
client.stopCalls();
|
|
assert.fail(`Client NACKED ${typeUrl} resource with message ${responseState.errorMessage}`);
|
|
}
|
|
});
|
|
const cluster = new FakeEdsCluster('cluster1', 'endpoint1', [{backends: [backend], locality:{region: 'region1'}}]);
|
|
const routeGroup = new FakeRouteGroup('listener1', 'route1', [{cluster: cluster}]);
|
|
routeGroup.startAllBackends(xdsServer).then(() => {
|
|
xdsServer.setEdsResource(cluster.getEndpointConfig());
|
|
xdsServer.setCdsResource(cluster.getClusterConfig());
|
|
xdsServer.setRdsResource(routeGroup.getRouteConfiguration());
|
|
xdsServer.setLdsResource(routeGroup.getListener());
|
|
const serverRoute = new FakeServerRoute(backend.getPort(), 'serverRoute');
|
|
xdsServer.setRdsResource(serverRoute.getRouteConfiguration());
|
|
xdsServer.setLdsResource(serverRoute.getListener());
|
|
client = XdsTestClient.createFromServer('listener1', xdsServer, {
|
|
'grpc.client_idle_timeout_ms': 1000,
|
|
});
|
|
client.sendOneCall(error => {
|
|
assert.ifError(error);
|
|
assert.strictEqual(client.getConnectivityState(), connectivityState.READY);
|
|
setTimeout(() => {
|
|
assert.strictEqual(client.getConnectivityState(), connectivityState.IDLE);
|
|
client.sendOneCall(error => {
|
|
done(error);
|
|
})
|
|
}, 1100);
|
|
});
|
|
}, reason => done(reason));
|
|
}, reason => done(reason));
|
|
});
|
|
});
|