fix bug

py-client/tests/test_zt_state.py

@@ -1,5 +1,6 @@
 import tempfile
 import unittest
+from unittest.mock import patch
 from pathlib import Path
 
 from libs.state import State
@@ -49,6 +50,32 @@ class ZTStateTests(unittest.TestCase):
         self.assertEqual(self.state.state['600000.SH']['base_qty'], 100)
         self.assertEqual(self.state.deals['sell']['is_arch'], 0)
 
+    def test_failed_initialization_leaves_original_database_empty(self):
+        invalid = self.position(100)
+        invalid.open_price = float('inf')
+        with self.assertRaises(ValueError):
+            sync_account_state(self.state, [invalid], [self.deal('old', 100)], initialize=True)
+        restarted = State(self.state.path)
+        self.assertEqual((restarted.state, restarted.deals), ({}, {}))
+        sync_account_state(restarted, [self.position(100)], [self.deal('old', 100)], initialize=True)
+        self.assertEqual(restarted.deals['old']['is_arch'], 1)
+
+    def test_initialization_cannot_overwrite_existing_database(self):
+        sync_account_state(self.state, [self.position(100)], [], initialize=True)
+        with self.assertRaises(ValueError):
+            sync_account_state(self.state, [], [], initialize=True)
+        self.assertEqual(State(self.state.path).state['600000.SH']['base_qty'], 100)
+
+    def test_no_new_deals_still_retries_failed_archiving(self):
+        sync_account_state(self.state, [self.position(100)], [], initialize=True)
+        sell = self.deal('sell', 100, flag=24)
+        with patch.object(self.state, 'archiving'):
+            with self.assertRaises(ValueError):
+                sync_account_state(self.state, [], [sell])
+        sync_account_state(self.state, [], [sell])
+        self.assertEqual(self.state.state, {})
+        self.assertEqual(self.state.deals['sell']['is_arch'], 1)
+
 
 if __name__ == '__main__':
     unittest.main()