Refine ClientTest

Author: sinall

tests/config/config.ini.template

@@ -2,6 +2,7 @@
 host=localhost
 port=8888
 key=
+client=
 
 [JoinQuant]
 username=

tests/shipane_sdk/test_client.py

@@ -17,55 +17,57 @@
 
 
 class ClientTest(unittest.TestCase):
-    def setUp(self):
+    @classmethod
+    def setUpClass(cls):
         logging.basicConfig(level=logging.DEBUG)
         config = ConfigParser()
         dir_path = os.path.dirname(os.path.realpath(__file__))
         config.read('{}/../config/config.ini'.format(dir_path))
-        self.client = Client(logging.getLogger(), host=config.get('ShiPanE', 'host'), key=config.get('ShiPanE', 'key'))
-        self.client.start_clients()
+        cls.client = Client(logging.getLogger(), **dict(config.items('ShiPanE')))
+        cls.client_param = config.get('ShiPanE', 'client')
+        cls.client.start_clients()
 
     def test_get_account(self):
         try:
-            self.client.get_account()
+            self.client.get_account(self.client_param)
         except HTTPError as e:
             self.fail()
 
     def test_get_positions(self):
         try:
-            data = self.client.get_positions()
+            data = self.client.get_positions(self.client_param)
             sub_accounts = data['sub_accounts']
-            self.assertGreater(sub_accounts['总资产']['人民币'], 0)
+            self.assertGreaterEqual(sub_accounts['余额']['人民币'], 0)
             positions = data['positions']
             self.assertIsNotNone(positions['证券代码'][0])
         except HTTPError as e:
             self.fail()
 
     def test_buy_stock(self):
         try:
-            order = self.client.buy(symbol='000001', price=8.11, amount=100)
+            order = self.client.buy(self.client_param, symbol='000001', price=8.11, amount=100)
             self.assertIsNotNone(order['id'])
         except HTTPError as e:
             result = e.response.json()
             self.assertIsNotNone(result['message'])
 
     def test_sell_stock(self):
         try:
-            order = self.client.sell(symbol='000001', price=9.51, amount=100)
+            order = self.client.sell(self.client_param, symbol='000001', price=9.51, amount=100)
             self.assertIsNotNone(order['id'])
         except HTTPError as e:
             result = e.response.json()
             self.assertIsNotNone(result['message'])
 
     def test_cancel_all(self):
         try:
-            self.client.cancel_all()
+            self.client.cancel_all(self.client_param)
         except HTTPError as e:
             self.fail()
 
     def test_query(self):
         try:
-            df = self.client.query(None, '查询>资金股份')
+            df = self.client.query(self.client_param, '查询>资金股份')
             self.assertIsNotNone(df['证券代码'][0])
         except HTTPError as e:
             self.fail()