diff --git a/patroni/dcs/consul.py b/patroni/dcs/consul.py index a03564fa..51aa4dd8 100644 --- a/patroni/dcs/consul.py +++ b/patroni/dcs/consul.py @@ -158,7 +158,8 @@ class Consul(AbstractDCS): if config.get('key') and config.get('cert'): config['cert'] = (config['cert'], config['key']) - kwargs = {p: config.get(p) for p in ('host', 'port', 'token', 'scheme', 'cert', 'ca_cert') if config.get(p)} + config_keys = ('host', 'port', 'token', 'scheme', 'cert', 'ca_cert', 'dc') + kwargs = {p: config.get(p) for p in config_keys if config.get(p)} verify = config.get('verify') if not isinstance(verify, bool): diff --git a/tests/test_consul.py b/tests/test_consul.py index 898b50d2..5703eb28 100644 --- a/tests/test_consul.py +++ b/tests/test_consul.py @@ -74,7 +74,7 @@ class TestConsul(unittest.TestCase): @patch.object(consul.Consul.KV, 'delete', Mock()) def setUp(self): Consul({'ttl': 30, 'scope': 't', 'name': 'p', 'url': 'https://l:1', 'retry_timeout': 10, - 'verify': 'on', 'key': 'foo', 'cert': 'bar', 'cacert': 'buz', 'token': 'asd'}) + 'verify': 'on', 'key': 'foo', 'cert': 'bar', 'cacert': 'buz', 'token': 'asd', 'dc': 'dc1'}) Consul({'ttl': 30, 'scope': 't', 'name': 'p', 'url': 'https://l:1', 'retry_timeout': 10, 'verify': 'on', 'cert': 'bar', 'cacert': 'buz'}) self.c = Consul({'ttl': 30, 'scope': 'test', 'name': 'postgresql1', 'host': 'localhost:1', 'retry_timeout': 10})