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")]
