def setUp(self):
self.addr = ("localhost", 59334)
self.s1 = Session(connection_handler=self.dummy_conn_handler)
self.s1.listen(self.addr)
self.s1.set_global_handler(self.global_handler_s1)
self.s2 = Session()
self.s2.set_global_handler(self.global_handler_s2)
class ServerTests(unittest.TestCase):
def setUp(self):
self.addr = ("localhost", 59334)
self.s1 = Session(connection_handler=self.dummy_conn_handler)
self.s1.listen(self.addr)
self.s1.set_global_handler(self.global_handler_s1)
self.s2 = Session()
self.s2.set_global_handler(self.global_handler_s2)
def dummy_conn_handler(self, session):
print "new connexion"
def global_handler_s1(self, action):
if action.action == ACTIONS['start']:
self.s1.accept_start()
else:
print "Got unknown action %s" % str(action)
def global_handler_s2(self, action):
if action.action == ACTIONS['accept_start']:
if self.s2.status == STATUS['connecting']:
self.s2.status = STATUS['connected']
else:
print "Got unknown action %s" % str(action)
def testStartSession(self):
count = 10
self.failUnless(self.s2.start(self.addr))
time.sleep(0.3) # wait a little for the socket to be created
while self.s2.status == STATUS['connecting'] and count > 0:
self.s1.check()
count -= 1
time.sleep(0.01)
self.s2.check()
self.failUnlessEqual(self.s1.status, STATUS['connected'])
self.failUnlessEqual(self.s1.status, self.s2.status)
def tearDown(self):
self.s2.close()
self.s1.close()