package test

import (
	"context"
	"net"
)

type TestIPGetter struct {
	PublicIPProp net.IP
	PublicIPFunc func(ctx context.Context) (net.IP, error)
}

func (g *TestIPGetter) GetCurrentPublicIP(ctx context.Context) (net.IP, error) {
	if g.PublicIPFunc == nil {
		return g.PublicIPProp, nil
	}
	return g.PublicIPFunc(ctx)
}