Add OKX env toggle to show or hide perpetual funds in the header.

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
dekun
2026-08-05 17:55:15 +08:00
parent 55e94fe059
commit 22e6b68e6d
11 changed files with 105 additions and 10 deletions
@@ -108,6 +108,10 @@ class TestEnvSchema(unittest.TestCase):
self.assertEqual(by_key["KEY_AUTO_ORDER_ENABLED"]["label"], "关键位自动单")
self.assertEqual(by_key["KEY_AUTO_ORDER_ENABLED"]["type"], "bool")
self.assertIn("KEY_AUTO_MIN_PLANNED_RR", by_key)
self.assertIn("OKX_SHOW_PERP_FUNDS", by_key)
self.assertEqual(by_key["OKX_SHOW_PERP_FUNDS"]["label"], "显示永续资金")
self.assertEqual(by_key["OKX_SHOW_PERP_FUNDS"]["type"], "bool")
self.assertTrue(by_key["OKX_SHOW_PERP_FUNDS"].get("hot_reload"))
groups_v = [{"title": "t", "fields": [by_key["TRADE_DIRECTION"]]}]
clean, errors = validate_env_updates(groups_v, {"TRADE_DIRECTION": "long_only"})
@@ -117,5 +121,22 @@ class TestEnvSchema(unittest.TestCase):
self.assertTrue(bad)
class TestShowPerpFunds(unittest.TestCase):
def test_okx_default_on(self):
from lib.instance.instance_embed_context_lib import show_perp_funds_enabled
old = os.environ.pop("OKX_SHOW_PERP_FUNDS", None)
try:
self.assertTrue(show_perp_funds_enabled(exchange_key="okx"))
os.environ["OKX_SHOW_PERP_FUNDS"] = "false"
self.assertFalse(show_perp_funds_enabled(exchange_key="okx"))
self.assertTrue(show_perp_funds_enabled(exchange_key="gate"))
finally:
if old is None:
os.environ.pop("OKX_SHOW_PERP_FUNDS", None)
else:
os.environ["OKX_SHOW_PERP_FUNDS"] = old
if __name__ == "__main__":
unittest.main()