package xrandr import ( "time" "github.com/stretchr/testify/mock" "edge-infra.dev/pkg/sds/lib/xorg/xrandr" ) type fakeXrandr struct { Mock mock.Mock } func NewFakeXrandr() (xrandr.Xrandr, *mock.Mock) { x := &fakeXrandr{} return x, &x.Mock } func (x *fakeXrandr) GetOutputs() (xrandr.Outputs, error) { args := x.Mock.Called() if args.Get(0) == nil { return nil, args.Error(1) } return args.Get(0).(xrandr.Outputs), args.Error(1) } func (x *fakeXrandr) GetPoweredOutputs() (xrandr.Outputs, error) { args := x.Mock.Called() if args.Get(0) == nil { return nil, args.Error(1) } return args.Get(0).(xrandr.Outputs), args.Error(1) } func (x *fakeXrandr) WaitUntilReady(minWaitTime, timeout time.Duration) error { args := x.Mock.Called(minWaitTime, timeout) return args.Error(0) }