def test_pickle(self) -> None:
err = WebSocketError(WSCloseCode.PROTOCOL_ERROR, "Something invalid")
err.foo = "bar"
for proto in range(pickle.HIGHEST_PROTOCOL + 1):
pickled = pickle.dumps(err, proto)
err2 = pickle.loads(pickled)
assert err2.code == WSCloseCode.PROTOCOL_ERROR
assert str(err2) == "Something invalid"
assert err2.foo == "bar"
def test_ctor(self) -> None:
err = WebSocketError(WSCloseCode.PROTOCOL_ERROR, "Something invalid")
assert err.code == WSCloseCode.PROTOCOL_ERROR
assert str(err) == "Something invalid"