Jerko Steiner
92a5f2063c
This should not cause compile issues when simple-peer is not installed (because client-side JS will be precompiled after pkg is published in NPM repository) because types are not used in runtime.
118 lines
2.8 KiB
TypeScript
118 lines
2.8 KiB
TypeScript
import { EventEmitter } from 'events'
|
|
import { Socket } from 'socket.io'
|
|
import { TypedIO } from '../shared'
|
|
import handleSocket from './socket'
|
|
|
|
describe('server/socket', () => {
|
|
type SocketMock = Socket & {
|
|
id: string
|
|
room?: string
|
|
join: jest.Mock
|
|
leave: jest.Mock
|
|
emit: jest.Mock
|
|
}
|
|
|
|
let socket: SocketMock
|
|
let io: TypedIO & {
|
|
in: jest.Mock<(room: string) => SocketMock>
|
|
to: jest.Mock<(room: string) => SocketMock>
|
|
}
|
|
let rooms: Record<string, {emit: any}>
|
|
beforeEach(() => {
|
|
socket = new EventEmitter() as SocketMock
|
|
socket.id = 'socket0'
|
|
socket.join = jest.fn()
|
|
socket.leave = jest.fn()
|
|
rooms = {}
|
|
|
|
io = {} as any
|
|
io.in = io.to = jest.fn().mockImplementation(room => {
|
|
return (rooms[room] = rooms[room] || {
|
|
emit: jest.fn(),
|
|
})
|
|
})
|
|
|
|
io.sockets = {
|
|
adapter: {
|
|
rooms: {
|
|
room1: {
|
|
socket0: true,
|
|
} as any,
|
|
room2: {
|
|
socket0: true,
|
|
} as any,
|
|
room3: {
|
|
sockets: {
|
|
'socket0': true,
|
|
'socket1': true,
|
|
'socket2': true,
|
|
},
|
|
} as any,
|
|
},
|
|
} as any,
|
|
} as any
|
|
|
|
socket.leave = jest.fn()
|
|
socket.join = jest.fn()
|
|
})
|
|
|
|
it('should be a function', () => {
|
|
expect(typeof handleSocket).toBe('function')
|
|
})
|
|
|
|
describe('socket events', () => {
|
|
beforeEach(() => handleSocket(socket, io))
|
|
|
|
describe('signal', () => {
|
|
it('should broadcast signal to specific user', () => {
|
|
const signal = { type: 'signal' }
|
|
|
|
socket.emit('signal', { userId: 'a', signal })
|
|
|
|
expect(io.to.mock.calls).toEqual([[ 'a' ]])
|
|
expect((io.to('a').emit as jest.Mock).mock.calls).toEqual([[
|
|
'signal', {
|
|
userId: 'socket0',
|
|
signal,
|
|
},
|
|
]])
|
|
})
|
|
})
|
|
|
|
describe('ready', () => {
|
|
it('should call socket.leave if socket.room', () => {
|
|
socket.room = 'room1'
|
|
socket.emit('ready', 'room2')
|
|
|
|
expect(socket.leave.mock.calls).toEqual([[ 'room1' ]])
|
|
expect(socket.join.mock.calls).toEqual([[ 'room2' ]])
|
|
})
|
|
|
|
it('should call socket.join to room', () => {
|
|
socket.emit('ready', 'room3')
|
|
expect(socket.join.mock.calls).toEqual([[ 'room3' ]])
|
|
})
|
|
|
|
it('should emit users', () => {
|
|
socket.emit('ready', 'room3')
|
|
|
|
expect(io.to.mock.calls).toEqual([[ 'room3' ]])
|
|
expect((io.to('room3').emit as jest.Mock).mock.calls).toEqual([
|
|
[
|
|
'users', {
|
|
initiator: 'socket0',
|
|
users: [{
|
|
id: 'socket0',
|
|
}, {
|
|
id: 'socket1',
|
|
}, {
|
|
id: 'socket2',
|
|
}],
|
|
},
|
|
],
|
|
])
|
|
})
|
|
})
|
|
})
|
|
})
|