Skip to content

Commit 732bd21

Browse files
aviauborntyping
authored andcommitted
Added clear_queue function to QueuedClient (closes #12)
1 parent 5037d19 commit 732bd21

2 files changed

Lines changed: 8 additions & 0 deletions

File tree

riemann_client/client.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,3 +102,6 @@ def flush(self):
102102
def send_event(self, event):
103103
self.queue.events.add().MergeFrom(event)
104104
return event
105+
106+
def clear_queue(self):
107+
self.queue = riemann_client.riemann_pb2.Msg()

riemann_client/tests/test_riemann_queued_client.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,3 +65,8 @@ def test_deciqueue_output(queued_client, large_queue):
6565
def test_deciqueue_flush(queued_client, large_queue):
6666
queued_client.flush()
6767
assert len(queued_client.queue.events) == 0
68+
69+
70+
def test_clear_queue(queued_client, using_simple_queue):
71+
queued_client.clear_queue()
72+
assert len(queued_client.queue.events) == 0

0 commit comments

Comments
 (0)