Author: Armin Rigo <[email protected]>
Branch: stm-gc
Changeset: r52630:1e22b3703ea6
Date: 2012-02-19 14:08 +0100
http://bitbucket.org/pypy/pypy/changeset/1e22b3703ea6/
Log: Fix test_rstm.
diff --git a/pypy/rlib/rstm.py b/pypy/rlib/rstm.py
--- a/pypy/rlib/rstm.py
+++ b/pypy/rlib/rstm.py
@@ -1,4 +1,4 @@
-import thread
+import threading
from pypy.rlib.objectmodel import specialize, we_are_translated
from pypy.rlib.objectmodel import keepalive_until_here
from pypy.rlib.debug import ll_assert
@@ -9,7 +9,7 @@
llhelper)
from pypy.translator.stm.stmgcintf import StmOperations
-_global_lock = thread.allocate_lock()
+_global_lock = threading.RLock()
@specialize.memo()
def _get_stm_callback(func, argcls):
@@ -32,8 +32,6 @@
def perform_transaction(func, argcls, arg):
ll_assert(arg is None or isinstance(arg, argcls),
"perform_transaction: wrong class")
- ll_assert(argcls._alloc_nonmovable_,
- "perform_transaction: XXX") # XXX kill me
if we_are_translated():
llarg = cast_instance_to_base_ptr(arg)
llarg = rffi.cast(rffi.VOIDP, llarg)
@@ -63,7 +61,10 @@
if not we_are_translated(): _global_lock.release()
def _debug_get_state():
- return StmOperations._debug_get_state()
+ if not we_are_translated(): _global_lock.acquire()
+ res = StmOperations._debug_get_state()
+ if not we_are_translated(): _global_lock.release()
+ return res
def thread_id():
return StmOperations.thread_id()
diff --git a/pypy/rlib/test/test_rstm.py b/pypy/rlib/test/test_rstm.py
--- a/pypy/rlib/test/test_rstm.py
+++ b/pypy/rlib/test/test_rstm.py
@@ -1,38 +1,37 @@
import os, thread, time
-from pypy.rlib.debug import debug_print
+from pypy.rlib.debug import debug_print, ll_assert
from pypy.rlib import rstm
from pypy.translator.stm.test.support import CompiledSTMTests
class Arg(object):
- _alloc_nonmovable_ = True
+ pass
+arg = Arg()
def setx(arg, retry_counter):
debug_print(arg.x)
- assert rstm.debug_get_state() == 1
+ assert rstm._debug_get_state() == 1
if arg.x == 303:
# this will trigger stm_become_inevitable()
os.write(1, "hello\n")
- assert rstm.debug_get_state() == 2
+ assert rstm._debug_get_state() == 2
arg.x = 42
-
-def test_stm_perform_transaction(initial_x=202):
- arg = Arg()
+def stm_perform_transaction(initial_x=202):
arg.x = initial_x
- assert rstm.debug_get_state() == -1
+ ll_assert(rstm._debug_get_state() == -2, "bad debug_get_state (1)")
rstm.descriptor_init()
- assert rstm.debug_get_state() == 0
+ ll_assert(rstm._debug_get_state() == 0, "bad debug_get_state (2)")
rstm.perform_transaction(setx, Arg, arg)
- assert rstm.debug_get_state() == 0
+ ll_assert(rstm._debug_get_state() == 0, "bad debug_get_state (3)")
rstm.descriptor_done()
- assert rstm.debug_get_state() == -1
- assert arg.x == 42
+ ll_assert(rstm._debug_get_state() == -2, "bad debug_get_state (4)")
+ ll_assert(arg.x == 42, "bad arg.x")
def test_stm_multiple_threads():
ok = []
def f(i):
- test_stm_perform_transaction()
+ stm_perform_transaction()
ok.append(i)
for i in range(10):
thread.start_new_thread(f, (i,))
@@ -58,20 +57,34 @@
assert dataout == ''
assert '102' in dataerr.splitlines()
- def test_perform_transaction(self):
+ def build_perform_transaction(self):
+ from pypy.module.thread import ll_thread
+ class Done: done = False
+ done = Done()
+ def g():
+ stm_perform_transaction(done.initial_x)
+ done.done = True
def f(argv):
- test_stm_perform_transaction()
+ done.initial_x = int(argv[1])
+ assert rstm._debug_get_state() == -1 # main thread
+ ll_thread.start_new_thread(g, ())
+ for i in range(20):
+ if done.done: break
+ time.sleep(0.1)
+ else:
+ print "timeout!"
+ raise Exception
return 0
t, cbuilder = self.compile(f)
- dataout, dataerr = cbuilder.cmdexec('', err=True)
+ return cbuilder
+
+ def test_perform_transaction(self):
+ cbuilder = self.build_perform_transaction()
+ #
+ dataout, dataerr = cbuilder.cmdexec('202', err=True)
assert dataout == ''
assert '202' in dataerr.splitlines()
-
- def test_perform_transaction_inevitable(self):
- def f(argv):
- test_stm_perform_transaction(303)
- return 0
- t, cbuilder = self.compile(f)
- dataout, dataerr = cbuilder.cmdexec('', err=True)
+ #
+ dataout, dataerr = cbuilder.cmdexec('303', err=True)
assert 'hello' in dataout.splitlines()
assert '303' in dataerr.splitlines()
_______________________________________________
pypy-commit mailing list
[email protected]
http://mail.python.org/mailman/listinfo/pypy-commit
