def testheartbeat_timeout(self):
server = TestStompServer('127.0.0.1', 60000)
server.start()
try:
server.add_frame('''CONNECTED
version: 1.1
session: 1
server: test
heart-beat: 1000,1000\x00''')
conn = stomp.Connection([('127.0.0.1', 60000)],
version=1.1,
heartbeats=(1000, 1000))
listener = TestListener()
conn.set_listener('', listener)
conn.start()
conn.connect()
time.sleep(5)
server.running = False
self.assert_(listener.heartbeat_timeouts >= 1,
'should have received a heartbeat timeout')
finally:
server.stop()
def testheartbeat_timeout(self):
server = TestStompServer('127.0.0.1', 60000)
server.start()
try:
server.add_frame('''CONNECTED
version: 1.1
session: 1
server: test
heart-beat: 1000,1000\x00''')
conn = stomp.Connection([('127.0.0.1', 60000)], version = 1.1, heartbeats = (1000, 1000))
listener = TestListener()
conn.set_listener('', listener)
conn.start()
conn.connect()
time.sleep(5)
server.running = False
self.assert_(listener.heartbeat_timeouts >= 1, 'should have received a heartbeat timeout')
finally:
server.stop()
def test_disconnect(self):
server = TestStompServer('127.0.0.1', 60000)
try:
server.start()
server.add_frame('''CONNECTED
version: 1.1
session: 1
server: test
heart-beat: 1000,1000\x00''')
conn = stomp.Connection([('127.0.0.1', 60000)], version = 1.0)
listener = TestListener()
conn.set_listener('', listener)
conn.start()
conn.connect()
time.sleep(2)
server.stop()
for count in range(100):
if server.stopped:
break
time.sleep(0.1)
else:
assert False, 'server never disconnected'
try:
conn.send('test disconnect', destination='/test/disconnectqueue')
self.fail('Should not have successfully sent a message at this point')
except Exception:
_, e, _ = sys.exc_info()
log.debug('stopping conn after expected exception %s' % e)
# lost connection, now restart the server
try:
conn.stop()
except:
pass
time.sleep(2)
server.add_frame('''CONNECTED
version: 1.1
session: 1
server: test
heart-beat: 1000,1000\x00''')
server.start()
conn.start()
conn.connect()
time.sleep(5)
self.assert_(listener.connections >= 2, 'should have received 2 connection acknowledgements')
time.sleep(2)
finally:
server.stop()
def test_disconnect(self):
server = TestStompServer('127.0.0.1', 60000)
try:
server.start()
server.add_frame('''CONNECTED
version: 1.1
session: 1
server: test
heart-beat: 1000,1000\x00''')
conn = stomp.Connection([('127.0.0.1', 60000)], version=1.0)
listener = TestListener()
conn.set_listener('', listener)
conn.start()
conn.connect()
time.sleep(2)
server.stop()
for count in range(100):
if server.stopped:
break
time.sleep(0.1)
else:
assert False, 'server never disconnected'
try:
conn.send('test disconnect',
destination='/test/disconnectqueue')
self.fail(
'Should not have successfully sent a message at this point'
)
except Exception:
_, e, _ = sys.exc_info()
log.debug('stopping conn after expected exception %s' % e)
# lost connection, now restart the server
try:
conn.stop()
except:
pass
time.sleep(2)
server.add_frame('''CONNECTED
version: 1.1
session: 1
server: test
heart-beat: 1000,1000\x00''')
server.start()
conn.start()
conn.connect()
time.sleep(5)
self.assert_(listener.connections >= 2,
'should have received 2 connection acknowledgements')
time.sleep(2)
finally:
server.stop()