Browse Source

Improve reader_test to find geoip2 databases

Ask Bjørn Hansen 7 years ago
parent
commit
0889af5410
1 changed files with 13 additions and 0 deletions
  1. 13 0
      zones/reader_test.go

+ 13 - 0
zones/reader_test.go

@@ -7,10 +7,23 @@ import (
 	"os"
 	"testing"
 
+	"github.com/abh/geodns/targeting"
+	"github.com/abh/geodns/targeting/geoip2"
 	"github.com/stretchr/testify/assert"
 )
 
 func loadZones(t *testing.T) *MuxManager {
+
+	if targeting.Geo() == nil {
+		t.Logf("Setting up geo provider")
+		geoprovider, err := geoip2.New(geoip2.FindDB())
+		if err == nil {
+			targeting.Setup(geoprovider)
+		} else {
+			t.Fatalf("error setting up geo provider: %s", err)
+		}
+	}
+
 	muxm, err := NewMuxManager("../dns", &NilReg{})
 	if err != nil {
 		t.Logf("loading zones: %s", err)