add tests for CONSUL_HTTP_ADDR with and without scheme and CONSUL_HTTP_SSL

asteven · asteven · commit 905b59834e14 · 2020-09-11T21:10:09.000+02:00
Signed-off-by: Steven Armstrong &lt;steven.armstrong@id.ethz.ch&gt;
diff --git a/tests/test_base.py b/tests/test_base.py
@@ -60,6 +60,19 @@ def test_CONSUL_HTTP_ADDR(self):
             c = Consul.from_env()
             assert c.http.base_uri == CONSUL_HTTP_ADDR
 
+    def test_CONSUL_HTTP_ADDR_scheme_http(self):
+        CONSUL_HTTP_ADDR = '127.0.0.23:4242'
+        with self.environ(CONSUL_HTTP_ADDR=CONSUL_HTTP_ADDR):
+            c = Consul.from_env()
+            assert c.http.base_uri == 'http://'+ CONSUL_HTTP_ADDR
+
+    def test_CONSUL_HTTP_ADDR_with_CONSUL_HTTP_SSL(self):
+        CONSUL_HTTP_ADDR = '127.0.0.23:4242'
+        with self.environ(CONSUL_HTTP_ADDR=CONSUL_HTTP_ADDR,
+                          CONSUL_HTTP_SSL='true'):
+            c = Consul.from_env()
+            assert c.http.base_uri == 'https://'+ CONSUL_HTTP_ADDR
+
     def test_CONSUL_HTTP_TOKEN(self):
         CONSUL_HTTP_TOKEN = '1bdc2cb4-9b02-4b3c-9df5-eb86214e1a6c'
         with self.environ(CONSUL_HTTP_TOKEN=CONSUL_HTTP_TOKEN):
