1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
package blocklist
import (
"os"
"path/filepath"
"testing"
)
func TestBlockZeroIP(t *testing.T) {
dir := t.TempDir()
p := filepath.Join(dir, "block.txt")
os.WriteFile(p, []byte("||example.com^\n0.0.0.0 doubleclick.net\n"), 0644)
b := New(ResponseZeroIP)
if err := b.LoadFile(p); err != nil {
t.Fatal(err)
}
cases := []struct {
domain string
block bool
}{
{"example.com.", true},
{"sub.example.com.", true},
{"notexample.com.", false},
{"doubleclick.net.", true},
{"ads.doubleclick.net.", true},
{"google.com.", false},
}
for _, c := range cases {
got := b.IsBlocked(c.domain)
if got != c.block {
t.Errorf("IsBlocked(%q) = %v, want %v", c.domain, got, c.block)
}
}
}
func TestException(t *testing.T) {
dir := t.TempDir()
p := filepath.Join(dir, "block.txt")
os.WriteFile(p, []byte("||example.com^\n@@||whitelist.example.com^\n"), 0644)
b := New(ResponseZeroIP)
if err := b.LoadFile(p); err != nil {
t.Fatal(err)
}
if !b.IsBlocked("example.com.") {
t.Error("expected blocked")
}
if b.IsBlocked("whitelist.example.com.") {
t.Error("expected NOT blocked (exception)")
}
if b.IsBlocked("sub.whitelist.example.com.") {
t.Error("expected NOT blocked (exception subdomain)")
}
}
func TestLoadFile(t *testing.T) {
dir := t.TempDir()
p := filepath.Join(dir, "block.txt")
os.WriteFile(p, []byte("0.0.0.0 ads.com\n||tracker.net^\n"), 0644)
b := New(ResponseZeroIP)
if err := b.LoadFile(p); err != nil {
t.Fatal(err)
}
if !b.IsBlocked("ads.com.") {
t.Error("expected blocked")
}
if !b.IsBlocked("sub.tracker.net.") {
t.Error("expected blocked")
}
if b.IsBlocked("safe.org.") {
t.Error("expected NOT blocked")
}
}
func TestResponseType(t *testing.T) {
b := New(ResponseNXDOMAIN)
if b.Response() != ResponseNXDOMAIN {
t.Error("expected NXDOMAIN")
}
b2 := New(ResponseZeroIP)
if b2.Response() != ResponseZeroIP {
t.Error("expected ZeroIP")
}
}
|
