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"