Merge pull request influxdata#178 from influxdb/issue_176

aviau · aviau · commit b022683ce116 · 2015-05-05T14:41:04.000-04:00
Allow mixed case in DSN (closes influxdata#176)
diff --git a/influxdb/client.py b/influxdb/client.py
@@ -142,7 +142,6 @@ def from_DSN(dsn, **kwargs):
             be used for the TCP connection; specify the UDP port with the
             additional `udp_port` parameter (cf. examples).
         """
-        dsn = dsn.lower()
 
         init_args = {}
         conn_params = urlparse(dsn)
@@ -760,7 +759,6 @@ def from_DSN(dsn, client_base_class=InfluxDBClient,
         [<influxdb.client.InfluxDBClient at 0x7feb480295d0>,
          <influxdb.client.InfluxDBClient at 0x7feb438ec950>]
         """
-        dsn = dsn.lower()
         conn_params = urlparse(dsn)
         netlocs = conn_params.netloc.split(',')
         cluster_client = InfluxDBClusterClient(
diff --git a/tests/influxdb/client_test.py b/tests/influxdb/client_test.py
@@ -853,3 +853,18 @@ def test_dsn_single_client(self):
             'https+influxdb://usr:pwd@host:8086/db',
             **{'ssl': False})
         self.assertEqual('http://host:8086', cli.clients[0]._baseurl)
+
+    def test_dsn_password_caps(self):
+        cli = InfluxDBClusterClient.from_DSN(
+            'https+influxdb://usr:pWd@host:8086/db')
+        self.assertEqual('pWd', cli.clients[0]._password)
+
+    def test_dsn_mixed_scheme_case(self):
+        cli = InfluxDBClusterClient.from_DSN(
+            'hTTps+inFLUxdb://usr:pWd@host:8086/db')
+        self.assertEqual('pWd', cli.clients[0]._password)
+        self.assertEqual('https://host:8086', cli.clients[0]._baseurl)
+
+        cli = InfluxDBClusterClient.from_DSN(
+            'uDP+influxdb://usr:pwd@host1:8086,usr:pwd@host2:8086/db')
+        self.assertTrue(cli.clients[0].use_udp)
