import os
import sys
import logging

logging.basicConfig(level=logging.INFO)

from kis_trader.utils.env import set_db, invalidate_merged_env_cache
from database import TradeDB
from kis_trader.strategies.breakout import BreakoutStrategy

def test_sleep():
    db = TradeDB()
    set_db(db)
    invalidate_merged_env_cache()
    
    print("DB value of SCAN_REJECT_SLEEP_MIN:", db.get_merged_env_snapshot().get("SCAN_REJECT_SLEEP_MIN"))
    
    # Initialize a dummy strategy instance
    class DummyWS:
        def __init__(self):
            self.tick_recorder = None
            self.trigger_eval_recorder = None
            self.ws_cache = None
            self.candle_agg = None
        def get_candles(self, *args, **kwargs): return []
        def get_current_candle(self, *args, **kwargs): return None
        def fill_gap(self, *args, **kwargs): pass
        def set_candle_tick_codes(self, *args, **kwargs): pass
    
    ws = DummyWS()
    strategy = BreakoutStrategy(db, ws, ws)
    
    loop_min = strategy._scan_sleep("loop")
    reject_min = strategy._scan_sleep("reject")
    print(f"Loop sleep: {loop_min}")
    print(f"Reject sleep: {reject_min}")

if __name__ == "__main__":
    test_sleep()