text/src/tests/services/WebsocketPolyfill.spec.js at 78f146197b7ca2fed41c6d3a7803033702f88ce3 · nextcloud/text · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
import initWebSocketPolyfill from '../../services/WebSocketPolyfill.js'

describe('Init function', () => {

	it('returns a websocket polyfill class', () => {
		const syncService = { on: jest.fn(), open: jest.fn(() => Promise.resolve({ version: 123, session: {} })) }
		const Polyfill = initWebSocketPolyfill(syncService)
		const websocket = new Polyfill('url')
		expect(websocket).toBeInstanceOf(Polyfill)
	})

	it('registers handlers', () => {
		const syncService = { on: jest.fn(), open: jest.fn(() => Promise.resolve({ version: 123, session: {} })) }
		const Polyfill = initWebSocketPolyfill(syncService)
		const websocket = new Polyfill('url')
		expect(syncService.on).toHaveBeenCalled()
	})

	it('opens sync service', () => {
		const syncService = { on: jest.fn(), open: jest.fn(() => Promise.resolve({ version: 123, session: {} })) }
		const fileId = 123
		const initialSession = { }
		const Polyfill = initWebSocketPolyfill(syncService, fileId, initialSession)
		const websocket = new Polyfill('url')
		expect(syncService.open).toHaveBeenCalledWith({ fileId, initialSession })
	})

	it('sends steps to sync service', async () => {
		const syncService = {
			on: jest.fn(),
			open: jest.fn(() => Promise.resolve({ version: 123, session: {} })),
			sendSteps: async getData => getData(),
		}
		const queue = [ 'initial' ]
		const data = { dummy: 'data' }
		const Polyfill = initWebSocketPolyfill(syncService, null, null, queue)
		const websocket = new Polyfill('url')
		const result = websocket.send(data)
		expect(result).toBeInstanceOf(Promise)
		expect(queue).toEqual([ 'initial' , data ])
		const dataSendOut = await result
		expect(queue).toEqual([])
		expect(dataSendOut).toHaveProperty('awareness')
		expect(dataSendOut).toHaveProperty('steps')
		expect(dataSendOut).toHaveProperty('version')
	})

	it('handles early reject', async () => {
		jest.spyOn(console, 'error').mockImplementation(() => {})
		const syncService = {
			on: jest.fn(),
			open: jest.fn(() => Promise.resolve({ version: 123, session: {} })),
			sendSteps: jest.fn().mockRejectedValue('error before reading steps in sync service'),
		}
		const queue = [ 'initial' ]
		const data = { dummy: 'data' }
		const Polyfill = initWebSocketPolyfill(syncService, null, null, queue)
		const websocket = new Polyfill('url')
		const result = websocket.send(data)
		expect(queue).toEqual([ 'initial' , data ])
		expect(result).toBeInstanceOf(Promise)
		const returned = await result
		expect(returned).toBeUndefined()
		expect(queue).toEqual([ 'initial' , data ])
	})

	it('handles reject after reading data', async () => {
		jest.spyOn(console, 'error').mockImplementation(() => {})
		const syncService = {
			on: jest.fn(),
			open: jest.fn(() => Promise.resolve({ version: 123, session: {} })),
			sendSteps: jest.fn().mockImplementation( async getData => {
				getData()
				throw 'error when sending in sync service'
			}),
		}
		const queue = [ 'initial' ]
		const data = { dummy: 'data' }
		const Polyfill = initWebSocketPolyfill(syncService, null, null, queue)
		const websocket = new Polyfill('url')
		const result = websocket.send(data)
		expect(queue).toEqual([ 'initial' , data ])
		expect(result).toBeInstanceOf(Promise)
		const returned = await result
		expect(returned).toBeUndefined()
		expect(queue).toEqual([ 'initial' , data ])
	})

	it('queue survives a close', async () => {
		jest.spyOn(console, 'error').mockImplementation(() => {})
		const syncService = {
			on: jest.fn(),
			open: jest.fn(() => Promise.resolve({ version: 123, session: {} })),
			sendSteps: jest.fn().mockImplementation( async getData => {
				getData()
				throw 'error when sending in sync service'
			}),
			sendStepsNow: jest.fn().mockImplementation( async getData => {
				getData()
				throw 'sendStepsNow error when sending'
			}),
			off: jest.fn(),
			close: jest.fn( async data => data ),
		}
		const queue = [ 'initial' ]
		const data = { dummy: 'data' }
		const Polyfill = initWebSocketPolyfill(syncService, null, null, queue)
		const websocket = new Polyfill('url')
		websocket.onclose = jest.fn()
		await websocket.send(data)
		const promise = websocket.close()
		expect(queue).toEqual([ 'initial' , data ])
		await promise
		expect(queue).toEqual([ 'initial' , data ])
	})

})