Add RedisCluster support

redbeat/schedulers.py

@@ -116,7 +116,11 @@ def get_redis(app=None):
     if not hasattr(app, 'redbeat_redis') or app.redbeat_redis is None:
         redis_options = conf.redbeat_redis_options
         retry_period = redis_options.get('retry_period')
-        if conf.redis_url.startswith('redis-sentinel') and 'sentinels' in redis_options:
+        if redis_options.get('cluster', False):
+            from redis.cluster import RedisCluster
+
+            connection = RedisCluster.from_url(conf.redis_url, **redis_options)
+        elif conf.redis_url.startswith('redis-sentinel') and 'sentinels' in redis_options:
             from redis.sentinel import Sentinel
 
             sentinel = Sentinel(
@@ -205,7 +209,7 @@ def __init__(
         **clsargs,
     ):
         super().__init__(
-            name=name,
+            name=str(name),
             task=task,
             schedule=schedule,
             args=args,

tests/test_scheduler.py

@@ -2,6 +2,7 @@
 import unittest
 from copy import deepcopy
 from datetime import datetime, timedelta
+from unittest import mock
 from unittest.mock import Mock, patch
 
 import pytz
@@ -200,6 +201,24 @@ def test_sentinel_scheduler(self):
         assert 'Sentinel' not in str(redis_client.connection_pool)
 
 
+class ClusterRedBeatCase(AppCase):
+    config_dict = {
+        'BROKER_URL': 'redis://',
+        'REDBEAT_REDIS_OPTIONS': {
+            'cluster': True,
+        },
+    }
+
+    def setup(self):
+        self.app.conf.update(self.config_dict)
+
+    def test_sentinel_scheduler(self):
+        # Fake redis doesn't really support redis cluster, but let's just make sure it was used.
+        with mock.patch('redis.RedisCluster.from_url') as from_url:
+            get_redis(app=self.app)
+            self.assertTrue(from_url.called)
+
+
 class SentinelRedBeatCase(AppCase):
     config_dict = {
         'REDBEAT_KEY_PREFIX': 'rb-tests:',