Raise PlotlyError if sign_in fails.

Author: theengineear

CHANGELOG.md

@@ -10,6 +10,8 @@ failures. Previously either a `PlotlyError`, `PlotlyRequestException`, or a
 `requests.exceptions.ReqestException` could be raised. In particular, scripts
 which depend on `try-except` blocks containing network requests should be
 revisited.
+- `plotly.py:sign_in` now validates to the plotly server specified in your
+  config. If it cannot make a successful request, it raises a `PlotlyError`.
 
 ### Deprecated
 - `plotly.tools.FigureFactory`. Use `plotly.figure_factory.*`.

plotly/plotly/plotly.py

@@ -48,7 +48,15 @@
 
 
 # don't break backwards compatibility
-sign_in = session.sign_in
+def sign_in(username, api_key, **kwargs):
+    session.sign_in(username, api_key, **kwargs)
+    try:
+        # The only way this can succeed is if the user can be authenticated
+        # with the given, username, api_key, and plotly_api_domain.
+        v2.users.current()
+    except exceptions.PlotlyRequestError:
+        raise exceptions.PlotlyError('Sign in failed.')
+
 update_plot_options = session.update_session_plot_options
 
 

plotly/tests/test_core/test_plotly/test_credentials.py

@@ -1,39 +1,43 @@
 from __future__ import absolute_import
 
-from unittest import TestCase
+from mock import patch
 
 import plotly.plotly.plotly as py
 import plotly.session as session
 import plotly.tools as tls
+from plotly import exceptions
+from plotly.tests.utils import PlotlyTestCase
 
 
-def test_get_credentials():
-    session_credentials = session.get_session_credentials()
-    if 'username' in session_credentials:
-        del session._session['credentials']['username']
-    if 'api_key' in session_credentials:
-        del session._session['credentials']['api_key']
-    creds = py.get_credentials()
-    file_creds = tls.get_credentials_file()
-    print(creds)
-    print(file_creds)
-    assert creds == file_creds
+class TestSignIn(PlotlyTestCase):
 
+    def setUp(self):
+        super(TestSignIn, self).setUp()
+        patcher = patch('plotly.api.v2.users.current')
+        self.users_current_mock = patcher.start()
+        self.addCleanup(patcher.stop)
 
-def test_sign_in():
-    un = 'anyone'
-    ak = 'something'
-    # TODO, add this!
-    # si = ['this', 'and-this']
-    py.sign_in(un, ak)
-    creds = py.get_credentials()
-    assert creds['username'] == un
-    assert creds['api_key'] == ak
-    # TODO, and check it!
-    # assert creds['stream_ids'] == si
+    def test_get_credentials(self):
+        session_credentials = session.get_session_credentials()
+        if 'username' in session_credentials:
+            del session._session['credentials']['username']
+        if 'api_key' in session_credentials:
+            del session._session['credentials']['api_key']
+        creds = py.get_credentials()
+        file_creds = tls.get_credentials_file()
+        self.assertEqual(creds, file_creds)
 
-
-class TestSignIn(TestCase):
+    def test_sign_in(self):
+        un = 'anyone'
+        ak = 'something'
+        # TODO, add this!
+        # si = ['this', 'and-this']
+        py.sign_in(un, ak)
+        creds = py.get_credentials()
+        self.assertEqual(creds['username'], un)
+        self.assertEqual(creds['api_key'], ak)
+        # TODO, and check it!
+        # assert creds['stream_ids'] == si
 
     def test_get_config(self):
         plotly_domain = 'test domain'
@@ -74,3 +78,10 @@ def test_sign_in_with_config(self):
         self.assertEqual(
             config['plotly_ssl_verification'], plotly_ssl_verification
         )
+
+    def test_sign_in_cannot_validate(self):
+        self.users_current_mock.side_effect = exceptions.PlotlyRequestError(
+            'msg', 400, 'foobar'
+        )
+        with self.assertRaisesRegexp(exceptions.PlotlyError, 'Sign in failed'):
+            py.sign_in('foo', 'bar')

plotly/tests/test_core/test_plotly/test_plot.py

@@ -12,6 +12,7 @@
 from requests.compat import json as _json
 
 from unittest import TestCase
+from mock import patch
 from nose.plugins.attrib import attr
 from nose.tools import raises
 
@@ -223,6 +224,14 @@ class TestPlotOptionLogic(PlotlyTestCase):
         {'world_readable': False, 'sharing': 'public'}
     )
 
+    def setUp(self):
+        super(TestPlotOptionLogic, self).setUp()
+
+        # Make sure we don't hit sign-in validation failures.
+        patcher = patch('plotly.api.v2.users.current')
+        self.users_current_mock = patcher.start()
+        self.addCleanup(patcher.stop)
+
     def test_default_options(self):
         options = py._plot_option_logic({})
         config_options = tls.get_config_file()