add test for last item in queue

platinummonkey · Oct 29, 2021 · 13f3d7f · 13f3d7f
1 parent 01eba8e
commit 13f3d7f
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/limiter/lifo_blocking_test.go b/limiter/lifo_blocking_test.go
@@ -123,6 +123,14 @@ func TestLifoQueue_Remove(t *testing.T) {
 	}
 	asrt.Equal(uint64(0), q.len())
 	asrt.Equal(7, len(seenElements))
+
+	q = lifoQueue{}
+	ctx = context.WithValue(context.Background(), testLifoQueueContextKey(1), 1)
+	q.push(ctx)
+
+	// Remove very last item leaving queue empty
+	q.remove(1)
+	asrt.Equal(uint64(0), q.len())
 }
 
 func TestLifoBlockingListener(t *testing.T) {