1 files changed, 7 insertions, 46 deletions
diff --git a/shared/tests/test_sa_models.py b/shared/tests/test_sa_models.py
index 67c3c82..c9311dd 100644
--- a/shared/tests/test_sa_models.py
+++ b/shared/tests/test_sa_models.py
@@ -14,6 +14,10 @@ def test_base_metadata_has_all_tables():
         "trades",
         "positions",
         "portfolio_snapshots",
+        "news_items",
+        "symbol_scores",
+        "market_sentiment",
+        "stock_selections",
     }
     assert expected == table_names
 
@@ -68,6 +72,9 @@ class TestSignalRow:
             "price",
             "quantity",
             "reason",
+            "conviction",
+            "stop_loss",
+            "take_profit",
             "created_at",
         }
         assert expected == cols
@@ -120,44 +127,6 @@ class TestOrderRow:
         assert fk_cols == {"signal_id": "signals.id"}
 
 
-class TestTradeRow:
-    def test_table_name(self):
-        from shared.sa_models import TradeRow
-
-        assert TradeRow.__tablename__ == "trades"
-
-    def test_columns(self):
-        from shared.sa_models import TradeRow
-
-        mapper = inspect(TradeRow)
-        cols = {c.key for c in mapper.column_attrs}
-        expected = {
-            "id",
-            "order_id",
-            "symbol",
-            "side",
-            "price",
-            "quantity",
-            "fee",
-            "traded_at",
-        }
-        assert expected == cols
-
-    def test_primary_key(self):
-        from shared.sa_models import TradeRow
-
-        mapper = inspect(TradeRow)
-        pk_cols = [c.name for c in mapper.mapper.primary_key]
-        assert pk_cols == ["id"]
-
-    def test_order_id_foreign_key(self):
-        from shared.sa_models import TradeRow
-
-        table = TradeRow.__table__
-        fk_cols = {fk.parent.name: fk.target_fullname for fk in table.foreign_keys}
-        assert fk_cols == {"order_id": "orders.id"}
-
-
 class TestPositionRow:
     def test_table_name(self):
         from shared.sa_models import PositionRow
@@ -229,11 +198,3 @@ class TestStatusDefault:
         status_col = table.c.status
         assert status_col.server_default is not None
         assert status_col.server_default.arg == "PENDING"
-
-    def test_trade_fee_server_default(self):
-        from shared.sa_models import TradeRow
-
-        table = TradeRow.__table__
-        fee_col = table.c.fee
-        assert fee_col.server_default is not None
-        assert fee_col.server_default.arg == "0"