Add tests for sentinel module

1 files changed, 173 insertions, 0 deletions

diff --git a/tests/test_sentinel.py b/tests/test_sentinel.py
new file mode 100644
index 0000000..ca85615
--- /dev/null
+++ b/tests/test_sentinel.py
@@ -0,0 +1,173 @@
+from __future__ import with_statement
+import pytest
+
+from redis import exceptions
+from redis.sentinel import (Sentinel, SentinelConnectionPool,
+                            MasterNotFoundError, SlaveNotFoundError)
+from redis._compat import next
+import redis.sentinel
+
+
+class SentinelTestClient(object):
+    def __init__(self, cluster, id):
+        self.cluster = cluster
+        self.id = id
+
+    def sentinel_masters(self):
+        self.cluster.connection_error_if_down(self)
+        return {self.cluster.service_name: self.cluster.master}
+
+    def sentinel_slaves(self, master_name):
+        self.cluster.connection_error_if_down(self)
+        if master_name != self.cluster.service_name:
+            return []
+        return self.cluster.slaves
+
+
+class SentinelTestCluster(object):
+    def __init__(self, service_name='mymaster', ip='127.0.0.1', port=6379):
+        self.clients = {}
+        self.master = {
+            'ip': ip,
+            'port': port,
+            'is_master': True,
+            'is_sdown': False,
+            'is_odown': False,
+            'num-other-sentinels': 0,
+            }
+        self.service_name = service_name
+        self.slaves = []
+        self.nodes_down = set()
+
+    def connection_error_if_down(self, node):
+        if node.id in self.nodes_down:
+            raise exceptions.ConnectionError
+
+    def client(self, host, port, **kwargs):
+        return SentinelTestClient(self, (host, port))
+
+
+@pytest.fixture()
+def cluster(request):
+    def teardown():
+        redis.sentinel.StrictRedis = saved_StrictRedis
+    cluster = SentinelTestCluster()
+    saved_StrictRedis = redis.sentinel.StrictRedis
+    redis.sentinel.StrictRedis = cluster.client
+    request.addfinalizer(teardown)
+    return cluster
+
+
+@pytest.fixture()
+def sentinel(request, cluster):
+    return Sentinel([('foo', 26379), ('bar', 26379)])
+
+
+def test_discover_master(sentinel):
+    address = sentinel.discover_master('mymaster')
+    assert address == ('127.0.0.1', 6379)
+
+
+def test_discover_master_error(sentinel):
+    with pytest.raises(MasterNotFoundError):
+        sentinel.discover_master('xxx')
+
+
+def test_discover_master_sentinel_down(cluster, sentinel):
+    # Put first sentinel 'foo' down
+    cluster.nodes_down.add(('foo', 26379))
+    address = sentinel.discover_master('mymaster')
+    assert address == ('127.0.0.1', 6379)
+    # 'bar' is now first sentinel
+    assert sentinel.sentinels[0].id == ('bar', 26379)
+
+
+def test_master_min_other_sentinels(cluster):
+    sentinel = Sentinel([('foo', 26379)], min_other_sentinels=1)
+    # min_other_sentinels
+    with pytest.raises(MasterNotFoundError):
+        sentinel.discover_master('mymaster')
+    cluster.master['num-other-sentinels'] = 2
+    address = sentinel.discover_master('mymaster')
+    assert address == ('127.0.0.1', 6379)
+
+
+def test_master_odown(cluster, sentinel):
+    cluster.master['is_odown'] = True
+    with pytest.raises(MasterNotFoundError):
+        sentinel.discover_master('mymaster')
+
+
+def test_master_sdown(cluster, sentinel):
+    cluster.master['is_sdown'] = True
+    with pytest.raises(MasterNotFoundError):
+        sentinel.discover_master('mymaster')
+
+
+def test_discover_slaves(cluster, sentinel):
+    assert sentinel.discover_slaves('mymaster') == []
+
+    cluster.slaves = [
+        {'ip': 'slave0', 'port': 1234, 'is_odown': False, 'is_sdown': False},
+        {'ip': 'slave1', 'port': 1234, 'is_odown': False, 'is_sdown': False},
+    ]
+    assert sentinel.discover_slaves('mymaster') == [
+        ('slave0', 1234), ('slave1', 1234)]
+
+    # slave0 -> ODOWN
+    cluster.slaves[0]['is_odown'] = True
+    assert sentinel.discover_slaves('mymaster') == [
+        ('slave1', 1234)]
+
+    # slave1 -> SDOWN
+    cluster.slaves[1]['is_sdown'] = True
+    assert sentinel.discover_slaves('mymaster') == []
+
+    cluster.slaves[0]['is_odown'] = False
+    cluster.slaves[1]['is_sdown'] = False
+
+    # node0 -> DOWN
+    cluster.nodes_down.add(('foo', 26379))
+    assert sentinel.discover_slaves('mymaster') == [
+        ('slave0', 1234), ('slave1', 1234)]
+
+
+def test_master_for(cluster, sentinel):
+    master = sentinel.master_for('mymaster', db=9)
+    assert master.ping()
+    assert master.connection_pool.master_address == ('127.0.0.1', 6379)
+
+    # Use internal connection check
+    master = sentinel.master_for('mymaster', db=9, check_connection=True)
+    assert master.ping()
+
+
+def test_slave_for(cluster, sentinel):
+    cluster.slaves = [
+        {'ip': '127.0.0.1', 'port': 6379,
+         'is_odown': False, 'is_sdown': False},
+    ]
+    slave = sentinel.slave_for('mymaster', db=9)
+    assert slave.ping()
+
+
+def test_slave_for_slave_not_found_error(cluster, sentinel):
+    cluster.master['is_odown'] = True
+    slave = sentinel.slave_for('mymaster', db=9)
+    with pytest.raises(SlaveNotFoundError):
+        slave.ping()
+
+
+def test_slave_round_robin(cluster, sentinel):
+    cluster.slaves = [
+        {'ip': 'slave0', 'port': 6379, 'is_odown': False, 'is_sdown': False},
+        {'ip': 'slave1', 'port': 6379, 'is_odown': False, 'is_sdown': False},
+    ]
+    pool = SentinelConnectionPool('mymaster', sentinel)
+    rotator = pool.rotate_slaves()
+    assert next(rotator) in (('slave0', 6379), ('slave1', 6379))
+    assert next(rotator) in (('slave0', 6379), ('slave1', 6379))
+    # Fallback to master
+    assert next(rotator) == ('127.0.0.1', 6379)
+    with pytest.raises(SlaveNotFoundError):
+        next(rotator)