Merge pull request plotly#261 from plotly/config-file-safe-tests

aneda · aneda · commit 6f2eb8a4d210 · 2015-07-16T12:37:12.000-04:00
Config file safe tests
diff --git a/plotly/tests/test_core/test_tools/test_file_tools.py b/plotly/tests/test_core/test_tools/test_file_tools.py
@@ -1,10 +1,9 @@
-from unittest import TestCase
-
 from plotly import tools
 from plotly import session
+from plotly.tests.utils import PlotlyTestCase
 
 
-class FileToolsTest(TestCase):
+class FileToolsTest(PlotlyTestCase):
 
     def test_set_config_file_all_entries(self):
 
diff --git a/plotly/tests/utils.py b/plotly/tests/utils.py
@@ -1,4 +1,50 @@
+import copy
+import json
 from numbers import Number as Num
+from unittest import TestCase
+
+from plotly import session
+from plotly.tools import CREDENTIALS_FILE, CONFIG_FILE, _file_permissions
+
+
+class PlotlyTestCase(TestCase):
+
+    # parent test case to assist with clean up of local credentials/config
+
+    def __init__(self, **kwargs):
+        self._file_credentials = None
+        self._file_config = None
+        self._session = None
+        super(PlotlyTestCase, self).__init__(**kwargs)
+
+    def setUp(self):
+        self.stash_file_credentials_and_config()
+
+    def tearDown(self):
+        self.restore_file_credentials_and_config()
+
+    def stash_file_credentials_and_config(self):
+        if _file_permissions:
+            with open(CREDENTIALS_FILE, 'r') as f:
+                self._file_credentials = json.load(f)
+            with open(CONFIG_FILE, 'r') as f:
+                self._file_config = json.load(f)
+
+    def restore_file_credentials_and_config(self):
+        if _file_permissions:
+            if self._file_credentials is not None:
+                with open(CREDENTIALS_FILE, 'w') as f:
+                    json.dump(self._file_credentials, f)
+            if self._file_config is not None:
+                with open(CONFIG_FILE, 'w') as f:
+                    json.load(self._file_config, f)
+
+    def stash_session(self):
+        self._session = copy.deepcopy(session._session)
+
+    def restore_session(self):
+        session._session.clear()  # clear and update to preserve references.
+        session._session.update(self._session)
 
 
 def compare_dict(dict1, dict2, equivalent=True, msg='', tol=10e-8):
