ENH: function to reset Google Analytics token store pandas-dev#2687

changhiskhan · changhiskhan · commit abb4f2b0f0fd · 2013-01-20T00:12:35.000-05:00
diff --git a/pandas/io/auth.py b/pandas/io/auth.py
@@ -127,3 +127,7 @@ def init_service(http):
     Use the given http object to build the analytics service object
     """
     return gapi.build('analytics', 'v3', http=http)
+
+def reset_default_token_store():
+    import os
+    os.remove(DEFAULT_TOKEN_FILE)
diff --git a/pandas/io/ga.py b/pandas/io/ga.py
@@ -86,6 +86,11 @@
     Local host redirect if unspecified
 """
 
+def reset_token_store():
+    """
+    Deletes the default token store
+    """
+    auth.reset_default_token_store()
 
 @Substitution(extras=_AUTH_PARAMS)
 @Appender(_GA_READER_DOC)
diff --git a/pandas/io/tests/test_ga.py b/pandas/io/tests/test_ga.py
@@ -13,6 +13,21 @@ class TestGoogle(unittest.TestCase):
 
     _multiprocess_can_split_ = True
 
+    def test_remove_token_store(self):
+        import os
+        try:
+            import pandas.io.auth as auth
+            from pandas.io.ga import reset_token_store
+        except ImportError:
+            raise nose.SkipTest
+
+        auth.DEFAULT_TOKEN_FILE = 'test.dat'
+        with open(auth.DEFAULT_TOKEN_FILE, 'w') as fh:
+            fh.write('test')
+
+        reset_token_store()
+        self.assert_(not os.path.exists(auth.DEFAULT_TOKEN_FILE))
+
     @slow
     @network
     def test_getdata(self):
