fix some things

Anant · Jan 25, 2024 · 4d2f5f4 · 4d2f5f4
1 parent 59b639f
commit 4d2f5f4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/airbyte-integrations/connectors/destination-astra/destination_astra/astra_client.py b/airbyte-integrations/connectors/destination-astra/destination_astra/astra_client.py
@@ -2,9 +2,9 @@
 
 import json
 from typing import Dict, List, Optional
-import urllib3
 
 import requests
+import urllib3
 
 
 class AstraClient:

diff --git a/airbyte-integrations/connectors/destination-astra/unit_tests/indexer_test.py b/airbyte-integrations/connectors/destination-astra/unit_tests/indexer_test.py
@@ -155,6 +155,7 @@ def test_astra_pre_sync():
 def test_astra_check(collection_name, describe_throws, reported_dimensions, check_succeeds, error_message):
     indexer = create_astra_indexer()
 
+    indexer.client.create_collection = MagicMock()
     indexer.client.find_collections = MagicMock()
     indexer.client.find_collections.return_value = [create_index_description(collection_name=collection_name, dimensions=reported_dimensions)]