Add demo test variant

We're piggybacking on our existing integration test framework to record  demos that we can include in our docs

1 files changed, 15 insertions, 1 deletions

diff --git a/pkg/gui/gui_driver.go b/pkg/gui/gui_driver.go
index 630da5b0b..8bab6218d 100644
--- a/pkg/gui/gui_driver.go
+++ b/pkg/gui/gui_driver.go
@@ -42,7 +42,11 @@ func (self *GuiDriver) PressKey(keyStr string) {
 		0,
 	)
 
-	// wait until lazygit is idle (i.e. all processing is done) before continuing
+	self.waitTillIdle()
+}
+
+// wait until lazygit is idle (i.e. all processing is done) before continuing
+func (self *GuiDriver) waitTillIdle() {
 	<-self.isIdleChan
 }
 
@@ -111,3 +115,13 @@ func (self *GuiDriver) View(viewName string) *gocui.View {
 	}
 	return view
 }
+
+func (self *GuiDriver) SetCaption(caption string) {
+	self.gui.setCaption(caption)
+	self.waitTillIdle()
+}
+
+func (self *GuiDriver) SetCaptionPrefix(prefix string) {
+	self.gui.setCaptionPrefix(prefix)
+	self.waitTillIdle()
+}