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)
	}
}