11import pytest
22
33from test .integration .connectors .utils .constants import DESTINATION_TAG , VECTOR_DB_TAG
4- from unstructured_ingest .error import ValueError
4+ from unstructured_ingest .error import DestinationConnectionError
5+ from unstructured_ingest .error import ValueError as IngestValueError
56from unstructured_ingest .processes .connectors .weaviate .cloud import (
67 CONNECTOR_TYPE ,
78 CloudWeaviateAccessConfig ,
89 CloudWeaviateConnectionConfig ,
10+ CloudWeaviateUploader ,
11+ CloudWeaviateUploaderConfig ,
912)
1013
1114
1215@pytest .mark .tags (CONNECTOR_TYPE , DESTINATION_TAG , VECTOR_DB_TAG )
1316def test_weaviate_failing_connection_config ():
14- with pytest .raises (ValueError ):
17+ with pytest .raises (IngestValueError ):
1518 CloudWeaviateConnectionConfig (
1619 access_config = CloudWeaviateAccessConfig (api_key = "my key" , password = "password" ),
1720 username = "username" ,
@@ -37,3 +40,20 @@ def test_weaviate_connection_config_anonymous():
3740 anonymous = True ,
3841 cluster_url = "clusterurl" ,
3942 )
43+
44+
45+ @pytest .mark .tags (CONNECTOR_TYPE , DESTINATION_TAG , VECTOR_DB_TAG )
46+ def test_weaviate_precheck_invalid_credentials ():
47+ """Test that precheck properly validates connection with invalid credentials."""
48+ connection_config = CloudWeaviateConnectionConfig (
49+ access_config = CloudWeaviateAccessConfig (api_key = "invalid-test-key-12345" ),
50+ cluster_url = "https://invalid-test-cluster.weaviate.cloud" ,
51+ anonymous = False ,
52+ )
53+ upload_config = CloudWeaviateUploaderConfig (collection = None )
54+ uploader = CloudWeaviateUploader (
55+ connection_config = connection_config ,
56+ upload_config = upload_config ,
57+ )
58+ with pytest .raises (DestinationConnectionError ):
59+ uploader .precheck ()
0 commit comments