http_test.go

package main

import (
	"bytes"
	"io"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/stretchr/testify/require"

	"github.com/abh/geodns/v3/targeting"
	"github.com/abh/geodns/v3/targeting/geoip2"
	"github.com/abh/geodns/v3/zones"
)

func TestHTTP(t *testing.T) {
	geoprovider, err := geoip2.New(geoip2.FindDB())
	if err == nil {
		targeting.Setup(geoprovider)
	}

	mm, err := zones.NewMuxManager("dns", &zones.NilReg{})
	if err != nil {
		t.Fatalf("loading zones: %s", err)
	}
	hs := NewHTTPServer(mm, serverInfo)

	srv := httptest.NewServer(hs.Mux())

	baseurl := srv.URL
	t.Logf("server base url: '%s'", baseurl)

	// metrics := NewMetrics()
	// go metrics.Updater()

	res, err := http.Get(baseurl + "/version")
	require.Nil(t, err)
	page, _ := io.ReadAll(res.Body)

	if !bytes.HasPrefix(page, []byte("GeoDNS ")) {
		t.Log("/version didn't start with 'GeoDNS '")
		t.Fail()
	}
}