def test_filter_limit(): trace = TraceStore(10) for i in range(5): trace.append(i) result = [x[1] for x in trace.filter(limit=3)] assert result == [2, 3, 4]
def test_filter(): trace = TraceStore(10) for i in range(5): trace.append(i) result = [x[1] for x in trace.filter()] assert result == [0, 1, 2, 3, 4]
def test_filter_category_limit(): cycle = itertools.cycle(range(3)) trace = TraceStore(10) for i in range(5): trace.append(1, str(next(cycle))) result = [x[2] for x in trace.filter(category="0", limit=1)] assert result == ["0"]
def test_filter_to_limit(): cycle = itertools.cycle(range(3)) trace = TraceStore(10) for i in range(5): msg = Message(body=str(i), to="{}@server".format(next(cycle)), sender="sender@server") trace.append(msg) result = [x[1].body for x in trace.filter(to="0@server", limit=1)] assert result == ["3"]
def test_filter_to_and_category_limit(): cycle = itertools.cycle(range(3)) trace = TraceStore(10) for i in range(5): c = str(next(cycle)) msg = Message(body=str(c), to="{}@server".format(c), sender="sender@server") trace.append(msg, c) result = [(x[1].body, x[2]) for x in trace.filter(to="1@server", category="1", limit=1)] assert result == [("1", "1")]