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)
