mirror of https://github.com/cbednarski/hostess
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
42 lines
1007 B
Go
42 lines
1007 B
Go
9 years ago
|
package hostess_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/cbednarski/hostess"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestContainsDomainIp(t *testing.T) {
|
||
|
|
hosts := []hostess.Hostname{
|
||
|
|
hostess.Hostname{domain, ip, false},
|
||
|
|
hostess.Hostname{"google.com", "8.8.8.8", true},
|
||
|
|
}
|
||
|
|
|
||
|
|
if !hostess.ContainsDomain(hosts, domain) {
|
||
|
|
t.Errorf("Expected to find %s", domain)
|
||
|
|
}
|
||
|
|
|
||
|
|
const extra_domain = "yahoo.com"
|
||
|
|
if hostess.ContainsDomain(hosts, extra_domain) {
|
||
|
|
t.Errorf("Did not expect to find %s", extra_domain)
|
||
|
|
}
|
||
|
|
|
||
|
|
if !hostess.ContainsIp(hosts, ip) {
|
||
|
|
t.Errorf("Expected to find %s", ip)
|
||
|
|
}
|
||
|
|
|
||
|
|
const extra_ip = "1.2.3.4"
|
||
|
|
if hostess.ContainsIp(hosts, extra_ip) {
|
||
|
|
t.Errorf("Did not expect to find %s", extra_ip)
|
||
|
|
}
|
||
|
|
|
||
|
|
hostname := hostess.Hostname{domain, ip, true}
|
||
|
|
if !hostess.ContainsHostname(hosts, hostname) {
|
||
|
|
t.Errorf("Expected to find %s", hostname)
|
||
|
|
}
|
||
|
|
|
||
|
|
extra_hostname := hostess.Hostname{"yahoo.com", "4.3.2.1", false}
|
||
|
|
if hostess.ContainsHostname(hosts, extra_hostname) {
|
||
|
|
t.Errorf("Did not expect to find %s", extra_hostname)
|
||
|
|
}
|
||
|
|
}
|