neilkumar101
diff --git a/‎plotly/tests/utils.py
Copy file name to clipboardExpand all lines: plotly/tests/utils.py
+36Lines changed: 36 additions & 0 deletions b/‎plotly/tests/utils.py
Copy file name to clipboardExpand all lines: plotly/tests/utils.py
+36Lines changed: 36 additions & 0 deletions
@@ -1,4 +1,40 @@
+import json
 from numbers import Number as Num
+from unittest import TestCase
+
+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._credentials = None
+        self._config = None
+        super(PlotlyTestCase, self).__init__(**kwargs)
+
+    def setUp(self):
+        self.stash_credentials_and_config()
+
+    def tearDown(self):
+        self.restore_credentials_and_config()
+
+    def stash_credentials_and_config(self):
+        if _file_permissions:
+            with open(CREDENTIALS_FILE, 'r') as f:
+                self._credentials = json.load(f)
+            with open(CONFIG_FILE, 'r') as f:
+                self._config = json.load(f)
+
+    def restore_credentials_and_config(self):
+        if _file_permissions:
+            if self._credentials is not None:
+                with open(CREDENTIALS_FILE, 'w') as f:
+                    json.dump(self._credentials, f)
+            if self._config is not None:
+                with open(CONFIG_FILE, 'w') as f:
+                    json.load(self._config, f)
 
 
 def compare_dict(dict1, dict2, equivalent=True, msg='', tol=10e-8):