allow postgres db

2025-07-16 10:08:07 +02:00
parent 4e3ef5d207
commit 22b06f775d
8 changed files with 476 additions and 794 deletions
@@ -0,0 +1,152 @@
+#!/usr/bin/env python3
+"""
+Test script to verify database configuration for both SQLite and PostgreSQL
+"""
+
+import os
+import sys
+import time
+import json
+
+# Add the app directory to the Python path
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'app'))
+
+def test_sqlite():
+    """Test SQLite database functionality"""
+    print("=== Testing SQLite Database ===")
+
+    # Clear any existing PostgreSQL environment variables
+    for var in ['POSTGRES_HOST', 'POSTGRES_DB', 'POSTGRES_USER', 'POSTGRES_PASSWORD']:
+        if var in os.environ:
+            del os.environ[var]
+
+    try:
+        from app.database import db_manager, init_db, get_cached_data, save_to_cache, cleanup_old_cache_entries
+
+        print(f"Database type: {db_manager.db_type}")
+
+        # Initialize database
+        init_db()
+        print("✓ Database initialized successfully")
+
+        # Test cache operations
+        test_url = "https://example.com"
+        test_route = "test"
+        test_data = {"title": "Test Page", "content": "Test content"}
+
+        # Save to cache
+        save_to_cache(test_url, test_route, test_data)
+        print("✓ Data saved to cache")
+
+        # Retrieve from cache
+        cached_data = get_cached_data(test_url, test_route)
+        if cached_data and cached_data == test_data:
+            print("✓ Data retrieved from cache successfully")
+        else:
+            print("✗ Failed to retrieve data from cache")
+            return False
+
+        # Test cache stats
+        from app.services.cache import get_cache_stats
+        stats = get_cache_stats()
+        if stats['status'] == 'success':
+            print("✓ Cache stats retrieved successfully")
+            print(f"  Database type: {stats['stats']['database_type']}")
+            print(f"  Total entries: {stats['stats']['total_entries']}")
+        else:
+            print("✗ Failed to get cache stats")
+            return False
+
+        print("✓ SQLite database test passed!")
+        return True
+
+    except Exception as e:
+        print(f"✗ SQLite test failed: {e}")
+        return False
+
+def test_postgres():
+    """Test PostgreSQL database functionality"""
+    print("\n=== Testing PostgreSQL Database ===")
+
+    # Check if PostgreSQL credentials are available
+    required_vars = ['POSTGRES_HOST', 'POSTGRES_DB', 'POSTGRES_USER', 'POSTGRES_PASSWORD']
+    missing_vars = [var for var in required_vars if not os.getenv(var)]
+
+    if missing_vars:
+        print(f"⚠ PostgreSQL test skipped - missing environment variables: {', '.join(missing_vars)}")
+        print("Set these variables to test PostgreSQL:")
+        print("  POSTGRES_HOST=your-host")
+        print("  POSTGRES_DB=your-database")
+        print("  POSTGRES_USER=your-username")
+        print("  POSTGRES_PASSWORD=your-password")
+        return True  # Not a failure, just skipped
+
+    try:
+        from app.database import db_manager, init_db, get_cached_data, save_to_cache
+
+        print(f"Database type: {db_manager.db_type}")
+
+        # Initialize database
+        init_db()
+        print("✓ Database initialized successfully")
+
+        # Test cache operations
+        test_url = "https://example.com"
+        test_route = "test"
+        test_data = {"title": "Test Page", "content": "Test content"}
+
+        # Save to cache
+        save_to_cache(test_url, test_route, test_data)
+        print("✓ Data saved to cache")
+
+        # Retrieve from cache
+        cached_data = get_cached_data(test_url, test_route)
+        if cached_data and cached_data == test_data:
+            print("✓ Data retrieved from cache successfully")
+        else:
+            print("✗ Failed to retrieve data from cache")
+            return False
+
+        # Test cache stats
+        from app.services.cache import get_cache_stats
+        stats = get_cache_stats()
+        if stats['status'] == 'success':
+            print("✓ Cache stats retrieved successfully")
+            print(f"  Database type: {stats['stats']['database_type']}")
+            print(f"  Total entries: {stats['stats']['total_entries']}")
+        else:
+            print("✗ Failed to get cache stats")
+            return False
+
+        print("✓ PostgreSQL database test passed!")
+        return True
+
+    except Exception as e:
+        print(f"✗ PostgreSQL test failed: {e}")
+        return False
+
+def main():
+    """Run all database tests"""
+    print("Database Configuration Test")
+    print("=" * 50)
+
+    # Test SQLite
+    sqlite_success = test_sqlite()
+
+    # Test PostgreSQL
+    postgres_success = test_postgres()
+
+    print("\n" + "=" * 50)
+    print("Test Results:")
+    print(f"SQLite: {'✓ PASSED' if sqlite_success else '✗ FAILED'}")
+    print(f"PostgreSQL: {'✓ PASSED' if postgres_success else '✗ FAILED'}")
+
+    if sqlite_success and postgres_success:
+        print("\n🎉 All tests passed! Database configuration is working correctly.")
+        return 0
+    else:
+        print("\n❌ Some tests failed. Please check the configuration.")
+        return 1
+
+if __name__ == "__main__":
+    sys.exit(main())