class TestHostStompEngine(unittest.TestCase):
engine = inject.attr('stompEngine')
def setUp(self):
self.stompProtocol = StompProtocolFactory().buildProtocol(
('127.0.0.1', 0))
self.fakeTransport = proto_helpers.StringTransport()
def tearDown(self):
self.fakeTransport.clear()
def test_connection(self):
self.stompProtocol.makeConnection(self.fakeTransport)
#ignore the connection request sent. We aren't testing that here
self.fakeTransport.clear()
#pretend we've received successful ack of our connection request.
#check that the host engine subscribes to the topics upon connection
connectedMsg = """CONNECTED
session:ID:snorky.local-49191-1185461799654-3:18"""
self.stompProtocol.dataReceived(connectedMsg)
msg1 = stomper.subscribe(destinations.CONN_DESTINATION)
msg2 = stomper.subscribe(destinations.CMD_RES_DESTINATION)
self.assertTrue(msg1 in self.fakeTransport.value())
self.assertTrue(msg2 in self.fakeTransport.value())
