introduce ViewLines functions

pkg/integration/tests/diff/diff.go

@@ -25,7 +25,10 @@ var Diff = NewIntegrationTest(NewIntegrationTestArgs{
 		input.SwitchToBranchesWindow()
 		assert.CurrentViewName("localBranches")
 
-		assert.SelectedLine(Contains("branch-a"))
+		assert.CurrentViewTopLines(
+			Contains("branch-a"),
+			Contains("branch-b"),
+		)
 		input.Press(keys.Universal.DiffingMenu)
 		input.Menu(Equals("Diffing"), Contains(`diff branch-a`))
 

pkg/integration/tests/diff/diff_and_apply_patch.go

@@ -24,8 +24,11 @@ var DiffAndApplyPatch = NewIntegrationTest(NewIntegrationTestArgs{
 	Run: func(shell *Shell, input *Input, assert *Assert, keys config.KeybindingConfig) {
 		input.SwitchToBranchesWindow()
 		assert.CurrentViewName("localBranches")
+		assert.CurrentViewLines(
+			Contains("branch-a"),
+			Contains("branch-b"),
+		)
 
-		assert.SelectedLine(Contains("branch-a"))
 		input.Press(keys.Universal.DiffingMenu)
 
 		input.Menu(Equals("Diffing"), Equals("diff branch-a"))

pkg/integration/tests/diff/diff_commits.go

@@ -22,7 +22,11 @@ var DiffCommits = NewIntegrationTest(NewIntegrationTestArgs{
 		input.SwitchToCommitsWindow()
 		assert.CurrentViewName("commits")
 
-		assert.SelectedLine(Contains("third commit"))
+		assert.CurrentViewLines(
+			Contains("third commit"),
+			Contains("second commit"),
+			Contains("first commit"),
+		)
 
 		input.Press(keys.Universal.DiffingMenu)
 		input.Menu(Equals("Diffing"), MatchesRegexp(`diff \w+`))
@@ -33,7 +37,6 @@ var DiffCommits = NewIntegrationTest(NewIntegrationTestArgs{
 
 		input.NextItem()
 		input.NextItem()
-
 		assert.SelectedLine(Contains("first commit"))
 
 		assert.MainViewContent(Contains("-second line\n-third line"))