Add sign_in test with config info.

Author: theengineear

plotly/plotly/plotly.py

@@ -52,12 +52,16 @@
 ### _credentials stuff ###
 
 
-def sign_in(username, api_key):
+def sign_in(username, api_key, **kwargs):
     """Set module-scoped _credentials for session. Verify with plotly."""
     global _credentials
     _credentials['username'], _credentials['api_key'] = username, api_key
     # TODO: verify these _credentials with plotly
 
+    global _config
+    _config['plotly_domain'] = kwargs.get('plotly_domain')
+    _config['plotly_streaming_domain'] = kwargs.get('plotly_streaming_domain')
+
 
 ### plot options stuff ###
 

plotly/tests/test_core/test_plotly/test_credentials.py

@@ -45,3 +45,20 @@ def test_get_config(self):
         self.assertEqual(
             config3['plotly_streaming_domain'], plotly_streaming_domain
         )
+
+    def test_sign_in_with_config(self):
+        username = 'place holder'
+        api_key = 'place holder'
+        plotly_domain = 'test domain'
+        plotly_streaming_domain = 'test streaming domain'
+        py.sign_in(
+            username,
+            api_key,
+            plotly_domain=plotly_domain,
+            plotly_streaming_domain=plotly_streaming_domain
+        )
+        config = py.get_config()
+        self.assertEqual(config['plotly_domain'], plotly_domain)
+        self.assertEqual(
+            config['plotly_streaming_domain'], plotly_streaming_domain
+        )