commit c6425a00e3c16268b101703c45f2a3e53154a372
author eesheesh <lironn@google.com> Fri Feb 12 15:07:06 2016 +0000
committer Jon Wayne Parrott <jon.wayne.parrott@gmail.com> Fri Mar 25 12:16:12 2016 -0700
tree 02936a1e3b8159dc6f1cdf5731a2c3b0d0c8215b
parent f99fa88df778c52950ed6eca185a619df9c22e58

Add retry on rate limiting API responses and network timeouts

tests/test_discovery.py

diff --git a/tests/test_discovery.py b/tests/test_discovery.py
index dea5631..dac581f 100644
--- a/tests/test_discovery.py
+++ b/tests/test_discovery.py
@@ -440,10 +440,10 @@
     self.orig_import = __import__
     self.mocked_api = mock.MagicMock()
 
-    def import_mock(name, *args):
+    def import_mock(name, *args, **kwargs):
       if name == 'google.appengine.api':
         return self.mocked_api
-      return self.orig_import(name, *args)
+      return self.orig_import(name, *args, **kwargs)
 
     import_fullname = '__builtin__.__import__'
     if sys.version_info[0] >= 3: