Browse Source

fix: lint errors

pull/1521/head
phi 7 months ago
parent
commit
6c899d7425
  1. 12
      tests/async/test_client.py
  2. 12
      tests/async/test_server.py
  3. 6
      tests/common/test_client.py
  4. 6
      tests/common/test_server.py

12
tests/async/test_client.py

@ -1248,7 +1248,7 @@ class TestAsyncClient:
args = {"foo": "bar"}
c = async_client.AsyncClient(serializer_args=args)
assert c.packet_class_args == args
def test_serializer_args_with_msgpack(self):
def default(o):
if isinstance(o, datetime):
@ -1256,7 +1256,8 @@ class TestAsyncClient:
raise TypeError("Unknown type")
args = {"dumps_default": default}
data = {"current": datetime.now(timezone(timedelta(0)))}
c = async_client.AsyncClient(serializer='msgpack', serializer_args=args)
c = async_client.AsyncClient(serializer='msgpack',
serializer_args=args)
p = c._create_packet(data=data)
p2 = c._create_packet(encoded_packet=p.encode())
@ -1265,9 +1266,10 @@ class TestAsyncClient:
assert "current" in p2.data
assert isinstance(p2.data["current"], str)
assert default(data["current"]) == p2.data["current"]
def test_invalid_serializer_args(self):
args = {"invalid_arg": 123}
c = async_client.AsyncClient(serializer='msgpack', serializer_args=args)
c = async_client.AsyncClient(serializer='msgpack',
serializer_args=args)
with pytest.raises(TypeError):
c._create_packet(data={"foo": "bar"}).encode()
c._create_packet(data={"foo": "bar"}).encode()

12
tests/async/test_server.py

@ -1095,7 +1095,7 @@ class TestAsyncServer:
args = {"foo": "bar"}
s = async_server.AsyncServer(serializer_args=args)
assert s.packet_class_args == args
def test_serializer_args_with_msgpack(self, eio):
def default(o):
if isinstance(o, datetime):
@ -1103,7 +1103,8 @@ class TestAsyncServer:
raise TypeError("Unknown type")
args = {"dumps_default": default}
data = {"current": datetime.now(timezone(timedelta(0)))}
s = async_server.AsyncServer(serializer='msgpack', serializer_args=args)
s = async_server.AsyncServer(serializer='msgpack',
serializer_args=args)
p = s._create_packet(data=data)
p2 = s._create_packet(encoded_packet=p.encode())
@ -1112,9 +1113,10 @@ class TestAsyncServer:
assert "current" in p2.data
assert isinstance(p2.data["current"], str)
assert default(data["current"]) == p2.data["current"]
def test_invalid_serializer_args(self, eio):
args = {"invalid_arg": 123}
s = async_server.AsyncServer(serializer='msgpack', serializer_args=args)
s = async_server.AsyncServer(serializer='msgpack',
serializer_args=args)
with pytest.raises(TypeError):
s._create_packet(data={"foo": "bar"}).encode()
s._create_packet(data={"foo": "bar"}).encode()

6
tests/common/test_client.py

@ -1392,7 +1392,7 @@ class TestClient:
args = {"foo": "bar"}
c = client.Client(serializer_args=args)
assert c.packet_class_args == args
def test_serializer_args_with_msgpack(self):
def default(o):
if isinstance(o, datetime):
@ -1409,9 +1409,9 @@ class TestClient:
assert "current" in p2.data
assert isinstance(p2.data["current"], str)
assert default(data["current"]) == p2.data["current"]
def test_invalid_serializer_args(self):
args = {"invalid_arg": 123}
c = client.Client(serializer='msgpack', serializer_args=args)
with pytest.raises(TypeError):
c._create_packet(data={"foo": "bar"}).encode()
c._create_packet(data={"foo": "bar"}).encode()

6
tests/common/test_server.py

@ -1038,7 +1038,7 @@ class TestServer:
args = {"foo": "bar"}
s = server.Server(serializer_args=args)
assert s.packet_class_args == args
def test_serializer_args_with_msgpack(self, eio):
def default(o):
if isinstance(o, datetime):
@ -1055,9 +1055,9 @@ class TestServer:
assert "current" in p2.data
assert isinstance(p2.data["current"], str)
assert default(data["current"]) == p2.data["current"]
def test_invalid_serializer_args(self, eio):
args = {"invalid_arg": 123}
s = server.Server(serializer='msgpack', serializer_args=args)
with pytest.raises(TypeError):
s._create_packet(data={"foo": "bar"}).encode()
s._create_packet(data={"foo": "bar"}).encode()

Loading…
Cancel
Save