from app.bars.models import Bar, Timeframe
from app.bars.store import InMemoryBarStore


def bar(t, close=1):
    return Bar(Timeframe.M1, t, 1, 2, 0, close, 10, False, "ES=F", "replay")


def test_store_replaces_forming_bar_and_bounds_history():
    store = InMemoryBarStore(2)
    store.put(bar(60))
    store.put(bar(60, 2))
    store.put(bar(120))
    store.put(bar(180))

    assert [value.t for value in store.get(Timeframe.M1)] == [120, 180]
    assert store.get(Timeframe.M1, 1)[0].t == 180