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"} args = {"foo": "bar"}
c = async_client.AsyncClient(serializer_args=args) c = async_client.AsyncClient(serializer_args=args)
assert c.packet_class_args == args assert c.packet_class_args == args
def test_serializer_args_with_msgpack(self): def test_serializer_args_with_msgpack(self):
def default(o): def default(o):
if isinstance(o, datetime): if isinstance(o, datetime):
@ -1256,7 +1256,8 @@ class TestAsyncClient:
raise TypeError("Unknown type") raise TypeError("Unknown type")
args = {"dumps_default": default} args = {"dumps_default": default}
data = {"current": datetime.now(timezone(timedelta(0)))} 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) p = c._create_packet(data=data)
p2 = c._create_packet(encoded_packet=p.encode()) p2 = c._create_packet(encoded_packet=p.encode())
@ -1265,9 +1266,10 @@ class TestAsyncClient:
assert "current" in p2.data assert "current" in p2.data
assert isinstance(p2.data["current"], str) assert isinstance(p2.data["current"], str)
assert default(data["current"]) == p2.data["current"] assert default(data["current"]) == p2.data["current"]
def test_invalid_serializer_args(self): def test_invalid_serializer_args(self):
args = {"invalid_arg": 123} 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): 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"} args = {"foo": "bar"}
s = async_server.AsyncServer(serializer_args=args) s = async_server.AsyncServer(serializer_args=args)
assert s.packet_class_args == args assert s.packet_class_args == args
def test_serializer_args_with_msgpack(self, eio): def test_serializer_args_with_msgpack(self, eio):
def default(o): def default(o):
if isinstance(o, datetime): if isinstance(o, datetime):
@ -1103,7 +1103,8 @@ class TestAsyncServer:
raise TypeError("Unknown type") raise TypeError("Unknown type")
args = {"dumps_default": default} args = {"dumps_default": default}
data = {"current": datetime.now(timezone(timedelta(0)))} 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) p = s._create_packet(data=data)
p2 = s._create_packet(encoded_packet=p.encode()) p2 = s._create_packet(encoded_packet=p.encode())
@ -1112,9 +1113,10 @@ class TestAsyncServer:
assert "current" in p2.data assert "current" in p2.data
assert isinstance(p2.data["current"], str) assert isinstance(p2.data["current"], str)
assert default(data["current"]) == p2.data["current"] assert default(data["current"]) == p2.data["current"]
def test_invalid_serializer_args(self, eio): def test_invalid_serializer_args(self, eio):
args = {"invalid_arg": 123} 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): 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"} args = {"foo": "bar"}
c = client.Client(serializer_args=args) c = client.Client(serializer_args=args)
assert c.packet_class_args == args assert c.packet_class_args == args
def test_serializer_args_with_msgpack(self): def test_serializer_args_with_msgpack(self):
def default(o): def default(o):
if isinstance(o, datetime): if isinstance(o, datetime):
@ -1409,9 +1409,9 @@ class TestClient:
assert "current" in p2.data assert "current" in p2.data
assert isinstance(p2.data["current"], str) assert isinstance(p2.data["current"], str)
assert default(data["current"]) == p2.data["current"] assert default(data["current"]) == p2.data["current"]
def test_invalid_serializer_args(self): def test_invalid_serializer_args(self):
args = {"invalid_arg": 123} args = {"invalid_arg": 123}
c = client.Client(serializer='msgpack', serializer_args=args) c = client.Client(serializer='msgpack', serializer_args=args)
with pytest.raises(TypeError): 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"} args = {"foo": "bar"}
s = server.Server(serializer_args=args) s = server.Server(serializer_args=args)
assert s.packet_class_args == args assert s.packet_class_args == args
def test_serializer_args_with_msgpack(self, eio): def test_serializer_args_with_msgpack(self, eio):
def default(o): def default(o):
if isinstance(o, datetime): if isinstance(o, datetime):
@ -1055,9 +1055,9 @@ class TestServer:
assert "current" in p2.data assert "current" in p2.data
assert isinstance(p2.data["current"], str) assert isinstance(p2.data["current"], str)
assert default(data["current"]) == p2.data["current"] assert default(data["current"]) == p2.data["current"]
def test_invalid_serializer_args(self, eio): def test_invalid_serializer_args(self, eio):
args = {"invalid_arg": 123} args = {"invalid_arg": 123}
s = server.Server(serializer='msgpack', serializer_args=args) s = server.Server(serializer='msgpack', serializer_args=args)
with pytest.raises(TypeError): with pytest.raises(TypeError):
s._create_packet(data={"foo": "bar"}).encode() s._create_packet(data={"foo": "bar"}).encode()

Loading…
Cancel
Save