mirror of
https://github.com/wxWidgets/Phoenix.git
synced 2026-01-07 12:30:07 +01:00
d0e440a304
git-svn-id: https://svn.wxwidgets.org/svn/wx/wxPython/Phoenix/trunk@73830 c3d73ce0-8a6f-49c7-b76d-6d57e0e08775
117 lines
4.2 KiB
Python
117 lines
4.2 KiB
Python
"""
|
|
|
|
:copyright: Copyright 2006-2009 by Oliver Schoenborn, all rights reserved.
|
|
:license: BSD, see LICENSE.txt for details.
|
|
|
|
|
|
"""
|
|
|
|
import imp_unittest, unittest
|
|
|
|
import wtc
|
|
|
|
from wx.lib.pubsub import pub
|
|
from wx.lib.pubsub.utils.notification import useNotifyByPubsubMessage
|
|
topicMgr = pub.getDefaultTopicMgr()
|
|
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
|
|
class lib_pubsub_Notify2(wtc.WidgetTestCase):
|
|
|
|
def test0_NotificationTopics(self):
|
|
assert not topicMgr.getTopic('pubsub', okIfNone=True)
|
|
useNotifyByPubsubMessage()
|
|
assert topicMgr.getTopic('pubsub')
|
|
|
|
assert topicMgr.getTopic('pubsub').hasSubtopic()
|
|
|
|
pubsubTopicNames = [obj.getName() for obj in topicMgr.getTopic('pubsub').getSubtopics()]
|
|
self.assertEqual(
|
|
set( pubsubTopicNames ),
|
|
set(['pubsub.sendMessage', 'pubsub.deadListener',
|
|
'pubsub.subscribe', 'pubsub.unsubscribe',
|
|
'pubsub.newTopic', 'pubsub.delTopic'])
|
|
)
|
|
|
|
|
|
def test1_SubscribeNotify(self):
|
|
class MyListener:
|
|
countSub = 0
|
|
countUnsub = 0
|
|
def listenerSub(self, msgTopic=pub.AUTO_TOPIC, listener=None,
|
|
topic=None, newSub=None):
|
|
self.assertEqual(msgTopic.getName(), 'pubsub.subscribe' )
|
|
assert topic.getName() in ('pubsub.unsubscribe', 'testSubscribeNotify')
|
|
if newSub:
|
|
self.countSub += 1
|
|
def listenerUnsub(self, msgTopic=pub.AUTO_TOPIC, topic=None,
|
|
listener=None, listenerRaw=None):
|
|
assert topic.getName() in ('testSubscribeNotify', 'pubsub.subscribe' )
|
|
self.assertEqual(msgTopic.getName(), 'pubsub.unsubscribe' )
|
|
if listener is not None:
|
|
self.countUnsub += 1
|
|
def listenerTest(self):
|
|
raise NotImplementedError # should never get here
|
|
|
|
pub.setNotificationFlags(subscribe=True, unsubscribe=True)
|
|
pub.getOrCreateTopic('testSubscribeNotify')
|
|
tmp = MyListener()
|
|
tmp.assertEqual = self.assertEqual
|
|
|
|
pub.subscribe(tmp.listenerSub, 'pubsub.subscribe')
|
|
self.assertEqual(tmp.countSub, 0) # don't notify of self subscription
|
|
self.assertEqual(tmp.countUnsub, 0)
|
|
sl, ok = pub.subscribe(tmp.listenerUnsub, 'pubsub.unsubscribe')
|
|
assert ok
|
|
self.assertEqual(tmp.countSub, 1)
|
|
self.assertEqual(tmp.countUnsub, 0)
|
|
|
|
pub.subscribe(tmp.listenerTest, 'testSubscribeNotify')
|
|
self.assertEqual(tmp.countUnsub, 0)
|
|
pub.unsubscribe(tmp.listenerTest, 'testSubscribeNotify')
|
|
self.assertEqual(tmp.countUnsub, 1)
|
|
|
|
pub.unsubscribe(tmp.listenerSub, 'pubsub.subscribe')
|
|
self.assertEqual(tmp.countSub, 2)
|
|
self.assertEqual(tmp.countUnsub, 2)
|
|
pub.unsubscribe(tmp.listenerUnsub, 'pubsub.unsubscribe')
|
|
self.assertEqual(tmp.countSub, 2)
|
|
self.assertEqual(tmp.countUnsub, 2) # don't notify of self unsubscription
|
|
|
|
|
|
def test2_SendNotify(self):
|
|
# trap the pubsub.sendMessage topic:
|
|
class SendHandler:
|
|
def __init__(self):
|
|
self.pre = self.post = 0
|
|
def __call__(self, topic=None, stage=None, listener=None, msgTopic=pub.AUTO_TOPIC):
|
|
if stage == 'pre':
|
|
self.pre += 1
|
|
else:
|
|
self.post += 1
|
|
self.assertEqual(msgTopic.getName(), 'pubsub.sendMessage')
|
|
self.assertEqual(topic.getName(), 'testSendNotify')
|
|
sh = SendHandler()
|
|
sh.assertEqual = self.assertEqual
|
|
|
|
pub.subscribe(sh, 'pubsub.sendMessage')
|
|
pub.setNotificationFlags(sendMessage=True)
|
|
|
|
# generate a message that will cause pubsub.sendMessage to be generated too
|
|
assert sh.pre == 0
|
|
assert sh.post == 0
|
|
pub.getOrCreateTopic('testSendNotify')
|
|
pub.sendMessage('testSendNotify')
|
|
assert sh.pre == 1
|
|
assert sh.post == 1
|
|
|
|
|
|
|
|
#---------------------------------------------------------------------------
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|