def test_presence_message(self):
# since we mock out the xmpp client in previous tests, we can't rely on it
# to call the xmpp_presence method. therefore, let's test it separately.
fake_subscribed_presence = flexmock(name='subscribed')
fake_subscribe_presence = flexmock(name='subscribe')
fake_from = flexmock(name='fake_from')
fake_from.should_receive('getStripped').and_return('me@public1')
flexmock(xmpp)
xmpp.should_receive('Presence').with_args(
to=fake_from,
typ='subscribed').and_return(fake_subscribed_presence)
xmpp.should_receive('Presence').with_args(
to=fake_from, typ='subscribe').and_return(fake_subscribe_presence)
fake_conn = flexmock(name='fake_conn')
fake_conn.should_receive('send').with_args(fake_subscribed_presence)
fake_conn.should_receive('send').with_args(fake_subscribe_presence)
fake_event = flexmock(name='fake_event')
fake_event.should_receive('getFrom').and_return(fake_from)
fake_event.should_receive('getPayload').and_return('doesnt matter')
fake_event.should_receive('getType').and_return('subscribe')
receiver = XMPPReceiver(self.appid, self.login_ip,
self.load_balancer_ip, self.password)
receiver.xmpp_presence(fake_conn, fake_event)
def test_receive_one_message(self):
# mock out the xmpp connection and have it connect and authenticate
fake_connection = flexmock(name='fake_connection', _sock="the socket")
fake_client = flexmock(name='fake_client', Connection=fake_connection)
fake_client.should_receive('connect').and_return(True)
fake_client.should_receive('auth').with_args(
self.appid, self.password, resource='').and_return(True)
# also add in mocks for when messages are received or when we see
# presence notifications
fake_client.should_receive('RegisterHandler').and_return()
# add in a mock for when we send our presence message to the XMPP server
fake_client.should_receive('sendInitPresence').and_return()
# and make sure that we only process one message
fake_client.should_receive('Process').with_args(1).once()
flexmock(xmpp)
xmpp.should_receive('Client').with_args(self.login_ip, debug=[]) \
.and_return(fake_client)
# finally, mock out 'select', and have it put in a message
flexmock(select)
message = {"the socket": "xmpp"}
select.should_receive('select').with_args(['the socket'], [], [], 1) \
.and_return(message, None, None)
receiver = XMPPReceiver(self.appid, self.login_ip,
self.load_balancer_ip, self.password)
actual_messages_sent = receiver.listen_for_messages(
messages_to_listen_for=1)
self.assertEquals(1, actual_messages_sent)
def test_message_results_in_post(self):
# since we mock out the xmpp client in previous tests, we can't rely on it
# to call the xmpp_message method. therefore, let's test it separately.
fake_conn = flexmock(name='fake_conn')
fake_from = flexmock(name='fake_from')
fake_from.should_receive('getStripped').and_return('me@public1')
fake_event = flexmock(name='fake_event')
fake_event.should_receive('getFrom').and_return(fake_from)
fake_event.should_receive('getBody').and_return('doesnt matter')
fake_event.should_receive('getType').and_return('chat')
# and mock out the httplib call
fake_response = flexmock(name='fake_response', status=200)
fake_http_connection = flexmock(name='fake_http_connection')
fake_http_connection.should_receive('request').with_args(
'POST', '/_ah/xmpp/message/chat/', str, XMPPReceiver.HEADERS)
fake_http_connection.should_receive('getresponse').and_return(
fake_response)
fake_http_connection.should_receive('close').and_return()
flexmock(httplib)
httplib.should_receive('HTTPConnection').with_args('publicip1', 1234) \
.and_return(fake_http_connection)
receiver = XMPPReceiver(self.appid, self.login_ip,
self.load_balancer_ip, self.password)
receiver.xmpp_message(fake_conn, fake_event)
def test_receive_one_message(self):
# mock out the xmpp connection and have it connect and authenticate
fake_connection = flexmock(name='fake_connection', _sock="the socket")
fake_client = flexmock(name='fake_client', Connection=fake_connection)
fake_client.should_receive('connect').and_return(True)
fake_client.should_receive('auth').with_args(
self.appid, self.password, resource='').and_return(True)
# also add in mocks for when messages are received or when we see
# presence notifications
fake_client.should_receive('RegisterHandler').and_return()
# add in a mock for when we send our presence message to the XMPP server
fake_client.should_receive('sendInitPresence').and_return()
# and make sure that we only process one message
fake_client.should_receive('Process').and_return(len('the message')).\
and_return(0)
flexmock(xmpp)
xmpp.should_receive('Client').with_args(self.login_ip, debug=[]) \
.and_return(fake_client)
receiver = XMPPReceiver(self.appid, self.login_ip,
self.load_balancer_ip, self.password)
receiver.listen_for_messages()
def test_connect_to_xmpp_but_it_is_down(self):
# mock out the xmpp connection and have it not connect
fake_client = flexmock(name='fake_client')
fake_client.should_receive('connect').and_return(None)
flexmock(xmpp)
xmpp.should_receive('Client').with_args(self.login_ip, debug=[]) \
.and_return(fake_client)
receiver = XMPPReceiver(self.appid, self.login_ip,
self.load_balancer_ip, self.password)
self.assertRaises(SystemExit, receiver.listen_for_messages)
def test_connect_to_xmpp_but_cannot_auth(self):
# mock out the xmpp connection and have it connect, but not authenticate
fake_client = flexmock(name='fake_client')
fake_client.should_receive('connect').and_return(True)
fake_client.should_receive('auth').with_args(
self.appid, self.password, resource='').and_return(None)
flexmock(xmpp)
xmpp.should_receive('Client').with_args(self.login_ip, debug=[]) \
.and_return(fake_client)
receiver = XMPPReceiver(self.appid, self.login_ip, self.password)
self.assertRaises(SystemExit,
receiver.listen_for_messages,
messages_to_listen_for=1)
def test_message_results_in_post(self):
# since we mock out the xmpp client in previous tests, we can't rely on it
# to call the xmpp_message method. therefore, let's test it separately.
fake_conn = flexmock(name='fake_conn')
fake_from = flexmock(name='fake_from')
fake_from.should_receive('getStripped').and_return('me@public1')
fake_event = flexmock(name='fake_event')
fake_event.should_receive('getFrom').and_return(fake_from)
fake_event.should_receive('getBody').and_return('doesnt matter')
fake_event.should_receive('getType').and_return('chat')
# and mock out the urllib call
flexmock(urllib)
urllib.should_receive('urlopen').with_args(
"http://publicip1:1234/_ah/xmpp/message/chat/", str).and_return()
receiver = XMPPReceiver(self.appid, self.login_ip, self.app_port,
self.password)
receiver.xmpp_message(fake_conn, fake_event)
