|
1 | 1 | from unittest import TestCase |
2 | 2 |
|
3 | | -from finterion_investing_algorithm_framework.market_service import \ |
4 | | - FinterionMarketService |
| 3 | +from finterion_investing_algorithm_framework.utils import \ |
| 4 | + convert_finterion_order_to_order |
5 | 5 |
|
6 | 6 |
|
7 | 7 | class TestOrderModels(TestCase): |
@@ -64,8 +64,18 @@ def test_order_model(self): |
64 | 64 | 'total': 3 |
65 | 65 | } |
66 | 66 |
|
67 | | - for order in response_data['items']: |
68 | | - order = FinterionMarketService( |
69 | | - None, initialize=False, market_credential_service=None |
70 | | - )\ |
71 | | - ._convert_order(order) |
| 67 | + for order_data in response_data['items']: |
| 68 | + order = convert_finterion_order_to_order(order_data) |
| 69 | + self.assertEqual(order.get_target_symbol(), order_data['target_symbol']) |
| 70 | + self.assertEqual(order.get_trading_symbol(), order_data['trading_symbol']) |
| 71 | + self.assertEqual(order.get_price(), order_data['price']) |
| 72 | + self.assertEqual(order.get_amount(), order_data['amount']) |
| 73 | + self.assertEqual(order.get_filled(), order_data['filled']) |
| 74 | + self.assertEqual(order.get_remaining(), order_data['remaining']) |
| 75 | + self.assertEqual(order.get_order_type(), order_data['order_type']) |
| 76 | + self.assertEqual( |
| 77 | + order.get_order_side().lower(), order_data['order_side'].lower() |
| 78 | + ) |
| 79 | + self.assertEqual( |
| 80 | + order.get_status().lower(), order_data['status'].lower() |
| 81 | + ) |
0 commit comments