a0246ccee2
Includes a few minor fixes: - frontend: support setting port for WHOIS server - proxy: fix handling of very long lines - proxy: refactor IP allowlist logic, parse allow IP list at startup
41 lines
880 B
Go
41 lines
880 B
Go
package main
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/magiconair/properties/assert"
|
|
)
|
|
|
|
func TestGetASNRepresentationDNS(t *testing.T) {
|
|
checkNetwork(t)
|
|
|
|
setting.dnsInterface = "asn.cymru.com"
|
|
setting.whoisServer = ""
|
|
cache := make(ASNCache)
|
|
result := cache.Lookup("6939")
|
|
if !strings.Contains(result, "HURRICANE") {
|
|
t.Errorf("Lookup AS6939 failed, got %s", result)
|
|
}
|
|
}
|
|
|
|
func TestGetASNRepresentationWhois(t *testing.T) {
|
|
checkNetwork(t)
|
|
|
|
setting.dnsInterface = ""
|
|
setting.whoisServer = "whois.arin.net"
|
|
cache := make(ASNCache)
|
|
result := cache.Lookup("6939")
|
|
if !strings.Contains(result, "HURRICANE") {
|
|
t.Errorf("Lookup AS6939 failed, got %s", result)
|
|
}
|
|
}
|
|
|
|
func TestGetASNRepresentationFallback(t *testing.T) {
|
|
setting.dnsInterface = ""
|
|
setting.whoisServer = ""
|
|
cache := make(ASNCache)
|
|
result := cache.Lookup("6939")
|
|
assert.Equal(t, result, "AS6939")
|
|
}
|