1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68
|
#!/usr/bin/env python
import time
import threading
import unittest
from random import random
from sample import ObjectType, Event
class Producer(ObjectType):
def __init__(self):
ObjectType.__init__(self)
self.data = None
self.read = False
def event(self, event):
self.data = random()
while not self.read:
time.sleep(0.01)
return True
class Collector(threading.Thread):
def __init__(self, objects):
threading.Thread.__init__(self)
self.max_runs = len(objects)
self.objects = objects
self.data = []
def run(self):
i = 0
while i < self.max_runs:
if self.objects[i].data is not None:
self.data.append(self.objects[i].data)
self.objects[i].read = True
i += 1
time.sleep(0.01)
class TestEventLoopWithThread(unittest.TestCase):
'''Communication between a python thread and an simple
event loop in C++'''
def testBasic(self):
'''Allowing threads and calling virtuals from C++'''
number = 10
objs = [Producer() for x in range(number)]
thread = Collector(objs)
thread.start()
evaluated = ObjectType.processEvent(objs,
Event(Event.BASIC_EVENT))
thread.join()
producer_data = [x.data for x in objs]
self.assertEqual(evaluated, number)
self.assertEqual(producer_data, thread.data)
if __name__ == '__main__':
unittest.main()
|