1 files changed, 57 insertions, 0 deletions
diff --git a/services/portfolio-manager/tests/test_portfolio.py b/services/portfolio-manager/tests/test_portfolio.py
new file mode 100644
index 0000000..26319ca
--- /dev/null
+++ b/services/portfolio-manager/tests/test_portfolio.py
@@ -0,0 +1,57 @@
+"""Tests for PortfolioTracker."""
+from decimal import Decimal
+
+from shared.models import Order, OrderSide, OrderStatus, OrderType
+from portfolio_manager.portfolio import PortfolioTracker
+
+
+def make_order(side: OrderSide, price: str, quantity: str) -> Order:
+    """Helper to create a filled Order."""
+    return Order(
+        signal_id="test-signal",
+        symbol="BTC/USDT",
+        side=side,
+        type=OrderType.MARKET,
+        price=Decimal(price),
+        quantity=Decimal(quantity),
+        status=OrderStatus.FILLED,
+    )
+
+
+def test_portfolio_add_buy_order() -> None:
+    tracker = PortfolioTracker()
+    order = make_order(OrderSide.BUY, "50000", "0.1")
+    tracker.apply_order(order)
+
+    position = tracker.get_position("BTC/USDT")
+    assert position is not None
+    assert position.quantity == Decimal("0.1")
+    assert position.avg_entry_price == Decimal("50000")
+
+
+def test_portfolio_add_multiple_buys() -> None:
+    tracker = PortfolioTracker()
+    tracker.apply_order(make_order(OrderSide.BUY, "50000", "0.1"))
+    tracker.apply_order(make_order(OrderSide.BUY, "52000", "0.1"))
+
+    position = tracker.get_position("BTC/USDT")
+    assert position is not None
+    assert position.quantity == Decimal("0.2")
+    assert position.avg_entry_price == Decimal("51000")
+
+
+def test_portfolio_sell_reduces_position() -> None:
+    tracker = PortfolioTracker()
+    tracker.apply_order(make_order(OrderSide.BUY, "50000", "0.2"))
+    tracker.apply_order(make_order(OrderSide.SELL, "55000", "0.1"))
+
+    position = tracker.get_position("BTC/USDT")
+    assert position is not None
+    assert position.quantity == Decimal("0.1")
+    assert position.avg_entry_price == Decimal("50000")
+
+
+def test_portfolio_no_position_returns_none() -> None:
+    tracker = PortfolioTracker()
+    position = tracker.get_position("ETH/USDT")
+    assert position is None