1 package dpms 2 3 import ( 4 "github.com/stretchr/testify/mock" 5 6 "edge-infra.dev/pkg/sds/lib/xorg/dpms" 7 ) 8 9 type fakeDPMS struct { 10 Mock mock.Mock 11 } 12 13 func NewFakeDPMS() (dpms.DPMS, *mock.Mock) { 14 d := &fakeDPMS{} 15 return d, &d.Mock 16 } 17 18 func (d *fakeDPMS) GetDPMSInfo() (*dpms.Info, error) { 19 args := d.Mock.Called() 20 if args.Get(0) == nil { 21 return nil, args.Error(1) 22 } 23 return args.Get(0).(*dpms.Info), args.Error(1) 24 } 25