fix compilation error

This commit is contained in:
Jakub Zimny
2025-01-20 23:38:56 +01:00
parent 84f9726b09
commit 5b43c79783

View File

@@ -13,6 +13,7 @@ type MockState struct {
EqualsError bool EqualsError bool
GetReturn EntityState GetReturn EntityState
GetError bool GetError bool
AllEntities []EntityState
} }
func (s MockState) AfterSunrise(_ ...DurationString) bool { func (s MockState) AfterSunrise(_ ...DurationString) bool {
@@ -33,6 +34,9 @@ func (s MockState) Get(eid string) (EntityState, error) {
} }
return s.GetReturn, nil return s.GetReturn, nil
} }
func (s MockState) List() ([]EntityState, error) {
return s.AllEntities, nil
}
func (s MockState) Equals(eid, state string) (bool, error) { func (s MockState) Equals(eid, state string) (bool, error) {
if s.EqualsError { if s.EqualsError {
return false, errors.New("some error") return false, errors.New("some error")