from db.redis_manager_delegate import RedisUtils
from trade.trade_settings import TradeStateManager, WantBuyCodesManager


def test_trade_state():
    print(TradeStateManager().is_can_buy(), TradeStateManager().is_can_buy_cache())
    TradeStateManager().open_buy()
    print(TradeStateManager().is_can_buy(), TradeStateManager().is_can_buy_cache())
    TradeStateManager().close_buy()
    print(TradeStateManager().is_can_buy_cache(), TradeStateManager().is_can_buy_cache())


def test_want_buy():
    code, buy_type = "000333", 1
    print(WantBuyCodesManager().add_code(code, buy_type))
    print(WantBuyCodesManager().list_code())
    print(WantBuyCodesManager().list_code_cache())
    print(WantBuyCodesManager().is_in(code, buy_type))
    print(WantBuyCodesManager().is_in_cache(code, buy_type))

    print(WantBuyCodesManager().remove_code(code, buy_type))
    print(WantBuyCodesManager().list_code())
    print(WantBuyCodesManager().list_code_cache())


if __name__ == "__main__":
    WantBuyCodesManager().is_in_cache("123230",1)
    # test_want_buy()
    RedisUtils.run_loop()