More accurate line count

zix99 · Oct 21, 2020 · 6f978d3 · 6f978d3
1 parent d686c04
commit 6f978d3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/multiterm/virtualterm.go b/pkg/multiterm/virtualterm.go
@@ -33,7 +33,7 @@ func (s *VirtualTerm) Get(line int) string {
 }
 
 func (s *VirtualTerm) LineCount() int {
-	return s.maxLine
+	return s.maxLine + 1
 }
 
 func (s *VirtualTerm) WriteToOutput(out io.Writer) {

diff --git a/pkg/multiterm/virtualterm_test.go b/pkg/multiterm/virtualterm_test.go
@@ -14,7 +14,7 @@ func TestVirtualTerm(t *testing.T) {
 
 	assert.Equal(t, "Hello", vt.Get(0))
 	assert.Equal(t, "", vt.Get(1))
-	assert.Equal(t, 2, vt.LineCount())
+	assert.Equal(t, 3, vt.LineCount())
 
 	vt.WriteToOutput(os.Stdout)
 }