def test_lock_cancel_mid_acquire():
a = 0
lock = tap.Lock()
def acquire():
release = yield lock.Acquire()
nonlocal a
a += 5
yield tap.Receive("unlock")
yield release
def fn():
yield tap.CallFork(acquire)
t = yield tap.CallFork(acquire)
yield tap.CallFork(acquire)
yield tap.Sleep(0)
assert a == 5
yield tap.Cancel(t)
yield tap.Broadcast("unlock")
yield tap.Sleep(0.01)
assert a == 10
yield tap.Broadcast("unlock")
yield tap.Sleep(0.001)
assert a == 10
tap.run(fn)
def test_create_acquires_out_of_order():
lock = tap.Lock()
def fn():
a1 = lock.Acquire()
a2 = lock.Acquire()
r2 = yield a2
yield r2
r1 = yield a1
yield r1
tap.run(fn)
def test_lock_hang():
lock = tap.Lock()
def fn():
yield lock.Acquire()
yield lock.Acquire()
with pytest.raises(tap.TapystryError) as x:
tap.run(fn)
# TODO fix
assert str(x.value).startswith("Hanging strands detected waiting for Receive(lock") or \
str(x.value).startswith("Hanging strands detected waiting for Call(Acquire")
def test_release_twice():
lock = tap.Lock()
def dummy(eff):
yield eff
def fn():
release = yield lock.Acquire()
yield tap.Call(dummy, (release, ))
yield release
with pytest.raises(tap.TapystryError) as x:
tap.run(fn)
# print(x.value)
assert str(x.value).startswith("Exception caught at")
assert str(x.value).count("in test_release_twice\n") == 1
assert str(x.value).count("in Acquire\n") == 1
assert str(x.value).count("Yielded same lock release multiple times?") == 1
def test_lock_cancel_after_acquire():
lock = tap.Lock()
def acquire():
release = yield lock.Acquire()
yield tap.Receive("unlock")
yield release
def fn():
t = yield tap.CallFork(acquire)
yield tap.Sleep(0.001)
yield tap.Cancel(t)
t = yield tap.CallFork(acquire)
yield tap.Sleep(0.001)
with pytest.raises(tap.TapystryError) as x:
tap.run(fn)
# TODO fix
assert str(x.value).startswith("Hanging strands detected waiting for Receive(lock") or \
str(x.value).startswith("Hanging strands detected waiting for Call(Acquire")
def test_lock():
a = 0
lock = tap.Lock()
def waits():
yield tap.Receive("msg")
nonlocal a
a += 1
release = yield lock.Acquire()
a += 2
yield release
def nowaits():
release = yield lock.Acquire()
nonlocal a
a += 5
yield tap.Receive("unlock")
yield release
def fn():
yield tap.CallFork(waits)
yield tap.CallFork(nowaits)
yield tap.CallFork(nowaits)
yield tap.Sleep(0)
assert a == 5
# the waiting strand finally gets to acquire lock, but it is the latest
yield tap.Broadcast("msg")
yield tap.Sleep(0)
assert a == 6
yield tap.Broadcast("unlock")
yield tap.Sleep(0.01)
assert a == 11
yield tap.Broadcast("unlock")
yield tap.Sleep(0.01)
assert a == 13
tap.run(fn)
def test_lock_cancel_mid_acquire_trickier():
a = 0
lock = tap.Lock()
def acquire(x, to_cancel):
yield tap.Receive(str(x))
release = yield lock.Acquire()
nonlocal a
a += 5
yield tap.Sequence([
tap.Receive(str(x)),
tap.Sequence([
tap.Cancel(to_cancel),
release,
]) if to_cancel is not None else release
])
def fn():
t1 = yield tap.CallFork(acquire, (1, None))
t2 = yield tap.CallFork(acquire, (2, t1))
t3 = yield tap.CallFork(acquire, (3, None))
yield tap.Broadcast("2")
yield tap.Broadcast("1")
yield tap.Sleep(0.001)
assert a == 5
yield tap.Broadcast("3")
yield tap.Broadcast(
"2") # this simultaneously tries to cancel 1 and unlocks
yield tap.Sleep(0.01)
assert a == 10
yield tap.Broadcast("1")
# 1 gets canceled after the acquire, so it's too late to release
yield tap.Sleep(0.001)
assert a == 10
# yield tap.Join([t1, t2, t3])
yield tap.Cancel(t3)
tap.run(fn)