cleaner test assertions

1 files changed, 7 insertions, 8 deletions

diff --git a/pkg/integration/tests/branch/delete.go b/pkg/integration/tests/branch/delete.go
index 55e9b6aa5..a873ebaef 100644
--- a/pkg/integration/tests/branch/delete.go
+++ b/pkg/integration/tests/branch/delete.go
@@ -18,9 +18,8 @@ var Delete = NewIntegrationTest(NewIntegrationTestArgs{
 	},
 	Run: func(shell *Shell, input *Input, assert *Assert, keys config.KeybindingConfig) {
 		input.SwitchToBranchesWindow()
-		assert.CurrentViewName("localBranches")
 
-		assert.CurrentViewLines(
+		assert.CurrentView().Name("localBranches").Lines(
 			MatchesRegexp(`\*.*branch-two`),
 			MatchesRegexp(`branch-one`),
 			MatchesRegexp(`master`),
@@ -34,11 +33,11 @@ var Delete = NewIntegrationTest(NewIntegrationTestArgs{
 		input.Press(keys.Universal.Remove)
 		input.AcceptConfirmation(Equals("Delete Branch"), Contains("Are you sure you want to delete the branch 'branch-one'?"))
 
-		assert.CurrentViewName("localBranches")
-		assert.CurrentViewLines(
-			MatchesRegexp(`\*.*branch-two`),
-			MatchesRegexp(`master`),
-		)
-		assert.CurrentLineIdx(1)
+		assert.CurrentView().Name("localBranches").
+			Lines(
+				MatchesRegexp(`\*.*branch-two`),
+				MatchesRegexp(`master`),
+			).
+			SelectedLineIdx(1)
 	},
 })