improve test

This commit is contained in:
syuilo 2022-01-02 02:11:25 +09:00
parent 6c990f3ce7
commit a05f47d5cc
1 changed files with 17 additions and 5 deletions

View File

@ -10,7 +10,10 @@ describe('Streaming', () => {
main.on('meUpdated', payload => { main.on('meUpdated', payload => {
mainChannelReceived.push(payload); mainChannelReceived.push(payload);
}); });
await server.connected;
const ws = await server.connected;
expect(new URLSearchParams(new URL(ws.url).search).get('i')).toEqual('TOKEN');
const msg = JSON.parse(await server.nextMessage as string); const msg = JSON.parse(await server.nextMessage as string);
const mainChannelId = msg.body.id; const mainChannelId = msg.body.id;
expect(msg.type).toEqual('connect'); expect(msg.type).toEqual('connect');
@ -44,7 +47,10 @@ describe('Streaming', () => {
messaging.on('message', payload => { messaging.on('message', payload => {
messagingChannelReceived.push(payload); messagingChannelReceived.push(payload);
}); });
await server.connected;
const ws = await server.connected;
expect(new URLSearchParams(new URL(ws.url).search).get('i')).toEqual('TOKEN');
const msg = JSON.parse(await server.nextMessage as string); const msg = JSON.parse(await server.nextMessage as string);
const messagingChannelId = msg.body.id; const messagingChannelId = msg.body.id;
expect(msg.type).toEqual('connect'); expect(msg.type).toEqual('connect');
@ -78,7 +84,8 @@ describe('Streaming', () => {
stream.useChannel('messaging', { otherparty: 'aaa' }); stream.useChannel('messaging', { otherparty: 'aaa' });
stream.useChannel('messaging', { otherparty: 'bbb' }); stream.useChannel('messaging', { otherparty: 'bbb' });
await server.connected; const ws = await server.connected;
expect(new URLSearchParams(new URL(ws.url).search).get('i')).toEqual('TOKEN');
const msg = JSON.parse(await server.nextMessage as string); const msg = JSON.parse(await server.nextMessage as string);
const messagingChannelId = msg.body.id; const messagingChannelId = msg.body.id;
@ -100,7 +107,9 @@ describe('Streaming', () => {
const messaging = stream.useChannel('messaging', { otherparty: 'aaa' }); const messaging = stream.useChannel('messaging', { otherparty: 'aaa' });
messaging.send('read', { id: 'aaa' }); messaging.send('read', { id: 'aaa' });
await server.connected; const ws = await server.connected;
expect(new URLSearchParams(new URL(ws.url).search).get('i')).toEqual('TOKEN');
const connectMsg = JSON.parse(await server.nextMessage as string); const connectMsg = JSON.parse(await server.nextMessage as string);
const channelId = connectMsg.body.id; const channelId = connectMsg.body.id;
const msg = JSON.parse(await server.nextMessage as string); const msg = JSON.parse(await server.nextMessage as string);
@ -122,7 +131,10 @@ describe('Streaming', () => {
main.on('meUpdated', payload => { main.on('meUpdated', payload => {
mainChannelReceived.push(payload); mainChannelReceived.push(payload);
}); });
await server.connected;
const ws = await server.connected;
expect(new URLSearchParams(new URL(ws.url).search).get('i')).toEqual('TOKEN');
const msg = JSON.parse(await server.nextMessage as string); const msg = JSON.parse(await server.nextMessage as string);
const mainChannelId = msg.body.id; const mainChannelId = msg.body.id;
expect(msg.type).toEqual('connect'); expect(msg.type).toEqual('connect');