airbyte-lib: Clean up test schema in Snowflake (#35015)

airbyte-lib/tests/conftest.py

@@ -19,6 +19,7 @@
 from _pytest.nodes import Item
 from google.cloud import secretmanager
 from pytest_docker.plugin import get_docker_ip
+from sqlalchemy import create_engine
 
 from airbyte_lib.caches import PostgresCacheConfig
 
@@ -193,6 +194,10 @@ def snowflake_config():
 
     yield config
 
+    engine = create_engine(config.get_sql_alchemy_url())
+    with engine.begin() as connection:
+        connection.execute(f"DROP SCHEMA IF EXISTS {config.schema_name}")
+
 
 @pytest.fixture(autouse=True)
 def source_test_registry(monkeypatch):

airbyte-lib/tests/integration_tests/test_source_test_fixture.py

@@ -680,9 +680,6 @@ def test_sync_to_snowflake(snowflake_config: SnowflakeCacheConfig, expected_test
 
     cache = SnowflakeSQLCache(config=snowflake_config)
 
-    with cache.get_sql_connection() as con:
-        con.execute("DROP SCHEMA IF EXISTS AIRBYTE_RAW")
-
     result: ReadResult = source.read(cache)
 
     assert result.processed_records == 3
@@ -693,6 +690,7 @@ def test_sync_to_snowflake(snowflake_config: SnowflakeCacheConfig, expected_test
             check_dtype=False,
         )
 
+
 def test_sync_limited_streams(expected_test_stream_data):
     source = ab.get_source("source-test", config={"apiKey": "test"})
     cache = ab.new_local_cache()