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:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user