From 13344ad06cbbd36938e777ec285b7a1d26f2b4b5 Mon Sep 17 00:00:00 2001
From: Piotr Maslanka <piotr.maslanka@henrietta.com.pl>
Date: Sun, 8 Oct 2017 04:45:07 +0200
Subject: [PATCH] o
---
coolamqp/clustering/cluster.py | 2 +-
tests/test_clustering/test_things.py | 3 +--
2 files changed, 2 insertions(+), 3 deletions(-)
diff --git a/coolamqp/clustering/cluster.py b/coolamqp/clustering/cluster.py
index ba120ff..5d77c96 100644
--- a/coolamqp/clustering/cluster.py
+++ b/coolamqp/clustering/cluster.py
@@ -178,7 +178,7 @@ class Cluster(object):
if self.on_fail is not None:
self.snr.on_fail.add(
(lambda snr, of: lambda: of() if not snr.terminating else None)(
- self.snr, self.on_fail)
+ self.listener, self.on_fail)
)
# Spawn a transactional publisher and a noack publisher
diff --git a/tests/test_clustering/test_things.py b/tests/test_clustering/test_things.py
index 4db5069..d5a04e6 100644
--- a/tests/test_clustering/test_things.py
+++ b/tests/test_clustering/test_things.py
@@ -25,8 +25,7 @@ class TestConnecting(unittest.TestCase):
def test_on_clean(self):
q = {'failed': False}
c = Cluster([NODE], on_fail=lambda: q.update(failed=True))
- c.start()
- time.sleep(5)
+ c.start(wait=True)
c.shutdown()
time.sleep(5)
self.assertFalse(q['failed'])
--
GitLab