aboutsummaryrefslogtreecommitdiff
path: root/internal/dnsbl/dnsbl_test.go
blob: ccadd64af492416c4660853cbbd6444fe813791c (plain)
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
package dnsbl

import (
	"fmt"
	"net"
	"os"
	"testing"
)

func TestReverse4(t *testing.T) {
	cases := []struct {
		inp, out string
	}{
		{"1.2.3.4", "4.3.2.1"},
	}

	for _, cs := range cases {
		t.Run(fmt.Sprintf("%s->%s", cs.inp, cs.out), func(t *testing.T) {
			out := reverse4(net.ParseIP(cs.inp))

			if out != cs.out {
				t.Errorf("wanted %s\ngot:   %s", cs.out, out)
			}
		})
	}
}

func TestReverse6(t *testing.T) {
	cases := []struct {
		inp, out string
	}{
		{
			inp: "1234:5678:9ABC:DEF0:1234:5678:9ABC:DEF0",
			out: "0.f.e.d.c.b.a.9.8.7.6.5.4.3.2.1.0.f.e.d.c.b.a.9.8.7.6.5.4.3.2.1",
		},
	}

	for _, cs := range cases {
		t.Run(fmt.Sprintf("%s->%s", cs.inp, cs.out), func(t *testing.T) {
			out := reverse6(net.ParseIP(cs.inp))

			if out != cs.out {
				t.Errorf("wanted %s, got: %s", cs.out, out)
			}
		})
	}
}

func TestLookup(t *testing.T) {
	if os.Getenv("DONT_USE_NETWORK") != "" {
		t.Skip("test requires network egress")
		return
	}

	resp, err := Lookup("27.65.243.194")
	if err != nil {
		t.Fatalf("it broked: %v", err)
	}

	t.Logf("response: %d", resp)
}