From 57322c22abf7f21c9d76aede216a8ac5b21cf496 Mon Sep 17 00:00:00 2001
From: Sergey Azovskov <last-g@yandex-team.ru>
Date: Thu, 14 Jan 2016 01:11:00 +0500
Subject: [PATCH] [sentinel] Simple unittests for RedisSentinel transport
 support

---
 kombu/tests/transport/test_redis.py | 40 ++++++++++++++++++++++++++++-
 1 file changed, 39 insertions(+), 1 deletion(-)

diff --git a/kombu/tests/transport/test_redis.py b/kombu/tests/transport/test_redis.py
index a67a9653..56bd6633 100644
--- a/kombu/tests/transport/test_redis.py
+++ b/kombu/tests/transport/test_redis.py
@@ -14,7 +14,7 @@ from kombu.utils import eventio  # patch poll
 from kombu.utils.json import dumps
 
 from kombu.tests.case import (
-    Case, ContextMock, Mock, call, module_exists, skip_if_not_module, patch,
+    Case, ContextMock, Mock, call, module_exists, skip_if_not_module, patch, ANY,
 )
 
 
@@ -1280,3 +1280,41 @@ class test_Mutex(Case):
             with redis.Mutex(client, 'foo1', 100):
                 held = True
             self.assertTrue(held)
+
+
+class test_RedisSentinel(Case):
+    def test_method_called(self):
+        from kombu.transport.redis import SentinelChannel
+
+        with patch.object(SentinelChannel, '_sentinel_managed_pool') as patched:
+            connection = Connection('sentinel://localhost:65534/', transport_options={
+                'master_name': 'not_important'
+            })
+
+            connection.channel()
+            self.assertTrue(patched.called)
+
+    def test_getting_master_from_sentinel(self):
+        from redis.sentinel import Sentinel
+
+        with patch.object(Sentinel, '__new__') as patched:
+            connection = Connection('sentinel://localhost:65534/', transport_options={
+                'master_name': 'not_important'
+            })
+
+            connection.channel()
+            self.assertTrue(patched)
+
+            sentinel_obj = patched.return_value
+            self.assertTrue(sentinel_obj.master_for.called, 'master_for was not called')
+            sentinel_obj.master_for.assert_called_with('not_important', ANY)
+            self.assertTrue(sentinel_obj.master_for().connection_pool.get_connection.called, 'get_connection on redis connection pool was not called')
+
+    def test_can_create_connection(self):
+        from redis.exceptions import ConnectionError
+
+        with self.assertRaises(ConnectionError):
+            connection = Connection('sentinel://localhost:65534/', transport_options={
+                'master_name': 'not_important'
+            })
+            connection.channel()