add ssl_verification_enabled to streaming_specs + tests

BRONSOLO · BRONSOLO · commit 18baeb573a8f · 2016-06-03T17:37:36.000-04:00
diff --git a/plotly/plotly/plotly.py b/plotly/plotly/plotly.py
@@ -463,6 +463,7 @@ def get_streaming_specs(self):
 
         """
         streaming_url = get_config()['plotly_streaming_domain']
+        ssl_verification_enabled = get_config()['plotly_ssl_verification']
         ssl_enabled = 'https' in streaming_url
         port = self.HTTPS_PORT if ssl_enabled else self.HTTP_PORT
 
@@ -476,6 +477,7 @@ def get_streaming_specs(self):
             'server': host,
             'port': port,
             'ssl_enabled': ssl_enabled,
+            'ssl_verification_enabled': ssl_verification_enabled,
             'headers': headers
         }
 
diff --git a/plotly/tests/test_core/test_stream/test_stream.py b/plotly/tests/test_core/test_stream/test_stream.py
@@ -18,7 +18,8 @@
 tk = 'vaia8trjjb'
 config = {'plotly_domain': 'https://plot.ly',
           'plotly_streaming_domain': 'stream.plot.ly',
-          'plotly_api_domain': 'https://api.plot.ly'}
+          'plotly_api_domain': 'https://api.plot.ly', 
+          'plotly_ssl_verification': False}
 
 
 class TestStreaming(TestCase):
@@ -147,6 +148,7 @@ def test_stream_http(self):
             'server': 'stream.plot.ly',
             'port': 80,
             'ssl_enabled': False,
+            'ssl_verification_enabled': False, 
             'headers': {
                 'Host': 'stream.plot.ly',
                 'plotly-streamtoken': tk
@@ -158,15 +160,20 @@ def test_stream_http(self):
     def test_stream_https(self):
 
         # If the https scheme is used in the plotly_streaming_domain, port 443
-        # should be used for streaming and ssl_enabled should be True
+        # should be used for streaming, ssl_enabled should be True,
+        # and ssl_verification_enabled should equal plotly_ssl_verification
 
-        py.sign_in(un, ak,
-                   **{'plotly_streaming_domain': 'https://stream.plot.ly'})
+        ssl_stream_config = {
+            'plotly_streaming_domain': 'https://stream.plot.ly',
+            'plotly_ssl_verification': True
+        }
+        py.sign_in(un, ak, **ssl_stream_config)
         my_stream = py.Stream(tk)
         expected_streaming_specs = {
             'server': 'stream.plot.ly',
             'port': 443,
             'ssl_enabled': True,
+            'ssl_verification_enabled': True, 
             'headers': {
                 'Host': 'stream.plot.ly',
                 'plotly-streamtoken': tk
