Add test coverage for @listener.

This commit is contained in:
joodicator 2019-05-13 19:04:35 +02:00
parent 24ca96accb
commit 41ea36c642

View File

@ -224,26 +224,23 @@ class EarlyPacketListenerTest(ConnectTest):
the early packet listener is registered afterwards. the early packet listener is registered afterwards.
""" """
def _start_client(self, client): def _start_client(self, client):
@client.listener(clientbound.play.JoinGamePacket)
def handle_join(packet): def handle_join(packet):
assert early_handle_join.called, \ assert early_handle_join.called, \
'Ordinary listener called before early listener.' 'Ordinary listener called before early listener.'
handle_join.called = True handle_join.called = True
handle_join.called = False handle_join.called = False
client.register_packet_listener(
handle_join, clientbound.play.JoinGamePacket)
@client.listener(clientbound.play.JoinGamePacket, early=True)
def early_handle_join(packet): def early_handle_join(packet):
early_handle_join.called = True early_handle_join.called = True
client.register_packet_listener(
early_handle_join, clientbound.play.JoinGamePacket, early=True)
early_handle_join.called = False early_handle_join.called = False
@client.listener(clientbound.play.DisconnectPacket)
def handle_disconnect(packet): def handle_disconnect(packet):
assert early_handle_join.called, 'Early listener not called.' assert early_handle_join.called, 'Early listener not called.'
assert handle_join.called, 'Ordinary listener not called.' assert handle_join.called, 'Ordinary listener not called.'
raise fake_server.FakeServerTestSuccess raise fake_server.FakeServerTestSuccess
client.register_packet_listener(
handle_disconnect, clientbound.play.DisconnectPacket)
client.connect() client.connect()