diff options
Diffstat (limited to 'pkg/tasks/tasks_test.go')
| -rw-r--r-- | pkg/tasks/tasks_test.go | 11 |
1 files changed, 7 insertions, 4 deletions
diff --git a/pkg/tasks/tasks_test.go b/pkg/tasks/tasks_test.go index e42d13e0b..49eedad31 100644 --- a/pkg/tasks/tasks_test.go +++ b/pkg/tasks/tasks_test.go @@ -10,6 +10,7 @@ import ( "testing" "time" + "github.com/jesseduffield/gocui" "github.com/jesseduffield/lazygit/pkg/secureexec" "github.com/jesseduffield/lazygit/pkg/utils" ) @@ -31,7 +32,10 @@ func TestNewCmdTaskInstantStop(t *testing.T) { onEndOfInput, getOnEndOfInputCallCount := getCounter() onNewKey, getOnNewKeyCallCount := getCounter() onDone, getOnDoneCallCount := getCounter() - incBusyCount, decBusyCount, getBusyCount := getIncDecCounter(1) + task := &gocui.Task{} + newTask := func() *gocui.Task { + return task + } manager := NewViewBufferManager( utils.NewDummyLog(), @@ -40,8 +44,7 @@ func TestNewCmdTaskInstantStop(t *testing.T) { refreshView, onEndOfInput, onNewKey, - incBusyCount, - decBusyCount, + newTask, ) stop := make(chan struct{}) @@ -57,7 +60,7 @@ func TestNewCmdTaskInstantStop(t *testing.T) { fn := manager.NewCmdTask(start, "prefix\n", LinesToRead{20, -1}, onDone) - _ = fn(TaskOpts{Stop: stop, InitialContentLoaded: decBusyCount}) + _ = fn(TaskOpts{Stop: stop, InitialContentLoaded: func() { task.Done() }}) callCountExpectations := []struct { expected int |