Added skip_if_iam to skip tests when running with IAM auth

Author: emlaver

tests/unit/database_tests.py

@@ -38,7 +38,7 @@
 from cloudant.feed import Feed, InfiniteFeed
 from tests.unit._test_util import LONG_NUMBER
 
-from .unit_t_db_base import skip_if_not_cookie_auth, UnitTestDbBase
+from .unit_t_db_base import skip_if_not_cookie_auth, UnitTestDbBase, skip_if_iam
 from .. import unicode_
 
 class CloudantDatabaseExceptionTests(unittest.TestCase):
@@ -863,7 +863,7 @@ def test_get_show_result(self):
             resp,
             'Hello from doc001!'
         )
-
+    @skip_if_iam
     def test_create_doc_with_update_handler(self):
         """
         Test update_handler_result executes an update handler function
@@ -884,6 +884,7 @@ def test_create_doc_with_update_handler(self):
             'Created new doc: {"message":"hello","_id":"testDoc"}'
         )
 
+    @skip_if_iam
     def test_update_doc_with_update_handler(self):
         """
         Test update_handler_result executes an update handler function

tests/unit/design_document_tests.py

@@ -32,7 +32,7 @@
 from cloudant.view import View, QueryIndexView
 from cloudant.error import CloudantArgumentError, CloudantDesignDocumentException
 
-from .unit_t_db_base import UnitTestDbBase
+from .unit_t_db_base import UnitTestDbBase, skip_if_iam
 
 class CloudantDesignDocumentExceptionTests(unittest.TestCase):
     """
@@ -1252,6 +1252,7 @@ def test_get_search_index(self):
                 '{"store": true}); }\n}'}
         )
 
+    @skip_if_iam
     def test_rewrite_rule(self):
         """
         Test that design document URL is rewritten to the expected test document.

tests/unit/unit_t_db_base.py

@@ -78,6 +78,13 @@ def wrapper(*args):
     return wrapper
 
 
+def skip_if_iam(f):
+    def wrapper(*args):
+        if os.environ.get('IAM_API_KEY'):
+            raise unittest.SkipTest('Test only supports non-IAM authentication')
+        return f(*args)
+    return wrapper
+
 class UnitTestDbBase(unittest.TestCase):
     """
     The base class for all unit tests targeting a database