dekun
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
"""半腿失败:手动补开 vs 自动平."""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from lib.hedge_plan.hedge_plan_orders_lib import (
|
|
manual_complete_on_partial,
|
|
partial_auto_close_enabled,
|
|
)
|
|
|
|
|
|
class PartialManualTests(unittest.TestCase):
|
|
def test_manual_default_forces_auto_close_off(self):
|
|
with mock.patch.dict(os.environ, {}, clear=False):
|
|
os.environ.pop("HEDGE_PLAN_MANUAL_COMPLETE_ON_PARTIAL", None)
|
|
os.environ["HEDGE_PLAN_PARTIAL_AUTO_CLOSE_OPTION"] = "true"
|
|
self.assertTrue(manual_complete_on_partial())
|
|
self.assertFalse(partial_auto_close_enabled())
|
|
|
|
def test_manual_off_allows_auto_close(self):
|
|
with mock.patch.dict(
|
|
os.environ,
|
|
{
|
|
"HEDGE_PLAN_MANUAL_COMPLETE_ON_PARTIAL": "false",
|
|
"HEDGE_PLAN_PARTIAL_AUTO_CLOSE_OPTION": "true",
|
|
},
|
|
):
|
|
self.assertFalse(manual_complete_on_partial())
|
|
self.assertTrue(partial_auto_close_enabled())
|
|
|
|
def test_both_off(self):
|
|
with mock.patch.dict(
|
|
os.environ,
|
|
{
|
|
"HEDGE_PLAN_MANUAL_COMPLETE_ON_PARTIAL": "false",
|
|
"HEDGE_PLAN_PARTIAL_AUTO_CLOSE_OPTION": "false",
|
|
},
|
|
):
|
|
self.assertFalse(partial_auto_close_enabled())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|