class PeerTest(unittest.TestCase):
def setUp(self):
self.peer = Peer("bob", lambda x: x)
self.msg = Message.create_message(
"HELLO 127.0.0.1:23501 data youpi")
def test_server(self):
self.assert_(isinstance(self.peer.server.current_state, PeerState))
self.assertRaises(SecurityAlert, self.peer.server.connected, None)
self.assertRaises(SecurityAlert, self.peer.server.disconnected)
self.assertRaises(SecurityAlert, self.peer.server.execute, self.msg)
# registered
self.peer.server.current_state = self.peer.server.registered_state
self.assertRaises(SecurityAlert, self.peer.server.execute, self.msg)
# connected
self.peer.server.connected("transport")
self.assert_(isinstance(self.peer.server.current_state, PeerConnected))
# disconnected
self.peer.server.disconnected()
self.assert_(isinstance(self.peer.server.current_state, PeerDisconnected))
self.assertRaises(SecurityAlert, self.peer.server.execute, self.msg)
# connected
self.peer.server.connected("transport")
self.assert_(isinstance(self.peer.server.current_state, PeerConnected))
def test_peer(self):
msg = self.peer.wrap_message(MESSAGE_HELLO)
self.assertEquals("HELLO ?:-1 ", str(msg))
def setUp(self):
self.peer = Peer("bob", lambda x: x)
self.msg = Message.create_message(
"HELLO 127.0.0.1:23501 data youpi")
