@@ -25,7 +25,8 @@ describe('E2E WebSocket proxy', () => {
25
25
wss = new WebSocketServer ( { port : WS_SERVER_PORT } ) ;
26
26
27
27
wss . on ( 'connection' , ( websocket ) => {
28
- websocket . on ( 'message' , ( message ) => {
28
+ websocket . on ( 'message' , ( data , isBinary ) => {
29
+ const message = isBinary ? data : data . toString ( ) ;
29
30
websocket . send ( message ) ; // echo received message
30
31
} ) ;
31
32
} ) ;
@@ -66,7 +67,8 @@ describe('E2E WebSocket proxy', () => {
66
67
} ) ;
67
68
68
69
it ( 'should proxy to path' , ( done ) => {
69
- ws . on ( 'message' , ( message ) => {
70
+ ws . on ( 'message' , ( data , isBinary ) => {
71
+ const message = isBinary ? data : data . toString ( ) ;
70
72
expect ( message ) . toBe ( 'foobar' ) ;
71
73
done ( ) ;
72
74
} ) ;
@@ -84,7 +86,8 @@ describe('E2E WebSocket proxy', () => {
84
86
} ) ;
85
87
86
88
it ( 'should proxy to path' , ( done ) => {
87
- ws . on ( 'message' , ( message ) => {
89
+ ws . on ( 'message' , ( data , isBinary ) => {
90
+ const message = isBinary ? data : data . toString ( ) ;
88
91
expect ( message ) . toBe ( 'foobar' ) ;
89
92
done ( ) ;
90
93
} ) ;
@@ -109,7 +112,8 @@ describe('E2E WebSocket proxy', () => {
109
112
} ) ;
110
113
111
114
it ( 'should proxy to path' , ( done ) => {
112
- ws . on ( 'message' , ( message ) => {
115
+ ws . on ( 'message' , ( data , isBinary ) => {
116
+ const message = isBinary ? data : data . toString ( ) ;
113
117
expect ( message ) . toBe ( 'foobar' ) ;
114
118
done ( ) ;
115
119
} ) ;
@@ -136,7 +140,8 @@ describe('E2E WebSocket proxy', () => {
136
140
} ) ;
137
141
138
142
it ( 'should proxy to path' , ( done ) => {
139
- ws . on ( 'message' , ( message ) => {
143
+ ws . on ( 'message' , ( data , isBinary ) => {
144
+ const message = isBinary ? data : data . toString ( ) ;
140
145
expect ( message ) . toBe ( 'foobar' ) ;
141
146
done ( ) ;
142
147
} ) ;
0 commit comments