def testBasic(self): socket_map = {} xmpp_connection = xmppserver.XmppConnection(self.fake_socket, socket_map, self, ('', 0), True) self.assertEqual(len(socket_map), 1) self.assertEqual(len(self.connections), 0) xmpp_connection.HandshakeDone(xmppserver.Jid('foo', 'bar')) self.assertEqual(len(socket_map), 1) self.assertEqual(len(self.connections), 1) sent_data = self.fake_socket.GetSentData() # Test subscription request. self.assertEqual(len(sent_data), 0) xmpp_connection.collect_incoming_data( '<iq><subscribe xmlns="google:push"></subscribe></iq>') self.assertEqual(len(sent_data), 1) # Test acks. xmpp_connection.collect_incoming_data('<iq type="result"/>') self.assertEqual(len(sent_data), 1) # Test notification. xmpp_connection.collect_incoming_data( '<message><push xmlns="google:push"/></message>') self.assertEqual(len(sent_data), 2) # Test unexpected stanza. def SendUnexpectedStanza(): xmpp_connection.collect_incoming_data('<foo/>') self.assertRaises(xmppserver.UnexpectedXml, SendUnexpectedStanza) # Test unexpected notifier command. def SendUnexpectedNotifierCommand(): xmpp_connection.collect_incoming_data( '<iq><foo xmlns="google:notifier"/></iq>') self.assertRaises(xmppserver.UnexpectedXml, SendUnexpectedNotifierCommand) # Test close. xmpp_connection.close() self.assertEqual(len(socket_map), 0) self.assertEqual(len(self.connections), 0)
def testBasic(self): socket_map = {} xmpp_connection = xmppserver.XmppConnection(self, socket_map, self, ('', 0)) self.assertEqual(len(socket_map), 1) self.assertEqual(len(self.connections), 0) xmpp_connection.HandshakeDone(xmppserver.Jid('foo', 'bar')) self.assertEqual(len(socket_map), 1) self.assertEqual(len(self.connections), 1) # Test subscription request. self.assertEqual(len(self.data), 0) xmpp_connection.collect_incoming_data( '<iq><getAll xmlns="google:notifier">' '<SubscribedServiceUrl/></getAll></iq>') self.assertEqual(len(self.data), 1) # Test acks. xmpp_connection.collect_incoming_data('<iq type="result"/>') self.assertEqual(len(self.data), 1) # Test notification. xmpp_connection.collect_incoming_data( '<iq><set xmlns="google:notifier"/></iq>') self.assertEqual(len(self.data), 2) # Test unexpected stanza. def SendUnexpectedStanza(): xmpp_connection.collect_incoming_data('<foo/>') self.assertRaises(xmppserver.UnexpectedXml, SendUnexpectedStanza) # Test unexpected notifier command. def SendUnexpectedNotifierCommand(): xmpp_connection.collect_incoming_data( '<iq><foo xmlns="google:notifier"/></iq>') self.assertRaises(xmppserver.UnexpectedXml, SendUnexpectedNotifierCommand) # Test close xmpp_connection.close() self.assertEqual(len(socket_map), 0) self.assertEqual(len(self.connections), 0)
def testBasicUnauthenticated(self): socket_map = {} xmpp_connection = xmppserver.XmppConnection(self.fake_socket, socket_map, self, ('', 0), False) self.assertEqual(len(socket_map), 1) self.assertEqual(len(self.connections), 0) xmpp_connection.HandshakeDone(None) self.assertEqual(len(socket_map), 0) self.assertEqual(len(self.connections), 0) # Test unexpected stanza. def SendUnexpectedStanza(): xmpp_connection.collect_incoming_data('<foo/>') self.assertRaises(xmppserver.UnexpectedXml, SendUnexpectedStanza) # Test redundant close. xmpp_connection.close() self.assertEqual(len(socket_map), 0) self.assertEqual(len(self.connections), 0)