mirror of
https://github.com/jesseduffield/lazygit.git
synced 2025-12-04 04:22:37 +03:00
112 lines
3.6 KiB
Go
112 lines
3.6 KiB
Go
package components
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/jesseduffield/gocui"
|
|
)
|
|
|
|
type ViewAsserter struct {
|
|
// context is prepended to any error messages e.g. 'context: "current view"'
|
|
context string
|
|
getView func() *gocui.View
|
|
assert *Assert
|
|
}
|
|
|
|
// asserts that the view has the expected name. This is typically used in tandem with the CurrentView method i.e.;
|
|
// assert.CurrentView().Name("commits") to assert that the current view is the commits view.
|
|
func (self *ViewAsserter) Name(expected string) *ViewAsserter {
|
|
self.assert.assertWithRetries(func() (bool, string) {
|
|
actual := self.getView().Name()
|
|
return actual == expected, fmt.Sprintf("%s: Expected view name to be '%s', but got '%s'", self.context, expected, actual)
|
|
})
|
|
|
|
return self
|
|
}
|
|
|
|
// asserts that the view has the expected title
|
|
func (self *ViewAsserter) Title(expected *matcher) *ViewAsserter {
|
|
self.assert.assertWithRetries(func() (bool, string) {
|
|
actual := self.getView().Title
|
|
return expected.context(fmt.Sprintf("%s title", self.context)).test(actual)
|
|
})
|
|
|
|
return self
|
|
}
|
|
|
|
// asserts that the view has lines matching the given matchers. So if three matchers
|
|
// are passed, we only check the first three lines of the view.
|
|
// This method is convenient when you have a list of commits but you only want to
|
|
// assert on the first couple of commits.
|
|
func (self *ViewAsserter) TopLines(matchers ...*matcher) *ViewAsserter {
|
|
self.assert.assertWithRetries(func() (bool, string) {
|
|
lines := self.getView().BufferLines()
|
|
return len(lines) >= len(matchers), fmt.Sprintf("unexpected number of lines in view. Expected at least %d, got %d", len(matchers), len(lines))
|
|
})
|
|
|
|
view := self.getView()
|
|
|
|
for i, matcher := range matchers {
|
|
self.assert.matchString(matcher, fmt.Sprintf("Unexpected content in view '%s'.", view.Name()),
|
|
func() string {
|
|
return view.BufferLines()[i]
|
|
},
|
|
)
|
|
}
|
|
|
|
return self
|
|
}
|
|
|
|
// asserts that the view has lines matching the given matchers. One matcher must be passed for each line.
|
|
// If you only care about the top n lines, use the TopLines method instead.
|
|
func (self *ViewAsserter) Lines(matchers ...*matcher) *ViewAsserter {
|
|
self.assert.assertWithRetries(func() (bool, string) {
|
|
lines := self.getView().BufferLines()
|
|
return len(lines) == len(matchers), fmt.Sprintf("unexpected number of lines in view. Expected %d, got %d", len(matchers), len(lines))
|
|
})
|
|
|
|
view := self.getView()
|
|
|
|
for i, matcher := range matchers {
|
|
self.assert.matchString(matcher, fmt.Sprintf("Unexpected content in view '%s'.", view.Name()),
|
|
func() string {
|
|
return view.BufferLines()[i]
|
|
},
|
|
)
|
|
}
|
|
|
|
return self
|
|
}
|
|
|
|
// asserts on the content of the view i.e. the stuff within the view's frame.
|
|
func (self *ViewAsserter) Content(matcher *matcher) *ViewAsserter {
|
|
self.assert.matchString(matcher, fmt.Sprintf("%s: Unexpected content.", self.context),
|
|
func() string {
|
|
return self.getView().Buffer()
|
|
},
|
|
)
|
|
|
|
return self
|
|
}
|
|
|
|
// asserts on the selected line of the view
|
|
func (self *ViewAsserter) SelectedLine(matcher *matcher) *ViewAsserter {
|
|
self.assert.matchString(matcher, fmt.Sprintf("%s: Unexpected selected line.", self.context),
|
|
func() string {
|
|
return self.getView().SelectedLine()
|
|
},
|
|
)
|
|
|
|
return self
|
|
}
|
|
|
|
// asserts on the index of the selected line. 0 is the first index, representing the line at the top of the view.
|
|
func (self *ViewAsserter) SelectedLineIdx(expected int) *ViewAsserter {
|
|
self.assert.assertWithRetries(func() (bool, string) {
|
|
actual := self.getView().SelectedLineIdx()
|
|
return expected == actual, fmt.Sprintf("%s: Expected selected line index to be %d, got %d", self.context, expected, actual)
|
|
})
|
|
|
|
return self
|
|
}
|