fixing test cases

int_tests/managers/test_api_keys.py

@@ -27,6 +27,8 @@ def setUpClass(cls):
 
         # Create corpus
         response = cls.client.corpora.create(name=cls.corpus_name, key=cls.corpus_key)
+        # Small wait for corpus to be fully provisioned
+        time.sleep(2)
         cls.key = response.key
 
     def _create_api_key(self, name="test-key", api_key_role="serving"):

int_tests/managers/test_document.py

@@ -25,6 +25,8 @@ def setUpClass(cls):
         timestamp = int(time.time())
         corpus_name = f"test-document-manager-{timestamp}"
         response = cls.client.corpora.create(name=corpus_name, key=corpus_name)
+        # Small wait for corpus to be fully provisioned
+        time.sleep(2)
         cls.corpus_key = response.key
         cls.created_corpora.add(cls.corpus_key)
 

int_tests/managers/test_upload.py

@@ -86,6 +86,8 @@ def setUpClass(cls):
         timestamp = int(time.time())
         corpus_name = f"test-upload-{timestamp}"
         response = cls.client.corpora.create(key=corpus_name, name=corpus_name)
+        # Small wait for corpus to be fully provisioned
+        time.sleep(2)
         cls.corpus = response
 
     def _get_test_file(self):

int_tests/test_chat.py

@@ -42,6 +42,8 @@ def setUpClass(cls):
         cls.corpus_name = f"test-chat-corpus-{timestamp}"
 
         cls.client.corpora.create(name=cls.corpus_name, key=cls.corpus_name)
+        # Small wait for corpus to be fully provisioned
+        time.sleep(2)
         cls.client.documents.create(cls.corpus_name, request=cls.TEST_DOCUMENT)
 
         # Create default search parameters

int_tests/test_query.py

@@ -56,6 +56,8 @@ def setUpClass(cls):
 
         for corpus_name, document in zip(cls.corpus_names, cls.test_documents):
             cls.client.corpora.create(name=corpus_name, key=corpus_name)
+            # Small wait for corpus to be fully provisioned
+            time.sleep(2)
             cls.client.documents.create(corpus_name, request=document)
 
     def setUp(self):