summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/test_context.py19
1 files changed, 17 insertions, 2 deletions
diff --git a/tests/test_context.py b/tests/test_context.py
index b5de372..6a9afcd 100644
--- a/tests/test_context.py
+++ b/tests/test_context.py
@@ -10,6 +10,21 @@ def test_temp_connection():
receiver = lambda sender: canary.append(sender)
sig.send(1)
+ with sig.connected_to(receiver):
+ sig.send(2)
+ sig.send(3)
+
+ assert canary == [2]
+ assert not sig.receivers
+
+
+def test_temp_connection_alias():
+ sig = Signal()
+
+ canary = []
+ receiver = lambda sender: canary.append(sender)
+
+ sig.send(1)
with sig.temporarily_connected_to(receiver):
sig.send(2)
sig.send(3)
@@ -24,7 +39,7 @@ def test_temp_connection_for_sender():
canary = []
receiver = lambda sender: canary.append(sender)
- with sig.temporarily_connected_to(receiver, sender=2):
+ with sig.connected_to(receiver, sender=2):
sig.send(1)
sig.send(2)
@@ -43,7 +58,7 @@ def test_temp_connection_failure():
try:
sig.send(1)
- with sig.temporarily_connected_to(receiver):
+ with sig.connected_to(receiver):
sig.send(2)
raise Failure
sig.send(3)