"""期权盈亏结算:模拟盘本地扣费;实盘以交易所历史为准并冻结折 U.""" from __future__ import annotations import unittest from lib.options.options_pnl_settle_lib import ( fee_to_premium_ccy, settle_closed_pnl, settle_exchange_realized_pnl, ) from lib.options.options_stats_lib import ( apply_exchange_pnl_freeze, compute_options_stats_from_history, history_pnl_to_usdt, ) class FeeConvertTests(unittest.TestCase): def test_fee_same_ccy(self): self.assertAlmostEqual( fee_to_premium_ccy(0.001, "ETH", premium_ccy="ETH", index_px=3000), 0.001, places=8, ) class SettleClosedPnlTests(unittest.TestCase): def test_sim_net_times_close_index(self): s = settle_closed_pnl( premium_paid=0.01, premium_received=0.02, open_fee=0.0005, close_fee=0.0005, premium_ccy="ETH", index_px=3000, ) self.assertAlmostEqual(float(s["realized_pnl"]), 0.009, places=8) self.assertAlmostEqual(float(s["realized_pnl_usdt"]), 27.0, places=4) def test_budget_ignored_for_usdt(self): # budget 不得抬高开仓成本 s = settle_closed_pnl( premium_paid=0.01, premium_received=0.02, open_fee=0, close_fee=0, premium_ccy="ETH", index_px=3000, budget_usdt=26.0, ) self.assertAlmostEqual(float(s["realized_pnl_usdt"]), 30.0, places=4) def test_usdc_freeze_is_net(self): s = settle_closed_pnl( premium_paid=10, premium_received=12, open_fee=0.1, close_fee=0.1, premium_ccy="USDC", index_px=None, ) self.assertAlmostEqual(float(s["realized_pnl"]), 1.8, places=4) self.assertAlmostEqual(float(s["realized_pnl_usdt"]), 1.8, places=4) def test_exchange_realized_not_fee_again(self): s = settle_exchange_realized_pnl( realized_pnl=0.009, premium_ccy="ETH", index_px=3000, fee_hint=0.001, ) self.assertAlmostEqual(float(s["realized_pnl"]), 0.009, places=8) self.assertAlmostEqual(float(s["realized_pnl_usdt"]), 27.0, places=4) class SimPremiumPnlStatsTests(unittest.TestCase): def test_sim_uses_premium_net_not_bridge_cash(self): """卖回−预算会把权利金本金算进盈利;必须用(收回−已付)×平仓指数.""" import sqlite3 from lib.options.options_db import init_options_tables from lib.options.options_spot_bridge_lib import ensure_bridge_table from lib.options.options_stats_lib import sync_sim_cash_realized_pnl_usdt conn = sqlite3.connect(":memory:") conn.row_factory = sqlite3.Row init_options_tables(conn) ensure_bridge_table(conn) # 桥上现金差额 952.96,但权利金净盈亏仅 0.01 ETH × 2500 = 25U conn.execute( """ INSERT INTO options_spot_bridge( id, underlying, status, budget_usdt, usdt_recovered ) VALUES (1, 'ETH', 'closed', 1000.0, 1952.96) """ ) conn.execute( """ INSERT INTO options_trades( inst_id, underlying, opt_type, sheets, eth_amount, premium_paid, premium_received, realized_pnl, premium_ccy, status, bridge_id, budget_usdt, realized_pnl_usdt, close_index_px, created_at, closed_at ) VALUES ( 'ETH-USD-261009-2500-P', 'ETH', 'P', 1, 0.01, 0.01, 0.02, 0.01, 'ETH', 'closed', 1, 1000.0, 1947.05, 2500.0, '2026-10-06 10:00:00', '2026-10-08 14:00:00' ) """ ) conn.commit() n = sync_sim_cash_realized_pnl_usdt(conn, None) self.assertEqual(n, 1) row = conn.execute( "SELECT realized_pnl_usdt, realized_pnl FROM options_trades WHERE id=1" ).fetchone() # (0.02-0.01)*2500 = 25, 绝不是 1952.96-1000 self.assertAlmostEqual(float(row["realized_pnl_usdt"]), 25.0, places=2) self.assertAlmostEqual(float(row["realized_pnl"]), 0.01, places=8) conn.close() def test_sim_deducts_fees_from_premium_pnl(self): import sqlite3 from lib.options.options_db import init_options_tables from lib.options.options_stats_lib import sync_sim_cash_realized_pnl_usdt conn = sqlite3.connect(":memory:") conn.row_factory = sqlite3.Row init_options_tables(conn) conn.execute( """ INSERT INTO options_trades( inst_id, underlying, opt_type, sheets, eth_amount, premium_paid, premium_received, realized_pnl, premium_ccy, open_fee, close_fee, status, realized_pnl_usdt, close_index_px, created_at, closed_at ) VALUES ( 'BTC-USD-261008-86500-P', 'BTC', 'P', 1, 0.01, 0.01, 0.02, 0.01, 'BTC', 0.0001, 0.0001, 'closed', 1947.05, 80000.0, '2026-10-05 05:11:30', '2026-10-07 10:03:53' ) """ ) conn.commit() sync_sim_cash_realized_pnl_usdt(conn, None) row = conn.execute( "SELECT realized_pnl_usdt FROM options_trades WHERE id=1" ).fetchone() # (0.02-0.01-0.0002)*80000 = 784 self.assertAlmostEqual(float(row["realized_pnl_usdt"]), 784.0, places=2) conn.close() class ExchangeFreezeStatsTests(unittest.TestCase): def test_freeze_uses_exchange_coin_pnl(self): import sqlite3 from lib.options.options_db import init_options_tables conn = sqlite3.connect(":memory:") conn.row_factory = sqlite3.Row init_options_tables(conn) class _Ex: def public_get_market_ticker(self, params): inst = (params or {}).get("instId") or "" if inst.startswith("BTC"): return {"data": [{"last": "100000"}]} return {"data": [{"last": "2500"}]} history = [ { "status": "closed", "source": "exchange", "history_key": "ex|BTC-USD-261009-84000-P|1", "inst_id": "BTC-USD-261009-84000-P", "underlying": "BTC", "premium_ccy": "BTC", "realized_pnl": 0.000138, "created_at": "2026-10-08 10:00:00", "closed_at": "2026-10-08 22:22:56", }, { "status": "closed", "source": "exchange", "history_key": "ex|ETH-USD-261006-2700-P|2", "inst_id": "ETH-USD-261006-2700-P", "underlying": "ETH", "premium_ccy": "ETH", "realized_pnl": -0.00648, "created_at": "2026-10-05 10:00:00", "closed_at": "2026-10-06 16:00:27", }, ] frozen = apply_exchange_pnl_freeze(conn, history, _Ex()) converted = history_pnl_to_usdt(frozen, _Ex()) # 0.000138*100000=13.8; -0.00648*2500=-16.2; net=-2.4 self.assertAlmostEqual(float(converted[0]["realized_pnl"]), 13.8, places=4) self.assertAlmostEqual(float(converted[1]["realized_pnl"]), -16.2, places=4) stats = compute_options_stats_from_history(converted) self.assertAlmostEqual(float(stats["net_realized_pnl"]), -2.4, places=4) # 指数再变也不改冻结 U class _Ex2: def public_get_market_ticker(self, params): return {"data": [{"last": "999999"}]} frozen2 = apply_exchange_pnl_freeze(conn, history, _Ex2()) converted2 = history_pnl_to_usdt(frozen2, _Ex2()) self.assertAlmostEqual(float(converted2[0]["realized_pnl"]), 13.8, places=4) conn.close() def test_frozen_usdt_ignores_live_index(self): history = [ { "status": "closed", "inst_id": "ETH-USD-260701-2500-C", "underlying": "ETH", "premium_ccy": "ETH", "realized_pnl": 0.01, "realized_pnl_usdt": 25.0, "close_index_px": 2500.0, "created_at": "2026-10-01 10:00:00", "closed_at": "2026-10-01 12:00:00", } ] class _Ex: def public_get_market_ticker(self, params): return {"data": [{"last": "99999"}]} converted = history_pnl_to_usdt(history, _Ex()) self.assertAlmostEqual(float(converted[0]["realized_pnl"]), 25.0, places=4) if __name__ == "__main__": unittest.main()