godnsbl_test.go

package godnsbl_test

import (
	"testing"

	"github.com/HelixSpiral/godnsbl"
)

func TestLookupIP(t *testing.T) {
	dnsbl := godnsbl.NewLookupService()

	dnsbl.DnsblListing = append(dnsbl.DnsblListing, godnsbl.Dnsbl{
		Name:         "Test - DroneBL",
		Address:      ".dnsbl.dronebl.org",
		BlockMessage: "%IPADDR found",
	})

	reply := dnsbl.LookupIP("127.0.0.2")

	if reply.Type != "BLOCK" {
		t.Errorf("Expended a ban, got: %+v\r\n", reply)
	}

	reply = dnsbl.LookupIP("2001:4860:4860::8888")

	if reply.Type != "CLEAR" {
		t.Errorf("Expended a clear, got: %+v\r\n", reply)
	}
}

func TestLookupIPGetAll(t *testing.T) {
	dnsbl := godnsbl.NewLookupService()

	dnsbl.DnsblListing = append(dnsbl.DnsblListing, godnsbl.Dnsbl{
		Name:         "Test - DroneBL",
		Address:      ".dnsbl.dronebl.org",
		BlockMessage: "%IPADDR found",
	})

	dnsbl.DnsblListing = append(dnsbl.DnsblListing, godnsbl.Dnsbl{
		Name:         "Test2 - DroneBL",
		Address:      ".dnsbl.dronebl.org",
		BlockMessage: "%IPADDR found",
	})

	replies := dnsbl.LookupIPGetAll("127.0.0.2")

	var foundBlock bool
	for _, reply := range replies {
		if reply.Type == "BLOCK" {
			foundBlock = true
			break
		}
	}

	if foundBlock != true {
		t.Errorf("Did not find blocking reply: %+v\r\n", replies)
	}

	replies = dnsbl.LookupIPGetAll("2001:4860:4860::8888")

	var foundClear bool
	for _, reply := range replies {
		if reply.Type == "CLEAR" {
			foundClear = true
			break
		}
	}

	if foundClear != true {
		t.Errorf("Did not find clearing reply: %+v\r\n", replies)
	}
}