157 lines
7.3 KiB
Python
157 lines
7.3 KiB
Python
import unittest
|
|
|
|
from minecraft import SUPPORTED_PROTOCOL_VERSIONS
|
|
from minecraft.networking.connection import ConnectionContext
|
|
from minecraft.networking import packets
|
|
from minecraft.networking import types
|
|
from minecraft.networking.packets import clientbound
|
|
from minecraft.networking.packets import serverbound
|
|
|
|
|
|
class LegacyPacketNamesTest(unittest.TestCase):
|
|
def test_legacy_packets_equal_current_packets(self):
|
|
self.assertEqual(packets.KeepAlivePacket,
|
|
packets.AbstractKeepAlivePacket)
|
|
|
|
self.assertEqual(packets.state_handshake_clientbound,
|
|
clientbound.handshake.get_packets)
|
|
self.assertEqual(packets.HandShakePacket,
|
|
serverbound.handshake.HandShakePacket)
|
|
self.assertEqual(packets.state_handshake_serverbound,
|
|
serverbound.handshake.get_packets)
|
|
|
|
self.assertEqual(packets.ResponsePacket,
|
|
clientbound.status.ResponsePacket)
|
|
self.assertEqual(packets.PingPacketResponse,
|
|
clientbound.status.PingResponsePacket)
|
|
self.assertEqual(packets.state_status_clientbound,
|
|
clientbound.status.get_packets)
|
|
self.assertEqual(packets.RequestPacket,
|
|
serverbound.status.RequestPacket)
|
|
self.assertEqual(packets.PingPacket,
|
|
serverbound.status.PingPacket)
|
|
self.assertEqual(packets.state_status_serverbound,
|
|
serverbound.status.get_packets)
|
|
|
|
self.assertEqual(packets.DisconnectPacket,
|
|
clientbound.login.DisconnectPacket)
|
|
self.assertEqual(packets.EncryptionRequestPacket,
|
|
clientbound.login.EncryptionRequestPacket)
|
|
self.assertEqual(packets.LoginSuccessPacket,
|
|
clientbound.login.LoginSuccessPacket)
|
|
self.assertEqual(packets.SetCompressionPacket,
|
|
clientbound.login.SetCompressionPacket)
|
|
self.assertEqual(packets.state_login_clientbound,
|
|
clientbound.login.get_packets)
|
|
self.assertEqual(packets.LoginStartPacket,
|
|
serverbound.login.LoginStartPacket)
|
|
self.assertEqual(packets.EncryptionResponsePacket,
|
|
serverbound.login.EncryptionResponsePacket)
|
|
self.assertEqual(packets.state_login_serverbound,
|
|
serverbound.login.get_packets)
|
|
|
|
self.assertEqual(packets.KeepAlivePacketClientbound,
|
|
clientbound.play.KeepAlivePacket)
|
|
self.assertEqual(packets.KeepAlivePacketServerbound,
|
|
serverbound.play.KeepAlivePacket)
|
|
self.assertEqual(packets.JoinGamePacket,
|
|
clientbound.play.JoinGamePacket)
|
|
self.assertEqual(packets.ChatMessagePacket,
|
|
clientbound.play.ChatMessagePacket)
|
|
self.assertEqual(packets.PlayerPositionAndLookPacket,
|
|
clientbound.play.PlayerPositionAndLookPacket)
|
|
self.assertEqual(packets.DisconnectPacketPlayState,
|
|
clientbound.play.DisconnectPacket)
|
|
self.assertEqual(packets.SetCompressionPacketPlayState,
|
|
clientbound.play.SetCompressionPacket)
|
|
self.assertEqual(packets.PlayerListItemPacket,
|
|
clientbound.play.PlayerListItemPacket)
|
|
self.assertEqual(packets.MapPacket,
|
|
clientbound.play.MapPacket)
|
|
self.assertEqual(packets.state_playing_clientbound,
|
|
clientbound.play.get_packets)
|
|
self.assertEqual(packets.ChatPacket,
|
|
serverbound.play.ChatPacket)
|
|
self.assertEqual(packets.PositionAndLookPacket,
|
|
serverbound.play.PositionAndLookPacket)
|
|
self.assertEqual(packets.TeleportConfirmPacket,
|
|
serverbound.play.TeleportConfirmPacket)
|
|
self.assertEqual(packets.AnimationPacketServerbound,
|
|
serverbound.play.AnimationPacket)
|
|
self.assertEqual(packets.state_playing_serverbound,
|
|
serverbound.play.get_packets)
|
|
|
|
|
|
class ClassMemberAliasesTest(unittest.TestCase):
|
|
def test_alias_values(self):
|
|
self.assertEqual(serverbound.play.AnimationPacket.HAND_MAIN,
|
|
types.RelativeHand.MAIN)
|
|
self.assertEqual(serverbound.play.AnimationPacket.HAND_OFF,
|
|
types.RelativeHand.OFF)
|
|
|
|
self.assertEqual(serverbound.play.ClientSettingsPacket.Hand.LEFT,
|
|
types.AbsoluteHand.LEFT)
|
|
self.assertEqual(serverbound.play.ClientSettingsPacket.Hand.RIGHT,
|
|
types.AbsoluteHand.RIGHT)
|
|
|
|
def test_block_change_packet(self):
|
|
context = ConnectionContext()
|
|
context.protocol_version = SUPPORTED_PROTOCOL_VERSIONS[-1]
|
|
bi, bm = 358, 9
|
|
packet = clientbound.play.BlockChangePacket(blockId=bi, blockMeta=bm)
|
|
self.assertEqual((packet.blockId, packet.blockMeta), (bi, bm))
|
|
self.assertEqual(packet.blockStateId, packet.block_state_id)
|
|
|
|
def test_join_game_packet(self):
|
|
GameMode = types.GameMode
|
|
context = ConnectionContext()
|
|
for pure_game_mode in (GameMode.SURVIVAL, GameMode.CREATIVE,
|
|
GameMode.ADVENTURE, GameMode.SPECTATOR):
|
|
for is_hardcore in (False, True):
|
|
context.protocol_version = 70
|
|
game_mode = \
|
|
pure_game_mode | GameMode.HARDCORE \
|
|
if is_hardcore else pure_game_mode
|
|
|
|
packet = clientbound.play.JoinGamePacket()
|
|
packet.game_mode = game_mode
|
|
packet.context = context
|
|
self.assertEqual(packet.pure_game_mode, pure_game_mode)
|
|
self.assertEqual(packet.is_hardcore, is_hardcore)
|
|
|
|
del packet.context
|
|
del packet.is_hardcore
|
|
packet.context = context
|
|
self.assertEqual(packet.game_mode, packet.pure_game_mode)
|
|
|
|
del packet.context
|
|
del packet.game_mode
|
|
packet.context = context
|
|
self.assertFalse(hasattr(packet, 'is_hardcore'))
|
|
|
|
packet = clientbound.play.JoinGamePacket()
|
|
packet.pure_game_mode = game_mode
|
|
packet.is_hardcore = is_hardcore
|
|
packet.context = context
|
|
self.assertEqual(packet.game_mode, game_mode)
|
|
|
|
context.protocol_version = 738
|
|
game_mode = pure_game_mode | GameMode.HARDCORE
|
|
|
|
packet = clientbound.play.JoinGamePacket()
|
|
packet.game_mode = game_mode
|
|
packet.is_hardcore = is_hardcore
|
|
packet.context = context
|
|
self.assertEqual(packet.game_mode, game_mode)
|
|
self.assertEqual(packet.pure_game_mode, game_mode)
|
|
self.assertEqual(packet.is_hardcore, is_hardcore)
|
|
|
|
del packet.context
|
|
packet.is_hardcore = is_hardcore
|
|
packet.context = context
|
|
self.assertEqual(packet.game_mode, game_mode)
|
|
self.assertEqual(packet.pure_game_mode, game_mode)
|
|
|
|
with self.assertRaises(AttributeError):
|
|
del packet.pure_game_mode
|