fixup! Add integration tests for checkout/new branch with autostash

Add assertions for modified file
jesseduffield · Sep 3, 2024 · 9eb2a4f · 9eb2a4f
1 parent 4c4c655
commit 9eb2a4f
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/pkg/integration/tests/branch/checkout_autostash.go b/pkg/integration/tests/branch/checkout_autostash.go
@@ -20,6 +20,11 @@ var CheckoutAutostash = NewIntegrationTest(NewIntegrationTestArgs{
 		shell.UpdateFile("file", "b\n\nb")
 	},
 	Run: func(t *TestDriver, keys config.KeybindingConfig) {
+		t.Views().Files().
+			Lines(
+				Contains("file"),
+			)
+
 		t.Views().Branches().
 			Focus().
 			Lines(
@@ -40,5 +45,10 @@ var CheckoutAutostash = NewIntegrationTest(NewIntegrationTestArgs{
 			)
 
 		t.Git().CurrentBranchName("master")
+
+		t.Views().Files().
+			Lines(
+				Contains("file"),
+			)
 	},
 })
diff --git a/pkg/integration/tests/branch/new_branch_autostash.go b/pkg/integration/tests/branch/new_branch_autostash.go
@@ -20,6 +20,11 @@ var NewBranchAutostash = NewIntegrationTest(NewIntegrationTestArgs{
 		shell.UpdateFile("file", "b\n\nb")
 	},
 	Run: func(t *TestDriver, keys config.KeybindingConfig) {
+		t.Views().Files().
+			Lines(
+				Contains("file"),
+			)
+
 		t.Views().Branches().
 			Focus().
 			Lines(
@@ -46,5 +51,10 @@ var NewBranchAutostash = NewIntegrationTest(NewIntegrationTestArgs{
 			)
 
 		t.Git().CurrentBranchName("new-branch")
+
+		t.Views().Files().
+			Lines(
+				Contains("file"),
+			)
 	},
 })