65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
|
package spf
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestError_Error(t *testing.T) {
|
||
|
|
err := &Error{error: "record returned neutral", errorType: ErrTypeNeutral}
|
||
|
|
if err.Error() != "record returned neutral" {
|
||
|
|
t.Error("Error() should return 'record returned neutral'")
|
||
|
|
} else if err.Type() != ErrTypeNeutral {
|
||
|
|
t.Error("Type() should return ErrTypeNeutral")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCheckIP(t *testing.T) {
|
||
|
|
includeProviders := true
|
||
|
|
if os.Getenv("SPF_FOSS_PROVIDERS_ONLY") == "1" {
|
||
|
|
fmt.Println("Running tests without non-FOSS providers")
|
||
|
|
includeProviders = false
|
||
|
|
}
|
||
|
|
|
||
|
|
// Tests for a simple IPv4 address
|
||
|
|
err := CheckIP("209.51.188.0", "gnu.org")
|
||
|
|
if err != nil {
|
||
|
|
t.Error("CheckIP should return nil, got", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if includeProviders {
|
||
|
|
// Tests for Google's complex nested IPv4 addresses with redirects and includes
|
||
|
|
err = CheckIP("35.190.247.0", "gmail.com")
|
||
|
|
if err != nil {
|
||
|
|
t.Error("CheckIP should return nil, got", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Tests for mailgun.org's complex nested IPv4 addresses with redirects and includes
|
||
|
|
err = CheckIP("146.20.113.0", "mailgun.org")
|
||
|
|
if err != nil {
|
||
|
|
t.Error("CheckIP should return nil, got", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Tests for PTR records which yahoo still uses for no good reason
|
||
|
|
err = CheckIP("27.123.204.128", "yahoo.com")
|
||
|
|
if err != nil {
|
||
|
|
t.Error("CheckIP should return nil, got", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Tests for cta.social's A and/or MX records
|
||
|
|
ctaSocialIP, _ := net.LookupIP("cta.social")
|
||
|
|
err = CheckIP(ctaSocialIP[0].String(), "cta.social")
|
||
|
|
if err != nil {
|
||
|
|
t.Error("CheckIP should return nil, got", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Tests for danwin1210.de's simple IPv6 address
|
||
|
|
err = CheckIP("2a01:4f8:c010:d56::1", "danwin1210.de")
|
||
|
|
if err != nil {
|
||
|
|
t.Error("CheckIP should return nil, got", err)
|
||
|
|
}
|
||
|
|
}
|