def test_error_frame_after_connect_raises_StompProtocolError(self): stomp = Stomp('localhost', 61613) stomp._socketConnect = Mock() stomp.receiveFrame = Mock() stomp.receiveFrame.return_value = {'cmd': 'ERROR', 'headers': {}, 'body': 'fake error'} stomp.socket = Mock() self.assertRaises(StompProtocolError, lambda: stomp.connect()) self.assertEquals(stomp.receiveFrame.call_count, 1, "receiveFrame not called")
def test_connect_writes_correct_frame(self): login = '******' passcode = 'george' stomp = Stomp('localhost', 61613) stomp._socketConnect = Mock() stomp.receiveFrame = Mock() stomp.receiveFrame.return_value = {'cmd': 'CONNECTED', 'headers': {}, 'body': ''} stomp.socket = Mock() stomp.connect(login=login,passcode=passcode) args,kargs = stomp.socket.sendall.call_args sentFrame = self.parseFrame(args[0]) self.assertEquals({'cmd': 'CONNECT', 'headers': {'login': login, 'passcode': passcode, }, 'body': ''}, sentFrame)