package dpms

import (
	"github.com/stretchr/testify/mock"

	"edge-infra.dev/pkg/sds/lib/xorg/dpms"
)

type fakeDPMS struct {
	Mock mock.Mock
}

func NewFakeDPMS() (dpms.DPMS, *mock.Mock) {
	d := &fakeDPMS{}
	return d, &d.Mock
}

func (d *fakeDPMS) GetDPMSInfo() (*dpms.Info, error) {
	args := d.Mock.Called()
	if args.Get(0) == nil {
		return nil, args.Error(1)
	}
	return args.Get(0).(*dpms.Info), args.Error(1)
}