Fix custom URL for HTTP ip method (fix #203)

This commit is contained in:
Quentin McGaw
2021-05-22 19:42:07 +00:00
parent 201df818d3
commit 398566850d
2 changed files with 8 additions and 0 deletions

View File

@@ -47,6 +47,10 @@ var (
)
func ValidateProvider(provider Provider, version ipversion.IPVersion) error {
if strings.HasPrefix(string(provider), "url:https://") { // custom HTTP url
return nil
}
for _, possible := range ListProviders() {
if provider == possible {
_, ok := provider.url(version)

View File

@@ -53,6 +53,10 @@ func Test_ValidateProvider(t *testing.T) {
provider: Google,
version: ipversion.IP4or6,
},
"custom url": {
provider: Provider("url:https://ip.com"),
version: ipversion.IP4or6,
},
"invalid for ip version": {
provider: Google,
version: ipversion.IP4,