@@ -205,30 +205,24 @@ def _make_handshake(
205
205
def test_handshake () -> None :
206
206
response , nonce = _make_handshake ([])
207
207
208
- response .headers = sorted (response .headers ) # For test determinism
209
- assert response == h11 .InformationalResponse (
210
- status_code = 101 ,
211
- headers = [
212
- (b"connection" , b"Upgrade" ),
213
- (b"sec-websocket-accept" , generate_accept_token (nonce )),
214
- (b"upgrade" , b"WebSocket" ),
215
- ],
216
- )
208
+ assert response .status_code == 101
209
+ assert sorted (response .headers ) == [
210
+ (b"connection" , b"Upgrade" ),
211
+ (b"sec-websocket-accept" , generate_accept_token (nonce )),
212
+ (b"upgrade" , b"WebSocket" ),
213
+ ]
217
214
218
215
219
216
def test_handshake_extra_headers () -> None :
220
217
response , nonce = _make_handshake ([], accept_headers = [(b"X-Foo" , b"bar" )])
221
218
222
- response .headers = sorted (response .headers ) # For test determinism
223
- assert response == h11 .InformationalResponse (
224
- status_code = 101 ,
225
- headers = [
226
- (b"connection" , b"Upgrade" ),
227
- (b"sec-websocket-accept" , generate_accept_token (nonce )),
228
- (b"upgrade" , b"WebSocket" ),
229
- (b"x-foo" , b"bar" ),
230
- ],
231
- )
219
+ assert response .status_code == 101
220
+ assert sorted (response .headers ) == [
221
+ (b"connection" , b"Upgrade" ),
222
+ (b"sec-websocket-accept" , generate_accept_token (nonce )),
223
+ (b"upgrade" , b"WebSocket" ),
224
+ (b"x-foo" , b"bar" ),
225
+ ]
232
226
233
227
234
228
@pytest .mark .parametrize ("accept_subprotocol" , ["one" , "two" ])
0 commit comments