mirror of
https://github.com/jesseduffield/lazygit.git
synced 2025-01-26 05:37:18 +02:00
56 lines
1.2 KiB
Go
56 lines
1.2 KiB
Go
package components
|
|
|
|
type MenuDriver struct {
|
|
t *TestDriver
|
|
hasCheckedTitle bool
|
|
}
|
|
|
|
func (self *MenuDriver) getViewDriver() *ViewDriver {
|
|
return self.t.Views().Menu()
|
|
}
|
|
|
|
// asserts that the popup has the expected title
|
|
func (self *MenuDriver) Title(expected *Matcher) *MenuDriver {
|
|
self.getViewDriver().Title(expected)
|
|
|
|
self.hasCheckedTitle = true
|
|
|
|
return self
|
|
}
|
|
|
|
func (self *MenuDriver) Confirm() {
|
|
self.checkNecessaryChecksCompleted()
|
|
|
|
self.getViewDriver().PressEnter()
|
|
}
|
|
|
|
func (self *MenuDriver) Cancel() {
|
|
self.checkNecessaryChecksCompleted()
|
|
|
|
self.getViewDriver().PressEscape()
|
|
}
|
|
|
|
func (self *MenuDriver) Select(option *Matcher) *MenuDriver {
|
|
self.getViewDriver().NavigateToLine(option)
|
|
|
|
return self
|
|
}
|
|
|
|
func (self *MenuDriver) Lines(matchers ...*Matcher) *MenuDriver {
|
|
self.getViewDriver().Lines(matchers...)
|
|
|
|
return self
|
|
}
|
|
|
|
func (self *MenuDriver) TopLines(matchers ...*Matcher) *MenuDriver {
|
|
self.getViewDriver().TopLines(matchers...)
|
|
|
|
return self
|
|
}
|
|
|
|
func (self *MenuDriver) checkNecessaryChecksCompleted() {
|
|
if !self.hasCheckedTitle {
|
|
self.t.Fail("You must check the title of a menu popup by calling Title() before calling Confirm()/Cancel().")
|
|
}
|
|
}
|