Skip to content

Commit 0dbce8a

Browse files
committed
test: Add integration tests for loaders and streaming features
New tests: - test_checkpoint_resume.py - End-to-end resume scenarios - test_resilient_streaming.py - Resilience with real databases - Enhanced Snowflake loader tests with state management - Enhanced PostgreSQL tests with reorg handling - Updated Redis, DeltaLake, Iceberg, LMDB loader tests Integration test features: - Real database containers (PostgreSQL, Redis, Snowflake) - State persistence and resume testing - Label joining with actual data - Reorg detection and invalidation - Parallel loading with multiple workers - Error injection and recovery Tests require Docker for database containers.
1 parent b8ffea3 commit 0dbce8a

File tree

9 files changed

+2159
-530
lines changed

9 files changed

+2159
-530
lines changed

tests/conftest.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,12 +81,14 @@ def snowflake_config():
8181
'warehouse': os.getenv('SNOWFLAKE_WAREHOUSE', 'test_warehouse'),
8282
'database': os.getenv('SNOWFLAKE_DATABASE', 'test_database'),
8383
'schema': os.getenv('SNOWFLAKE_SCHEMA', 'PUBLIC'),
84-
'use_stage': True,
84+
'loading_method': 'stage', # Default to stage loading for existing tests
8585
}
8686

8787
# Add optional parameters if they exist
8888
if os.getenv('SNOWFLAKE_PASSWORD'):
8989
config['password'] = os.getenv('SNOWFLAKE_PASSWORD')
90+
if os.getenv('SNOWFLAKE_PRIVATE_KEY'):
91+
config['private_key'] = os.getenv('SNOWFLAKE_PRIVATE_KEY')
9092
if os.getenv('SNOWFLAKE_ROLE'):
9193
config['role'] = os.getenv('SNOWFLAKE_ROLE')
9294
if os.getenv('SNOWFLAKE_AUTHENTICATOR'):

0 commit comments

Comments
 (0)