whitelist_test.go

package dnsp_test

import (
	"io/ioutil"
	"os"
	"testing"

	"github.com/gophergala/dnsp"
)

func TestIsAllowedWhite(t *testing.T) {
	t.Parallel()

	tmp, err := ioutil.TempFile("", "")
	if err != nil {
		t.Fatal(err)
	}
	defer os.Remove(tmp.Name())

	tmp.Write([]byte("*.wikipedia.org\n"))
	tmp.Write([]byte("github.com\n"))
	tmp.Write([]byte("google.com\n"))

	s, err := dnsp.NewServer(dnsp.Options{
		Whitelist: tmp.Name(),
	})
	if err != nil {
		t.Fatal(err)
	}

	for host, ok := range map[string]bool{
		"blocked.net.":      false,
		"en.wikipedia.org.": true,
		"example.com.":      false,
		"github.com.":       true,
		"google.com.":       true,
		"hu.wikipedia.org.": true,
		"wikipedia.org":     false,
	} {
		if act := s.IsAllowed(host); ok != act {
			t.Errorf("expected s.IsAllowed(%q) to be %v, got %v", host, ok, act)
		}
	}
}

func TestIsAllowedBlack(t *testing.T) {
	t.Parallel()

	tmp, err := ioutil.TempFile("", "")
	if err != nil {
		t.Fatal(err)
	}
	defer os.Remove(tmp.Name())

	tmp.Write([]byte("*.xxx\n"))
	tmp.Write([]byte("*.sex\n"))
	tmp.Write([]byte("doubleclick.net\n"))
	tmp.Write([]byte("porn.com\n"))

	s, err := dnsp.NewServer(dnsp.Options{
		Blacklist: tmp.Name(),
	})
	if err != nil {
		t.Fatal(err)
	}

	for host, ok := range map[string]bool{
		"bar.spam.xxx.":    false,
		"doubleclick.net.": false,
		"example.com.":     true,
		"foo.sex.":         false,
		"github.com.":      true,
		"google.com.":      true,
		"porn.com.":        false,
	} {
		if act := s.IsAllowed(host); ok != act {
			t.Errorf("expected s.IsAllowed(%q) to be %v, got %v", host, ok, act)
		}
	}
}