type TestProxy struct {
	run          func(<-chan error) error
	cleanup      func()
	blockChannel chan any
}

func (tp TestProxy) Run(stop <-chan error) error {
	if tp.run == nil {
		return nil
	}
	return tp.run(stop)
}

func (tp TestProxy) Drain(bool) error {
	tp.blockChannel <- "unblock"
	return nil
}

func (tp TestProxy) Cleanup() {
	if tp.cleanup != nil {
		tp.cleanup()
	}