testutil

type RobotsEvaluatorMock struct {
	// optional function to call at connect time
	ConnectFunc func() error
	// optional function to call when RobotsEvaluator is closing
	CloseFunc func() error
	// optional function to call when IsAllowed is called on RobotsEvaluator. IsAllowed will always return true if this
	// is not set
	IsAllowedFunc func(*robotsevaluatorV1.IsAllowedRequest) bool
}

type ScreenshotWriterMock struct {
	// optional function to call at connect time
	ConnectFunc func() error
	// optional function to call when RobotsEvaluator is closing
	CloseFunc func() error
	// optional function to call when Write is called on ScreenshotWriter. Write will always return nil if this
	// is not set
	WriteFunc func(data []byte, metadata screenshotwriter.Metadata) error
}