1 files changed, 12 insertions, 8 deletions
diff --git a/shared/tests/test_models.py b/shared/tests/test_models.py
index 04098ce..40bb791 100644
--- a/shared/tests/test_models.py
+++ b/shared/tests/test_models.py
@@ -1,8 +1,8 @@
 """Tests for shared models and settings."""
 
 import os
+from datetime import UTC, datetime
 from decimal import Decimal
-from datetime import datetime, timezone
 from unittest.mock import patch
 
 
@@ -12,8 +12,11 @@ def test_settings_defaults():
 
     with patch.dict(os.environ, {}, clear=False):
         settings = Settings()
-        assert settings.redis_url == "redis://localhost:6379"
-        assert settings.database_url == "postgresql://trading:trading@localhost:5432/trading"
+        assert settings.redis_url.get_secret_value() == "redis://localhost:6379"
+        assert (
+            settings.database_url.get_secret_value()
+            == "postgresql://trading:trading@localhost:5432/trading"
+        )
         assert settings.log_level == "INFO"
         assert settings.risk_max_position_size == 0.1
         assert settings.risk_stop_loss_pct == 5.0
@@ -25,7 +28,7 @@ def test_candle_creation():
     """Test Candle model creation."""
     from shared.models import Candle
 
-    now = datetime.now(timezone.utc)
+    now = datetime.now(UTC)
     candle = Candle(
         symbol="AAPL",
         timeframe="1m",
@@ -47,7 +50,7 @@ def test_candle_creation():
 
 def test_signal_creation():
     """Test Signal model creation."""
-    from shared.models import Signal, OrderSide
+    from shared.models import OrderSide, Signal
 
     signal = Signal(
         strategy="rsi_strategy",
@@ -69,9 +72,10 @@ def test_signal_creation():
 
 def test_order_creation():
     """Test Order model creation with defaults."""
-    from shared.models import Order, OrderSide, OrderType, OrderStatus
     import uuid
 
+    from shared.models import Order, OrderSide, OrderStatus, OrderType
+
     signal_id = str(uuid.uuid4())
     order = Order(
         signal_id=signal_id,
@@ -90,7 +94,7 @@ def test_order_creation():
 
 def test_signal_conviction_default():
     """Test Signal defaults for conviction, stop_loss, take_profit."""
-    from shared.models import Signal, OrderSide
+    from shared.models import OrderSide, Signal
 
     signal = Signal(
         strategy="rsi",
@@ -107,7 +111,7 @@ def test_signal_conviction_default():
 
 def test_signal_with_stops():
     """Test Signal with explicit conviction, stop_loss, take_profit."""
-    from shared.models import Signal, OrderSide
+    from shared.models import OrderSide, Signal
 
     signal = Signal(
         strategy="rsi",