list_test.go

package block

import (
	"os"
	"strings"
	"testing"
)

func TestListParse(t *testing.T) {
	var list = `
# 127.0.0.1	example.com
127.0.0.1	example.org	third
008.free-counter.co.uk
com
`

	b := new(Block)
	os.Remove("/tmp/list_test.db")
	b.setupDB("/tmp/list_test.db")

	r := strings.NewReader(list)
	l := make(map[string]DomainValue)
	listRead(r, l, -1)
	b.update = l

	err := b.UpdateDomains(b.update)
	if err != nil {
		log.Fatal("failed to update list_test -- ", err)
	}

	tests := []struct {
		name    string
		blocked bool
	}{
		{"example.org.", false},
		{"example.com.", true},
		{"com.", true},
	}

	for _, test := range tests {
		retIP := ""
		retCNAME := ""
		hasPermit := false
		categories := []string{}

		got := b.blocked("1.2.3.4", test.name, &retIP, &retCNAME, &hasPermit, &categories)
		if got != test.blocked {
			t.Errorf("Expected %s to be blocked", test.name)
		}
	}
}