github/SleekXMPP
1
0
Fork 0
mirror of https://github.com/correl/SleekXMPP.git synced 2024-11-27 19:19:54 +00:00
Code Issues Projects Releases Packages Wiki Activity

fixed pubsub test

Browse source
This commit is contained in:
Nathan Fritz 2009-06-12 19:27:29 +00:00
parent fb3e0d85c4
commit 5fbf0f84d7
1 changed files with 17 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
sleekxmpp/tests/testpubsub.py
View file

@ -17,18 +17,19 @@
""" """
import logging import logging
import sleekxmpp.clientxmpp import sleekxmpp
from optparse import OptionParser from optparse import OptionParser
from xml.etree import cElementTree as ET from xml.etree import cElementTree as ET
import os import os
import time import time
import sys import sys
import queue import Queue
import thread
class testps(sleekxmpp.clientxmpp.ClientXMPP): class testps(sleekxmpp.ClientXMPP):
def __init__(self, jid, password, ssl=False, plugin_config = {}, plugin_whitelist=[], nodenum=0, pshost=None): def __init__(self, jid, password, ssl=False, plugin_config = {}, plugin_whitelist=[], nodenum=0, pshost=None):
sleekxmpp.clientxmpp.ClientXMPP.__init__(self, jid, password, ssl, plugin_config, plugin_whitelist) sleekxmpp.ClientXMPP.__init__(self, jid, password, ssl, plugin_config, plugin_whitelist)
self.registerPlugin('xep_0004') self.registerPlugin('xep_0004')
self.registerPlugin('xep_0030') self.registerPlugin('xep_0030')
self.registerPlugin('xep_0060') self.registerPlugin('xep_0060')
@ -36,7 +37,7 @@ class testps(sleekxmpp.clientxmpp.ClientXMPP):
self.add_handler("<message xmlns='jabber:client'><event xmlns='http://jabber.org/protocol/pubsub#event' /></message>", self.pubsubEventHandler, threaded=True) self.add_handler("<message xmlns='jabber:client'><event xmlns='http://jabber.org/protocol/pubsub#event' /></message>", self.pubsubEventHandler, threaded=True)
self.add_event_handler("session_start", self.start, threaded=True) self.add_event_handler("session_start", self.start, threaded=True)
self.add_handler("<iq type='error' />", self.handleError) self.add_handler("<iq type='error' />", self.handleError)
self.events = queue.Queue() self.events = Queue.Queue()
self.default_config = None self.default_config = None
self.ps = self.plugin['xep_0060'] self.ps = self.plugin['xep_0060']
self.node = "pstestnode_%s" self.node = "pstestnode_%s"
@ -57,7 +58,7 @@ class testps(sleekxmpp.clientxmpp.ClientXMPP):
#TODO: make this configurable #TODO: make this configurable
self.getRoster() self.getRoster()
self.sendPresence(ppriority=20) self.sendPresence(ppriority=20)
self.test_all() thread.start_new(self.test_all, tuple())
def sprint(self, msg, end=False, color=False): def sprint(self, msg, end=False, color=False):
length = len(msg) length = len(msg)
@ -176,12 +177,12 @@ class testps(sleekxmpp.clientxmpp.ClientXMPP):
def test_addItem(self): def test_addItem(self):
"Adding item, waiting for notification" "Adding item, waiting for notification"
item = ET.Element('test') item = ET.Element('test')
result = self.ps.setItem(self.pshost, self.node % self.leafnode, {'test_node1': item}) result = self.ps.setItem(self.pshost, self.node % self.leafnode, (('test_node1', item),))
if result == False: if result == False:
return False return False
try: try:
event = self.events.get(True, 10) event = self.events.get(True, 10)
except queue.Empty: except Queue.Empty:
return False return False
if event == 'test_node1': if event == 'test_node1':
return True return True
@ -191,12 +192,12 @@ class testps(sleekxmpp.clientxmpp.ClientXMPP):
"Updating item, waiting for notification" "Updating item, waiting for notification"
item = ET.Element('test') item = ET.Element('test')
item.attrib['crap'] = 'yup, right here' item.attrib['crap'] = 'yup, right here'
result = self.ps.setItem(self.pshost, self.node % self.leafnode, {'test_node1': item}) result = self.ps.setItem(self.pshost, self.node % self.leafnode, (('test_node1', item),))
if result == False: if result == False:
return False return False
try: try:
event = self.events.get(True, 10) event = self.events.get(True, 10)
except queue.Empty: except Queue.Empty:
return False return False
if event == 'test_node1': if event == 'test_node1':
return True return True
@ -209,7 +210,7 @@ class testps(sleekxmpp.clientxmpp.ClientXMPP):
return False return False
try: try:
event = self.events.get(True, 10) event = self.events.get(True, 10)
except queue.Empty: except Queue.Empty:
self.lasterror = "No Notification" self.lasterror = "No Notification"
return False return False
if event == 'test_node1': if event == 'test_node1':
@ -243,7 +244,7 @@ class testps(sleekxmpp.clientxmpp.ClientXMPP):
return False return False
try: try:
event = self.events.get(True, 10) event = self.events.get(True, 10)
except queue.Empty: except Queue.Empty:
self.lasterror = "No Notification" self.lasterror = "No Notification"
return False return False
if event == self.node % self.leafnode: if event == self.node % self.leafnode:
@ -265,7 +266,7 @@ class testps(sleekxmpp.clientxmpp.ClientXMPP):
return False return False
try: try:
event = self.events.get(True, 10) event = self.events.get(True, 10)
except queue.Empty: except Queue.Empty:
self.lasterror = "No Notification" self.lasterror = "No Notification"
return False return False
if event == self.node % self.leafnode: if event == self.node % self.leafnode:
@ -287,7 +288,7 @@ class testps(sleekxmpp.clientxmpp.ClientXMPP):
return False return False
try: try:
event = self.events.get(True, 10) event = self.events.get(True, 10)
except queue.Empty: except Queue.Empty:
self.lasterror = "No Notification" self.lasterror = "No Notification"
return False return False
if event == self.node % self.leafnode: if event == self.node % self.leafnode:
@ -309,7 +310,7 @@ class testps(sleekxmpp.clientxmpp.ClientXMPP):
return False return False
try: try:
event = self.events.get(True, 10) event = self.events.get(True, 10)
except queue.Empty: except Queue.Empty:
self.lasterror = "No Notification" self.lasterror = "No Notification"
return False return False
if event == self.node % self.leafnode: if event == self.node % self.leafnode:
@ -355,5 +356,5 @@ if __name__ == '__main__':
con.connect() con.connect()
else: else:
con.connect((config.attrib['server'], 5222)) con.connect((config.attrib['server'], 5222))
con.process() con.process(threaded=False)
print("") print("")