feat: add trading service workflow

2026-07-08 17:25:37 +08:00
parent 0f4dfead6f
commit f282d4a38b
3 changed files with 366 additions and 0 deletions
--- a/tests/test_services.py
+++ b/tests/test_services.py
@@ -0,0 +1,131 @@
+from datetime import date, timedelta
+from decimal import Decimal
+
+import pytest
+
+from grid_trading.domain.models import Instrument, Trade, TradeGroup, TradeSide
+from grid_trading.services.trading_service import TradingService
+
+
+def test_service_creates_default_account_and_computes_summary(tmp_path):
+    service = TradingService(tmp_path / "grid.db")
+    service.ensure_defaults()
+    account = service.get_active_account()
+    assert account.name == "默认账户"
+
+    account = service.save_account(account.__class__(id=account.id, name="主账户", initial_cash=Decimal("100000")))
+    instrument = service.add_instrument(
+        Instrument(id=None, code="510300", name="沪深300ETF", market="ETF", manual_price=Decimal("4.00"))
+    )
+    service.save_trade(
+        Trade(
+            id=None,
+            account_id=account.id,
+            instrument_id=instrument.id,
+            trade_date=date(2026, 7, 7),
+            side=TradeSide.BUY,
+            price=Decimal("3.90"),
+            quantity=1000,
+            commission=Decimal("5"),
+            trade_group=TradeGroup.GRID,
+        )
+    )
+    service.save_trade(
+        Trade(
+            id=None,
+            account_id=account.id,
+            instrument_id=instrument.id,
+            trade_date=date(2026, 7, 8),
+            side=TradeSide.SELL,
+            price=Decimal("4.10"),
+            quantity=500,
+            commission=Decimal("5"),
+            stamp_tax=Decimal("1.03"),
+            trade_group=TradeGroup.GRID,
+        )
+    )
+
+    positions = service.get_position_summaries(as_of=date(2026, 7, 8))
+    summary = service.get_account_summary(as_of=date(2026, 7, 8))
+
+    assert len(positions) == 1
+    assert positions[0].total_quantity == 500
+    assert positions[0].grid_profit == Decimal("91.47")
+    assert summary.cash == Decimal("98138.97")
+    assert summary.market_value == Decimal("2000.00")
+    assert summary.total_assets == Decimal("100138.97")
+
+
+def test_service_validates_lot_size_and_available_sell_quantity(tmp_path):
+    service = TradingService(tmp_path / "grid.db")
+    service.ensure_defaults()
+    account = service.get_active_account()
+    instrument = service.add_instrument(Instrument(id=None, code="600000", name="浦发银行"))
+
+    with pytest.raises(ValueError, match="100"):
+        service.save_trade(
+            Trade(
+                id=None,
+                account_id=account.id,
+                instrument_id=instrument.id,
+                trade_date=date(2026, 7, 8),
+                side=TradeSide.BUY,
+                price=Decimal("10"),
+                quantity=50,
+                trade_group=TradeGroup.BASE,
+            )
+        )
+
+    service.save_trade(
+        Trade(
+            id=None,
+            account_id=account.id,
+            instrument_id=instrument.id,
+            trade_date=date(2026, 7, 8),
+            side=TradeSide.BUY,
+            price=Decimal("10"),
+            quantity=100,
+            trade_group=TradeGroup.BASE,
+        )
+    )
+
+    with pytest.raises(ValueError, match="T\\+1"):
+        service.save_trade(
+            Trade(
+                id=None,
+                account_id=account.id,
+                instrument_id=instrument.id,
+                trade_date=date(2026, 7, 8),
+                side=TradeSide.SELL,
+                price=Decimal("10.1"),
+                quantity=100,
+                trade_group=TradeGroup.BASE,
+            )
+        )
+
+    service.save_trade(
+        Trade(
+            id=None,
+            account_id=account.id,
+            instrument_id=instrument.id,
+            trade_date=date(2026, 7, 8) - timedelta(days=1),
+            side=TradeSide.BUY,
+            price=Decimal("9.9"),
+            quantity=100,
+            trade_group=TradeGroup.BASE,
+        )
+    )
+    saved_sell = service.save_trade(
+        Trade(
+            id=None,
+            account_id=account.id,
+            instrument_id=instrument.id,
+            trade_date=date(2026, 7, 8),
+            side=TradeSide.SELL,
+            price=Decimal("10.1"),
+            quantity=100,
+            trade_group=TradeGroup.BASE,
+        )
+    )
+
+    assert saved_sell.id is not None