|
1 | 1 | // @ts-nocheck |
2 | 2 | import { SSEHandlerFactory } from '..'; |
3 | | -import { PUSH_SUBSYSTEM_UP, PUSH_NONRETRYABLE_ERROR, PUSH_SUBSYSTEM_DOWN, PUSH_RETRYABLE_ERROR, SEGMENT_UPDATE, SPLIT_KILL, SPLIT_UPDATE, MEMBERSHIPS_MS_UPDATE, MEMBERSHIPS_LS_UPDATE, ControlType } from '../../constants'; |
| 3 | +import { PUSH_SUBSYSTEM_UP, PUSH_NONRETRYABLE_ERROR, PUSH_SUBSYSTEM_DOWN, PUSH_RETRYABLE_ERROR, SEGMENT_UPDATE, SPLIT_KILL, SPLIT_UPDATE, RB_SEGMENT_UPDATE, MEMBERSHIPS_MS_UPDATE, MEMBERSHIPS_LS_UPDATE, ControlType } from '../../constants'; |
4 | 4 | import { loggerMock } from '../../../../logger/__tests__/sdkLogger.mock'; |
5 | 5 |
|
6 | 6 | // update messages |
7 | 7 | import splitUpdateMessage from '../../../../__tests__/mocks/message.SPLIT_UPDATE.1457552620999.json'; |
| 8 | +import rbsegmentUpdateMessage from '../../../../__tests__/mocks/message.RB_SEGMENT_UPDATE.1457552620999.json'; |
8 | 9 | import splitKillMessage from '../../../../__tests__/mocks/message.SPLIT_KILL.1457552650000.json'; |
9 | 10 | import segmentUpdateMessage from '../../../../__tests__/mocks/message.SEGMENT_UPDATE.1457552640000.json'; |
10 | 11 |
|
@@ -144,6 +145,10 @@ test('`handlerMessage` for update notifications (NotificationProcessor) and stre |
144 | 145 | sseHandler.handleMessage(splitUpdateMessage); |
145 | 146 | expect(pushEmitter.emit).toHaveBeenLastCalledWith(SPLIT_UPDATE, ...expectedParams); // must emit SPLIT_UPDATE with the message change number |
146 | 147 |
|
| 148 | + expectedParams = [{ type: 'RB_SEGMENT_UPDATE', changeNumber: 1457552620999 }]; |
| 149 | + sseHandler.handleMessage(rbsegmentUpdateMessage); |
| 150 | + expect(pushEmitter.emit).toHaveBeenLastCalledWith(RB_SEGMENT_UPDATE, ...expectedParams); // must emit RB_SEGMENT_UPDATE with the message change number |
| 151 | + |
147 | 152 | expectedParams = [{ type: 'SPLIT_KILL', changeNumber: 1457552650000, splitName: 'whitelist', defaultTreatment: 'not_allowed' }]; |
148 | 153 | sseHandler.handleMessage(splitKillMessage); |
149 | 154 | expect(pushEmitter.emit).toHaveBeenLastCalledWith(SPLIT_KILL, ...expectedParams); // must emit SPLIT_KILL with the message change number, split name and default treatment |
|
0 commit comments