import pytest

from freqtrade.persistence import FtNoDBContext, PairLocks, Trade


@pytest.mark.parametrize('timeframe', ['', '5m', '1d'])
def test_FtNoDBContext(timeframe):
    PairLocks.timeframe = ''
    assert Trade.use_db is True
    assert PairLocks.use_db is True
    assert PairLocks.timeframe == ''

    with FtNoDBContext(timeframe):
        assert Trade.use_db is False
        assert PairLocks.use_db is False
        assert PairLocks.timeframe == timeframe

    with FtNoDBContext():
        assert Trade.use_db is False
        assert PairLocks.use_db is False
        assert PairLocks.timeframe == ''

    assert Trade.use_db is True
    assert PairLocks.use_db is True