optz print.

py-client/tests/test_deal_model.py

@@ -7,7 +7,7 @@ from pathlib import Path
 from types import SimpleNamespace
 from unittest.mock import Mock
 from libs.order import OrderBook as ActiveOrders
-from libs.orderbook import OrderBook
+from libs.state import State
 from sdk.models import Assets, DealItem, OrderItem, PositionItem
 from sdk.portfolio import PortfolioMixin
 
@@ -76,16 +76,21 @@ class ApiModelTests(unittest.TestCase):
         deal = self.client.deals()[0]
         with tempfile.TemporaryDirectory() as tmp:
             path = Path(tmp) / 'state.db'
-            book = OrderBook(path)
+            book = State(path)
             book.sync_deals([deal])
-            loaded = OrderBook(path).deals['sys1']
+            loaded = State(path).deals['sys1']
             self.assertEqual(loaded['volume'], deal.volume)
             self.assertEqual(loaded['trade_date'], '2026-09-07')
-        deal.trade_amount = 0
-        self.assertEqual(OrderBook.deal_record(deal)['trade_amount'], 1000)
-        deal.price = 0
-        with self.assertRaises(ValueError):
-            OrderBook.deal_record(deal)
+            deal.order_sys_id = 'sys2'
+            deal.trade_amount = 0
+            book.sync_deals([deal, deal])
+            self.assertEqual(book.deals['sys2']['trade_amount'], 1000)
+            self.assertEqual(deal.trade_amount, 0)
+            deal.order_sys_id = 'sys3'
+            deal.price = 0
+            with self.assertRaises(ValueError):
+                book.sync_deals([deal])
+            self.assertEqual(set(State(path).deals), {'sys1', 'sys2'})
 
 
 if __name__ == '__main__':