amirho/nest
1
0
Fork 0
mirror of https://github.com/nestjs/nest.git synced 2026-02-21 23:11:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(microservices): add static headers for ClientNats

Browse Source
This commit is contained in:
Rick Dutour Geerling
2021-09-07 10:33:20 +02:00
parent f4002e493b
commit 7fed9a9e64
3 changed files with 129 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

113
packages/microservices/test/client/client-nats.spec.ts
View File

@@ -1,16 +1,16 @@
import { expect } from 'chai';
import { JSONCodec, headers as createHeaders } from 'nats';
import { headers as createHeaders, JSONCodec } from 'nats';
import * as sinon from 'sinon';
import { ClientNats } from '../../client/client-nats';
import { ReadPacket } from '../../interfaces';
import { NatsRecord } from '../../record-builders';
describe('ClientNats', () => {
const client = new ClientNats({});
let client: ClientNats;
describe('publish', () => {
let msg: ReadPacket;
const pattern = 'test';
const headers = createHeaders();
headers.set('1', '123');
const msg = { pattern, data: { headers, value: 'data' } };
const id = 3;
let subscribeSpy: sinon.SinonSpy,
@@ -23,6 +23,8 @@ describe('ClientNats', () => {
createClient: sinon.SinonStub;
beforeEach(() => {
client = new ClientNats({});
msg = { pattern, data: 'data' };
unsubscribeSpy = sinon.spy();
subscription = {
unsubscribe: unsubscribeSpy,
@@ -96,7 +98,54 @@ describe('ClientNats', () => {
expect(unsubscribeSpy.called).to.be.true;
});
});
describe('headers', () => {
it('should not generate headers if none are configured', () => {
client['publish'](msg, () => {});
expect(natsClient.publish.getCall(0).args[2].headers).to.be.undefined;
});
it('should send packet headers', () => {
const requestHeaders = createHeaders();
requestHeaders.set('1', '123');
msg.data = new NatsRecord('data', requestHeaders);
client['publish'](msg, () => {});
expect(natsClient.publish.getCall(0).args[2].headers.get('1')).to.eql(
'123',
);
});
it('should combine packet and static headers', () => {
const staticHeaders = { 'client-id': 'some-client-id' };
(client as any).options.headers = staticHeaders;
const requestHeaders = createHeaders();
requestHeaders.set('1', '123');
msg.data = new NatsRecord('data', requestHeaders);
client['publish'](msg, () => {});
expect(publishSpy.getCall(0).args[2].headers.get('client-id')).to.eql(
'some-client-id',
);
expect(publishSpy.getCall(0).args[2].headers.get('1')).to.eql('123');
});
it('should prefer packet headers over static headers', () => {
const staticHeaders = { 'client-id': 'some-client-id' };
(client as any).options.headers = staticHeaders;
const requestHeaders = createHeaders();
requestHeaders.set('client-id', 'override-client-id');
msg.data = new NatsRecord('data', requestHeaders);
client['publish'](msg, () => {});
expect(publishSpy.getCall(0).args[2].headers.get('client-id')).to.eql(
'override-client-id',
);
});
});
});
describe('createSubscriptionHandler', () => {
const pattern = 'test';
const msg = { pattern, data: 'data', id: '1' };
@@ -279,10 +328,12 @@ describe('ClientNats', () => {
});
});
describe('dispatchEvent', () => {
const msg = { pattern: 'pattern', data: 'data' };
let msg: ReadPacket;
let subscribeStub: sinon.SinonStub, natsClient: any;
beforeEach(() => {
client = new ClientNats({});
msg = { pattern: 'pattern', data: 'data' };
subscribeStub = sinon
.stub()
.callsFake((channel, options) => options.callback());
@@ -298,6 +349,7 @@ describe('ClientNats', () => {
expect(natsClient.publish.called).to.be.true;
});
it('should throw error', async () => {
subscribeStub.callsFake((channel, options) =>
options.callback(new Error()),
@@ -306,5 +358,54 @@ describe('ClientNats', () => {
expect(err).to.be.instanceOf(Error),
);
});
describe('headers', () => {
it('should not generate headers if none are configured', () => {
client['dispatchEvent'](msg);
expect(natsClient.publish.getCall(0).args[2].headers).to.be.undefined;
});
it('should send packet headers', () => {
const requestHeaders = createHeaders();
requestHeaders.set('1', '123');
msg.data = new NatsRecord('data', requestHeaders);
client['dispatchEvent'](msg);
expect(natsClient.publish.getCall(0).args[2].headers.get('1')).to.eql(
'123',
);
});
it('should combine packet and static headers', () => {
const staticHeaders = { 'client-id': 'some-client-id' };
(client as any).options.headers = staticHeaders;
const requestHeaders = createHeaders();
requestHeaders.set('1', '123');
msg.data = new NatsRecord('data', requestHeaders);
client['dispatchEvent'](msg);
expect(
natsClient.publish.getCall(0).args[2].headers.get('client-id'),
).to.eql('some-client-id');
expect(natsClient.publish.getCall(0).args[2].headers.get('1')).to.eql(
'123',
);
});
it('should prefer packet headers over static headers', () => {
const staticHeaders = { 'client-id': 'some-client-id' };
(client as any).options.headers = staticHeaders;
const requestHeaders = createHeaders();
requestHeaders.set('client-id', 'override-client-id');
msg.data = new NatsRecord('data', requestHeaders);
client['dispatchEvent'](msg);
expect(
natsClient.publish.getCall(0).args[2].headers.get('client-id'),
).to.eql('override-client-id');
});
});
});
});