QuantPlatformKit/tests/test_longbridge_execution.py at d37df79b81760cb75ad184dd35e1d8575afcf329 · QuantStrategyLab/QuantPlatformKit · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
from __future__ import annotations

import sys
import types
import unittest
from unittest.mock import patch

from quant_platform_kit.longbridge.execution import estimate_max_purchase_quantity, fetch_order_status, submit_order


class FakeEstimateResponse:
    def __init__(self, cash_max_qty):
        self.cash_max_qty = cash_max_qty


class FakeOrdersResponse:
    def __init__(self, orders):
        self.orders = orders


class FakeTradeContext:
    def estimate_max_purchase_quantity(self, **kwargs):
        self.estimate_kwargs = kwargs
        return FakeEstimateResponse("12")

    def submit_order(self, symbol, order_type, side, quantity, tif, **kwargs):
        self.submit_args = (symbol, order_type, side, quantity, tif, kwargs)
        return type("SubmitResponse", (), {"order_id": "OID-1"})()

    def today_orders(self, order_id):
        order = type(
            "Order",
            (),
            {
                "status": "Filled",
                "executed_quantity": "10",
                "executed_price": "101.5",
                "msg": "",
            },
        )()
        return FakeOrdersResponse([order])


class LongBridgeExecutionTests(unittest.TestCase):
    def test_estimate_max_purchase_quantity(self) -> None:
        longport_module = types.ModuleType("longport")
        openapi_module = types.ModuleType("longport.openapi")
        openapi_module.OrderSide = types.SimpleNamespace(Buy="Buy")
        openapi_module.OrderType = types.SimpleNamespace(LO="LO", MO="MO")

        ctx = FakeTradeContext()
        with patch.dict(sys.modules, {"longport": longport_module, "longport.openapi": openapi_module}):
            quantity = estimate_max_purchase_quantity(ctx, "SOXL.US", order_kind="limit", ref_price=100.5)

        self.assertEqual(quantity, 12)
        self.assertIs(ctx.estimate_kwargs["fractional_shares"], False)

    def test_estimate_max_purchase_quantity_can_request_fractional_buying_power(self) -> None:
        longport_module = types.ModuleType("longport")
        openapi_module = types.ModuleType("longport.openapi")
        openapi_module.OrderSide = types.SimpleNamespace(Buy="Buy")
        openapi_module.OrderType = types.SimpleNamespace(LO="LO", MO="MO")

        ctx = FakeTradeContext()
        with patch.dict(sys.modules, {"longport": longport_module, "longport.openapi": openapi_module}):
            quantity = estimate_max_purchase_quantity(
                ctx,
                "SOXX.US",
                order_kind="limit",
                ref_price=495.91,
                fractional_shares=True,
            )

        self.assertEqual(quantity, 12)
        self.assertIs(ctx.estimate_kwargs["fractional_shares"], True)

    def test_submit_order(self) -> None:
        longport_module = types.ModuleType("longport")
        openapi_module = types.ModuleType("longport.openapi")
        openapi_module.OrderSide = types.SimpleNamespace(Buy="Buy", Sell="Sell")
        openapi_module.OrderType = types.SimpleNamespace(LO="LO", MO="MO")
        openapi_module.TimeInForceType = types.SimpleNamespace(Day="Day")

        ctx = FakeTradeContext()
        with patch.dict(sys.modules, {"longport": longport_module, "longport.openapi": openapi_module}):
            report = submit_order(ctx, "SOXL.US", order_kind="limit", side="buy", quantity=5, submitted_price=100.25)

        self.assertEqual(report.status, "submitted")
        self.assertEqual(report.broker_order_id, "OID-1")

    def test_submit_order_allows_decimal_quantity_at_or_above_one_share(self) -> None:
        longport_module = types.ModuleType("longport")
        openapi_module = types.ModuleType("longport.openapi")
        openapi_module.OrderSide = types.SimpleNamespace(Buy="Buy", Sell="Sell")
        openapi_module.OrderType = types.SimpleNamespace(LO="LO", MO="MO")
        openapi_module.TimeInForceType = types.SimpleNamespace(Day="Day")

        ctx = FakeTradeContext()
        with patch.dict(sys.modules, {"longport": longport_module, "longport.openapi": openapi_module}):
            report = submit_order(ctx, "SOXL.US", order_kind="limit", side="buy", quantity=1.5, submitted_price=100.25)

        self.assertEqual(report.status, "submitted")
        self.assertEqual(str(ctx.submit_args[3]), "1.5")

    def test_submit_order_rejects_quantity_below_one_before_api_call(self) -> None:
        longport_module = types.ModuleType("longport")
        openapi_module = types.ModuleType("longport.openapi")
        openapi_module.OrderSide = types.SimpleNamespace(Buy="Buy", Sell="Sell")
        openapi_module.OrderType = types.SimpleNamespace(LO="LO", MO="MO")
        openapi_module.TimeInForceType = types.SimpleNamespace(Day="Day")

        ctx = FakeTradeContext()
        with patch.dict(sys.modules, {"longport": longport_module, "longport.openapi": openapi_module}):
            report = submit_order(
                ctx,
                "SOXX.US",
                order_kind="limit",
                side="buy",
                quantity=0.4326,
                submitted_price=495.91,
            )

        self.assertEqual(report.status, "rejected")
        self.assertIn("at least 1 share", report.raw_payload["detail"])
        self.assertFalse(hasattr(ctx, "submit_args"))

    def test_fetch_order_status(self) -> None:
        status = fetch_order_status(FakeTradeContext(), "OID-1")

        self.assertEqual(status["status"], "Filled")
        self.assertEqual(status["executed_qty"], "10")


if __name__ == "__main__":
    unittest.main()