def testHandleVersion(self): """Testing handleVersion""" assert self.u._protocol_ok == False, "_protocol_ok : False expected" # Messages should be empty, protocol is not established clear_all_messages() self.u._handleVersion() self.assertEquals(get_messages(), []) assert self.u._protocol_ok == False, "_protocol_ok change unexpected" self.u._packet = list('\n') # Messages should be empty, protocol is not established clear_all_messages() self.u._handleVersion() self.assertEquals(get_messages(), []) assert self.u.transport._loseConnection == True, "loseConnection not called" self.u.transport = FakeTransport() # transport re-init self.u._packet = list('CGI a.b\n') # Messages should be empty, protocol is not established clear_all_messages() self.u._handleVersion() self.assertEquals(get_messages(), []) assert self.u.transport._loseConnection == True, "loseConnection not called" self.u.transport = FakeTransport() # transport re-init vers = Version(protocol_number) PROTOCOL_MAJOR = "%03d" % vers.major() PROTOCOL_MINOR = "%d%02d" % (vers.medium(), vers.minor()) self.u._packet = list('CGI %s.%s \n' % (PROTOCOL_MAJOR, PROTOCOL_MINOR)) clear_all_messages() self.u._handleVersion() self.assertEquals(get_messages(), ["protocol established"]) assert self.u._protocol_ok == True, "_protocol_ok value unexpected"
def testHandleVersion(self): """Testing handleVersion""" assert self.u._protocol_ok == False , "_protocol_ok : False expected" # Messages should be empty, protocol is not established clear_all_messages() self.u._handleVersion() self.assertEquals(get_messages(), []) assert self.u._protocol_ok == False ,"_protocol_ok change unexpected" self.u._packet = list('\n') # Messages should be empty, protocol is not established clear_all_messages() self.u._handleVersion() self.assertEquals(get_messages(), []) assert self.u.transport._loseConnection == True , "loseConnection not called" self.u.transport = FakeTransport() # transport re-init self.u._packet = list('CGI a.b\n') # Messages should be empty, protocol is not established clear_all_messages() self.u._handleVersion() self.assertEquals(get_messages(), []) assert self.u.transport._loseConnection == True , "loseConnection not called" self.u.transport = FakeTransport() # transport re-init vers = Version(protocol_number) PROTOCOL_MAJOR = "%03d" % vers.major() PROTOCOL_MINOR = "%d%02d" % ( vers.medium(), vers.minor() ) self.u._packet = list( 'CGI %s.%s \n' % (PROTOCOL_MAJOR, PROTOCOL_MINOR )) clear_all_messages() self.u._handleVersion() self.assertEquals(get_messages(), ["protocol established"]) assert self.u._protocol_ok == True , "_protocol_ok value unexpected"
# Loic Dachary <*****@*****.**> # # from twisted.internet import reactor, protocol from twisted.python.runtime import seconds from pokerpackets.packets import Packet, PacketFactory, PACKET_PING from pokernetwork import protocol_number from pokernetwork.version import Version from pokernetwork import log as network_log log = network_log.get_child('protocol') protocol_version = Version(protocol_number) PROTOCOL_MAJOR = "%03d" % protocol_version.major() PROTOCOL_MINOR = "%d%02d" % ( protocol_version.medium(), protocol_version.minor() ) class Queue: def __init__(self): self.delay = 0 self.packets = [] class UGAMEProtocol(protocol.Protocol): """UGAMEProtocol""" _stats_read = 0 _stats_write = 0 log = log.get_child('UGAMEProtocol') def __init__(self):
# # from twisted.internet import reactor, protocol from twisted.python.runtime import seconds from pokerpackets.packets import Packet, PacketFactory, PACKET_PING from pokernetwork import protocol_number from pokernetwork.version import Version from pokernetwork import log as network_log log = network_log.get_child('protocol') protocol_version = Version(protocol_number) PROTOCOL_MAJOR = "%03d" % protocol_version.major() PROTOCOL_MINOR = "%d%02d" % (protocol_version.medium(), protocol_version.minor()) class Queue: def __init__(self): self.delay = 0 self.packets = [] class UGAMEProtocol(protocol.Protocol): """UGAMEProtocol""" _stats_read = 0 _stats_write = 0 log = log.get_child('UGAMEProtocol')