Make it possible to handle toasts in integration tests

Use it in two selected tests to demonstrate what it looks like.

1 files changed, 10 insertions, 0 deletions

diff --git a/pkg/gui/gui_driver.go b/pkg/gui/gui_driver.go
index c36e68e93..ddeb3fab4 100644
--- a/pkg/gui/gui_driver.go
+++ b/pkg/gui/gui_driver.go
@@ -20,6 +20,7 @@ import (
 type GuiDriver struct {
 	gui        *Gui
 	isIdleChan chan struct{}
+	toastChan  chan string
 }
 
 var _ integrationTypes.GuiDriver = &GuiDriver{}
@@ -133,3 +134,12 @@ func (self *GuiDriver) SetCaptionPrefix(prefix string) {
 	self.gui.setCaptionPrefix(prefix)
 	self.waitTillIdle()
 }
+
+func (self *GuiDriver) NextToast() *string {
+	select {
+	case t := <-self.toastChan:
+		return &t
+	default:
+		return nil
+	}
+}