def test_session_expires_on_given_condition(self): control = '/tmp/stream_test' open(control, 'w') session = StreamSessionManager() session.add('*****@*****.**', condition=lambda: os.path.exists(control)) session.expire() self.assertEquals(session.receivers, ['*****@*****.**']) session.expire() self.assertEquals(session.receivers, ['*****@*****.**']) os.remove(control) session.expire() self.assertEquals(session.receivers, [])
def test_session_expires_after_timeout(self): self.set_date('2011-09-21 01:05:10') session = StreamSessionManager() session.add('*****@*****.**', timeout = 10) self.set_date('2011-09-21 01:05:11') session.expire() self.assertEquals(session.receivers, ['*****@*****.**']) self.set_date('2011-09-21 01:05:21') session.expire() self.assertEquals(session.receivers, [])
def test_session_support_multiple_jids(self): self.set_date('2011-09-21 01:05:10') control = '/tmp/stream_test' open(control, 'w') session = StreamSessionManager() session.add('*****@*****.**', timeout = 10) session.add('*****@*****.**', timeout = 20, condition = lambda: os.path.exists(control)) session.add('*****@*****.**', timeout = 30) session.add('*****@*****.**', condition = lambda: os.path.exists(control)) self.set_date('2011-09-21 01:05:11') session.expire() self.assertEquals(len(session.receivers), 3) self.assertTrue('*****@*****.**' in session.receivers) self.assertTrue('*****@*****.**' in session.receivers) self.assertTrue('*****@*****.**' in session.receivers) # test1 expires self.set_date('2011-09-21 01:05:21') session.expire() self.assertEquals(len(session.receivers), 2) self.assertTrue('*****@*****.**' not in session.receivers) self.assertTrue('*****@*****.**' in session.receivers) self.assertTrue('*****@*****.**' in session.receivers) # test2 expires, even if condition is still true self.set_date('2011-09-21 01:05:31') session.expire() self.assertEquals(len(session.receivers), 1) self.assertTrue('*****@*****.**' not in session.receivers) self.assertTrue('*****@*****.**' not in session.receivers) self.assertTrue('*****@*****.**' in session.receivers) # test3 expires, but it remains there because there are two sessions self.set_date('2011-09-21 01:05:41') session.expire() self.assertEquals(len(session.receivers), 1) self.assertTrue('*****@*****.**' not in session.receivers) self.assertTrue('*****@*****.**' not in session.receivers) self.assertTrue('*****@*****.**' in session.receivers) # condition is false, now all receivers are gone os.remove(control) session.expire() self.assertEquals(len(session.receivers), 0) self.assertTrue('*****@*****.**' not in session.receivers) self.assertTrue('*****@*****.**' not in session.receivers) self.assertTrue('*****@*****.**' not in session.receivers)
def test_one_jid_can_be_removed_from_sessions(self): self.set_date('2011-09-21 01:05:10') control = '/tmp/stream_test' open(control, 'w') session = StreamSessionManager() session.add('*****@*****.**', timeout = 10) session.add('*****@*****.**', timeout = 20, condition = lambda: os.path.exists(control)) session.add('*****@*****.**', timeout = 30) session.add('*****@*****.**', condition = lambda: os.path.exists(control)) self.set_date('2011-09-21 01:05:11') self.assertEquals(len(session.receivers), 3) self.assertTrue('*****@*****.**' in session.receivers) self.assertTrue('*****@*****.**' in session.receivers) self.assertTrue('*****@*****.**' in session.receivers) session.remove('*****@*****.**') self.assertEquals(len(session.receivers), 2) self.assertTrue('*****@*****.**' in session.receivers) self.assertTrue('*****@*****.**' in session.receivers) self.assertTrue('*****@*****.**' not in session.receivers)
def test_expired_sessions_can_be_obtained(self): self.set_date('2011-09-21 01:05:10') control = '/tmp/stream_test' open(control, 'w') session = StreamSessionManager() session.add('*****@*****.**', timeout = 10) session.add('*****@*****.**', timeout = 20, condition = lambda: os.path.exists(control)) session.add('*****@*****.**', timeout = 30) session.add('*****@*****.**', condition = lambda: os.path.exists(control)) self.set_date('2011-09-21 01:05:11') expired = session.expire() self.assertEquals(expired, []) # test1 expires self.set_date('2011-09-21 01:05:21') expired = session.expire() self.assertEquals(expired, ['*****@*****.**']) # test2 expires, even if condition is still true self.set_date('2011-09-21 01:05:31') expired = session.expire() self.assertEquals(expired, ['*****@*****.**']) # test3 expires, but it remains there because there are two sessions self.set_date('2011-09-21 01:05:41') expired = session.expire() self.assertEquals(expired, []) # condition is false, now all receivers are gone os.remove(control) expired = session.expire() self.assertEquals(expired, ['*****@*****.**'])