From 56f5b475ebd057e3c29be009abbff8017d96e865 Mon Sep 17 00:00:00 2001 From: Zachary Yedidia Date: Tue, 25 Feb 2020 23:21:50 -0500 Subject: [PATCH] Improve buffer test --- internal/buffer/buffer_test.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/internal/buffer/buffer_test.go b/internal/buffer/buffer_test.go index e9e65fd5..d5a027b4 100644 --- a/internal/buffer/buffer_test.go +++ b/internal/buffer/buffer_test.go @@ -85,7 +85,9 @@ func check(t *testing.T, before []string, operations []operation, after []string checkText(after) - for b.UndoStack.Peek() != nil { + // must have exactly two events per operation (delete and insert) + for range operations { + b.UndoOneEvent() b.UndoOneEvent() } @@ -101,7 +103,8 @@ func check(t *testing.T, before []string, operations []operation, after []string } } - for b.RedoStack.Peek() != nil { + for range operations { + b.RedoOneEvent() b.RedoOneEvent() }