test: git rid of root folder

reata · Dec 31, 2023 · 0506489 · 0506489
1 parent 1fa5deb
commit 0506489
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tests/core/test_metadata_provider.py b/tests/core/test_metadata_provider.py
@@ -1,13 +1,18 @@
+import os
+
 import pytest
 
 from sqllineage.core.metadata.sqlalchemy import SQLAlchemyMetaDataProvider
 from sqllineage.exceptions import MetaDataProviderException
 
 
 def test_sqlalchemy_metadata_provider_connection_fail():
-    # connect to /root directory as sqlite db, which is not possible. Simulate connection failure
+    # connect to a directory as sqlite db, which is not possible. Simulate connection failure
     with pytest.raises(MetaDataProviderException):
-        SQLAlchemyMetaDataProvider("sqlite:////root/")
+        SQLAlchemyMetaDataProvider(f"sqlite:///{os.path.dirname(__file__)}")
+
+
+def test_sqlalchemy_metadata_provider_driver_not_install():
     # use an unknown driver to connect. Simulate driver not installed
     with pytest.raises(MetaDataProviderException):
         SQLAlchemyMetaDataProvider("sqlite+unknown_driver:///:memory:")