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
|
#!/usr/bin/env python
'''Simple test with a blocking C++ method that should allow python
threads to run.'''
import unittest
import threading
from sample import Bucket
class Unlocker(threading.Thread):
def __init__(self, bucket):
threading.Thread.__init__(self)
self.bucket = bucket
def run(self):
while not self.bucket.locked():
pass
self.bucket.unlock()
class MyBucket(Bucket):
def __init__(self):
Bucket.__init__(self)
def virtualBlockerMethod(self):
self.lock()
return True
class TestLockUnlock(unittest.TestCase):
def testBasic(self):
'''Locking in C++ and releasing in a python thread'''
bucket = Bucket()
unlocker = Unlocker(bucket)
unlocker.start()
bucket.lock()
unlocker.join()
def testVirtualBlocker(self):
'''Same as the basic case but blocker method is a C++ virtual called from C++.'''
bucket = Bucket()
unlocker = Unlocker(bucket)
unlocker.start()
result = bucket.callVirtualBlockerMethodButYouDontKnowThis()
unlocker.join()
self.assert_(result)
def testReimplementedVirtualBlocker(self):
'''Same as the basic case but blocker method is a C++ virtual reimplemented in Python and called from C++.'''
mybucket = MyBucket()
unlocker = Unlocker(mybucket)
unlocker.start()
result = mybucket.callVirtualBlockerMethodButYouDontKnowThis()
unlocker.join()
self.assert_(result)
if __name__ == '__main__':
unittest.main()
|