144 lines
4.3 KiB
Python
144 lines
4.3 KiB
Python
# Run the _testcapi module tests (tests for the Python/C API): by defn,
|
|
# these are all functions _testcapi exports whose name begins with 'test_'.
|
|
|
|
import sys
|
|
import time
|
|
import random
|
|
import unittest
|
|
import threading
|
|
from test import support
|
|
import _testcapi
|
|
|
|
|
|
def testfunction(self):
|
|
"""some doc"""
|
|
return self
|
|
|
|
class InstanceMethod:
|
|
id = _testcapi.instancemethod(id)
|
|
testfunction = _testcapi.instancemethod(testfunction)
|
|
|
|
class CAPITest(unittest.TestCase):
|
|
|
|
def test_instancemethod(self):
|
|
inst = InstanceMethod()
|
|
self.assertEqual(id(inst), inst.id())
|
|
self.assert_(inst.testfunction() is inst)
|
|
self.assertEqual(inst.testfunction.__doc__, testfunction.__doc__)
|
|
self.assertEqual(InstanceMethod.testfunction.__doc__, testfunction.__doc__)
|
|
|
|
InstanceMethod.testfunction.attribute = "test"
|
|
self.assertEqual(testfunction.attribute, "test")
|
|
self.assertRaises(AttributeError, setattr, inst.testfunction, "attribute", "test")
|
|
|
|
|
|
class TestPendingCalls(unittest.TestCase):
|
|
|
|
def pendingcalls_submit(self, l, n):
|
|
def callback():
|
|
#this function can be interrupted by thread switching so let's
|
|
#use an atomic operation
|
|
l.append(None)
|
|
|
|
for i in range(n):
|
|
time.sleep(random.random()*0.02) #0.01 secs on average
|
|
#try submitting callback until successful.
|
|
#rely on regular interrupt to flush queue if we are
|
|
#unsuccessful.
|
|
while True:
|
|
if _testcapi._pending_threadfunc(callback):
|
|
break;
|
|
|
|
def pendingcalls_wait(self, l, n):
|
|
#now, stick around until l[0] has grown to 10
|
|
count = 0;
|
|
while len(l) != n:
|
|
#this busy loop is where we expect to be interrupted to
|
|
#run our callbacks. Note that callbacks are only run on the
|
|
#main thread
|
|
if False and test_support.verbose:
|
|
print("(%i)"%(len(l),),)
|
|
for i in range(1000):
|
|
a = i*i
|
|
count += 1
|
|
self.failUnless(count < 10000,
|
|
"timeout waiting for %i callbacks, got %i"%(n, len(l)))
|
|
if False and test_support.verbose:
|
|
print("(%i)"%(len(l),))
|
|
|
|
def test_pendingcalls_threaded(self):
|
|
l = []
|
|
|
|
#do every callback on a separate thread
|
|
n = 32
|
|
threads = []
|
|
for i in range(n):
|
|
t = threading.Thread(target=self.pendingcalls_submit, args = (l, 1))
|
|
t.start()
|
|
threads.append(t)
|
|
|
|
self.pendingcalls_wait(l, n)
|
|
|
|
for t in threads:
|
|
t.join()
|
|
|
|
def test_pendingcalls_non_threaded(self):
|
|
#again, just using the main thread, likely they will all be dispathced at
|
|
#once. It is ok to ask for too many, because we loop until we find a slot.
|
|
#the loop can be interrupted to dispatch.
|
|
#there are only 32 dispatch slots, so we go for twice that!
|
|
l = []
|
|
n = 64
|
|
self.pendingcalls_submit(l, n)
|
|
self.pendingcalls_wait(l, n)
|
|
|
|
|
|
def test_main():
|
|
support.run_unittest(CAPITest)
|
|
|
|
for name in dir(_testcapi):
|
|
if name.startswith('test_'):
|
|
test = getattr(_testcapi, name)
|
|
if support.verbose:
|
|
print("internal", name)
|
|
test()
|
|
|
|
# some extra thread-state tests driven via _testcapi
|
|
def TestThreadState():
|
|
if support.verbose:
|
|
print("auto-thread-state")
|
|
|
|
idents = []
|
|
|
|
def callback():
|
|
idents.append(_thread.get_ident())
|
|
|
|
_testcapi._test_thread_state(callback)
|
|
a = b = callback
|
|
time.sleep(1)
|
|
# Check our main thread is in the list exactly 3 times.
|
|
if idents.count(_thread.get_ident()) != 3:
|
|
raise support.TestFailed(
|
|
"Couldn't find main thread correctly in the list")
|
|
|
|
try:
|
|
_testcapi._test_thread_state
|
|
have_thread_state = True
|
|
except AttributeError:
|
|
have_thread_state = False
|
|
|
|
if have_thread_state:
|
|
import _thread
|
|
import time
|
|
TestThreadState()
|
|
import threading
|
|
t = threading.Thread(target=TestThreadState)
|
|
t.start()
|
|
t.join()
|
|
|
|
support.run_unittest(TestPendingCalls)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|