add zt_open_hands

py-client/tests/test_orderbook.py

@@ -4,10 +4,9 @@ import unittest
 from contextlib import closing
 from dataclasses import asdict, fields
 from pathlib import Path
-from datetime import datetime
 from unittest.mock import patch
 
-from libs.state import State, StateItem
+from libs.state import FLAG_BUY, FLAG_SELL, State, StateItem
 from sdk import DealItem, PositionItem
 
 
@@ -21,7 +20,7 @@ class OrderBookTests(unittest.TestCase):
         prefix = {'base': 'zt-base-', 'sell': 'zt-t-sell-', 'buy': 'zt-t-buy-'}[kind]
         return DealItem(
             order_sys_id=sys_order_id, stock_code='600000.SH',
-            offset_flag=24 if kind == 'sell' else 23,
+            offset_flag=FLAG_SELL if kind == 'sell' else FLAG_BUY,
             volume=qty, price=price, trade_amount=qty * price,
             trade_date=date, trade_time='10:00:00', remark=prefix + 'order1|zt',
         )
@@ -64,7 +63,7 @@ class OrderBookTests(unittest.TestCase):
         self.assertEqual(book.deals, {})
         self.assertEqual(book.deals_sys_ids, set())
         self.assertEqual(State(self.path).deals, {})
-        invalid.offset_flag = 23
+        invalid.offset_flag = FLAG_BUY
         book.sync_deals([first, invalid])
         self.assertEqual(book.deals_sys_ids, {'d1', 'd2'})