Skip to content
Snippets Groups Projects
Commit e419132c authored by Herwin van Welbergen's avatar Herwin van Welbergen
Browse files

made more stuff python 2.6 compliant

parent 51a7e0c3
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
ipaacalib/python/test/src/testnotifier.py
+ 122
− 122
View file @ e419132c
''' '''
Created on Dec 20, 2012 Created on Dec 20, 2012
@author: hvanwelbergen @author: hvanwelbergen
''' '''
import unittest import unittest
from mockito import mock from mockito import mock
from mockito import verify from mockito import verify
from mockito import any from mockito import any
from mockito import when from mockito import when
from mockito import times from mockito import times
from notifier import ComponentNotifier from notifier import ComponentNotifier
from hamcrest.core.base_matcher import BaseMatcher from hamcrest.core.base_matcher import BaseMatcher
from ipaaca import IUEventType from ipaaca import IUEventType
from ipaaca import Message from ipaaca import Message
from ipaaca import InputBuffer from ipaaca import InputBuffer
from ipaaca import OutputBuffer from ipaaca import OutputBuffer
import time import time
import os import os
class IUCategoryMatcher(BaseMatcher): class IUCategoryMatcher(BaseMatcher):
def __init__(self, expected_cat): def __init__(self, expected_cat):
self.expected_cat = expected_cat self.expected_cat = expected_cat
def _matches_(self, iu): def _matches_(self, iu):
return iu.category==self.expected_cat return iu.category==self.expected_cat
def describe_to(self, description): def describe_to(self, description):
description.append_text("IU with category :"+self.expected_cat) description.append_text("IU with category :"+self.expected_cat)
class ComponentNotifierTest(unittest.TestCase): class ComponentNotifierTest(unittest.TestCase):
RECV_CAT = set(["testrec1", "testrc2"]) RECV_CAT = set(["testrec1", "testrc2"])
SEND_CAT = set(["testsnd1", "testsnd2", "testsnd3"]) SEND_CAT = set(["testsnd1", "testsnd2", "testsnd3"])
def setUp(self): def setUp(self):
self.mockOutBuffer = mock() self.mockOutBuffer = mock()
self.mockInBuffer = mock() self.mockInBuffer = mock()
self.notifier = ComponentNotifier("testcomp","testfunc", ComponentNotifierTest.SEND_CAT, ComponentNotifierTest.RECV_CAT, self.mockOutBuffer, self.mockInBuffer) self.notifier = ComponentNotifier("testcomp","testfunc", ComponentNotifierTest.SEND_CAT, ComponentNotifierTest.RECV_CAT, self.mockOutBuffer, self.mockInBuffer)
self.notifier.initialize() self.notifier.initialize()
def tearDown(self): def tearDown(self):
pass pass
def _sendNotify(self, state, receiveCats): def _sendNotify(self, state, receiveCats):
mockIUNotify = Message(ComponentNotifier.NOTIFY_CATEGORY) mockIUNotify = Message(ComponentNotifier.NOTIFY_CATEGORY)
mockIUNotify.payload[ComponentNotifier.STATE] = state; mockIUNotify.payload[ComponentNotifier.STATE] = state;
mockIUNotify.payload[ComponentNotifier.NAME] = "namex"; mockIUNotify.payload[ComponentNotifier.NAME] = "namex";
mockIUNotify.payload[ComponentNotifier.SEND_CATEGORIES] = ""; mockIUNotify.payload[ComponentNotifier.SEND_CATEGORIES] = "";
mockIUNotify.payload[ComponentNotifier.RECEIVE_CATEGORIES] = ",".join(receiveCats); mockIUNotify.payload[ComponentNotifier.RECEIVE_CATEGORIES] = ",".join(receiveCats);
self.notifier._handle_iu_event(mockIUNotify, IUEventType.ADDED, False) self.notifier._handle_iu_event(mockIUNotify, IUEventType.ADDED, False)
def testNotifyAtInit(self): def testNotifyAtInit(self):
verify(self.mockOutBuffer).add(any()) verify(self.mockOutBuffer).add(any())
#TODO: python mockito cannot yet use hamcrest matchers, so cannot easily test if the message is correct :( #TODO: python mockito cannot yet use hamcrest matchers, so cannot easily test if the message is correct :(
#assertEquals(ComponentNotifier.NOTIFY_CATEGORY, iu.getCategory()); #assertEquals(ComponentNotifier.NOTIFY_CATEGORY, iu.getCategory());
#assertEquals("new", iu.getPayload().get(ComponentNotifier.STATE)); #assertEquals("new", iu.getPayload().get(ComponentNotifier.STATE));
#assertThat(ImmutableSet.copyOf(iu.getPayload().get(ComponentNotifier.RECEIVE_CATEGORIES).split(",")), #assertThat(ImmutableSet.copyOf(iu.getPayload().get(ComponentNotifier.RECEIVE_CATEGORIES).split(",")),
# IsIterableContainingInAnyOrder.containsInAnyOrder(RECV_CAT.toArray(new String[0]))); # IsIterableContainingInAnyOrder.containsInAnyOrder(RECV_CAT.toArray(new String[0])));
#assertThat(ImmutableSet.copyOf(iu.getPayload().get(ComponentNotifier.SEND_CATEGORIES).split(",")), #assertThat(ImmutableSet.copyOf(iu.getPayload().get(ComponentNotifier.SEND_CATEGORIES).split(",")),
# IsIterableContainingInAnyOrder.containsInAnyOrder(SEND_CAT.toArray(new String[0]))); # IsIterableContainingInAnyOrder.containsInAnyOrder(SEND_CAT.toArray(new String[0])));
def testNotifyAtNotifyNew(self): def testNotifyAtNotifyNew(self):
self._sendNotify("new", {"testsnd1"}); self._sendNotify("new", set(["testsnd1"]));
verify(self.mockOutBuffer, times(2)).add(any()) verify(self.mockOutBuffer, times(2)).add(any())
#TODO: python mockito cannot yet use hamcrest matchers, so cannot easily test if the message is correct :( #TODO: python mockito cannot yet use hamcrest matchers, so cannot easily test if the message is correct :(
#ArgumentCaptor<LocalIU> argument = ArgumentCaptor.forClass(LocalIU.class); #ArgumentCaptor<LocalIU> argument = ArgumentCaptor.forClass(LocalIU.class);
#verify(mockOutBuffer, times(2)).add(argument.capture()); #verify(mockOutBuffer, times(2)).add(argument.capture());
#LocalIU iu = argument.getAllValues().get(1); #LocalIU iu = argument.getAllValues().get(1);
#assertEquals("componentNotify", iu.getCategory()); #assertEquals("componentNotify", iu.getCategory());
#assertEquals("old", iu.getPayload().get("state")); #assertEquals("old", iu.getPayload().get("state"));
def testNoNotifyAtNotifyOld(self): def testNoNotifyAtNotifyOld(self):
self._sendNotify("old", {"testsnd1"}); self._sendNotify("old", set(["testsnd1"]));
verify(self.mockOutBuffer, times(1)).add(any()) verify(self.mockOutBuffer, times(1)).add(any())
class MyListener(object): class MyListener(object):
def __init__(self): def __init__(self):
self.numCalled = 0 self.numCalled = 0
def handle(self, iu, mytype, local): def handle(self, iu, mytype, local):
self.numCalled += 1 self.numCalled += 1
class ComponentNotifierIntegrationTest(unittest.TestCase): class ComponentNotifierIntegrationTest(unittest.TestCase):
def setUp(self): def setUp(self):
pass pass
def tearDown(self): def tearDown(self):
pass pass
def _setupCompNotifier(self, idx, sendList, recvList): def _setupCompNotifier(self, idx, sendList, recvList):
inBuffer = InputBuffer(idx+"in", {ComponentNotifier.NOTIFY_CATEGORY}) inBuffer = InputBuffer(idx+"in", {ComponentNotifier.NOTIFY_CATEGORY})
outBuffer = OutputBuffer(idx+"out") outBuffer = OutputBuffer(idx+"out")
return ComponentNotifier(idx, "testfunction", sendList, recvList, outBuffer, inBuffer) return ComponentNotifier(idx, "testfunction", sendList, recvList, outBuffer, inBuffer)
# bug: this somehow remains active after running # bug: this somehow remains active after running
# def testSelf(self): # def testSelf(self):
# notifier = self._setupCompNotifier("not", {"a1","b1"}, {"a3","b1"}); # notifier = self._setupCompNotifier("not", {"a1","b1"}, {"a3","b1"});
# listener = MyListener() # listener = MyListener()
# notifier.add_notification_handler(listener.handle); # notifier.add_notification_handler(listener.handle);
# #
# notifier.initialize(); # notifier.initialize();
# time.sleep(0.5); # time.sleep(0.5);
# #
# self.assertEquals(0, listener.numCalled); # self.assertEquals(0, listener.numCalled);
def testTwo(self): def testTwo(self):
notifier1 = self._setupCompNotifier("not1", {"a1", "b1"}, {"a3", "b2"}); notifier1 = self._setupCompNotifier("not1", {"a1", "b1"}, {"a3", "b2"});
notifier2 = self._setupCompNotifier("not2", {"a2", "b2"}, {"a3", "b1"}); notifier2 = self._setupCompNotifier("not2", {"a2", "b2"}, {"a3", "b1"});
listener1 = MyListener() listener1 = MyListener()
listener2 = MyListener() listener2 = MyListener()
notifier1.add_notification_handler(listener1.handle) notifier1.add_notification_handler(listener1.handle)
notifier2.add_notification_handler(listener2.handle) notifier2.add_notification_handler(listener2.handle)
notifier1.initialize() notifier1.initialize()
time.sleep(0.5) time.sleep(0.5)
notifier2.initialize() notifier2.initialize()
time.sleep(0.5) time.sleep(0.5)
self.assertEqual(1, listener1.numCalled) self.assertEqual(1, listener1.numCalled)
self.assertEqual(1, listener2.numCalled) self.assertEqual(1, listener2.numCalled)
if __name__ == "__main__": if __name__ == "__main__":
#import sys;sys.argv = ['', 'Test.testName'] #import sys;sys.argv = ['', 'Test.testName']
unittest.main() unittest.main()
os._exit(0) os._exit(0)
\ No newline at end of file
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment