// SPDX-License-Identifier: BSD-3-Clause

package query_test

import (
	"testing"
	"time"

	"git.froth.zone/sam/awl/query"
	"git.froth.zone/sam/awl/util"
	"github.com/miekg/dns"
	"gotest.tools/v3/assert"
)

func TestResolve(t *testing.T) {
	t.Parallel()

	opts := util.Options{
		Logger: util.InitLogger(0),
		Port:   53,
		Request: util.Request{
			Server:  "8.8.4.1",
			Type:    dns.TypeA,
			Name:    "example.com.",
			Timeout: time.Second / 2,
			Retries: 0,
		},
	}
	resolver, err := query.LoadResolver(opts)
	assert.NilError(t, err)

	msg := new(dns.Msg)
	msg.SetQuestion(opts.Request.Name, opts.Request.Type)

	_, err = resolver.LookUp(msg)
	assert.ErrorContains(t, err, "timeout")
}

func TestTruncate(t *testing.T) {
	t.Parallel()

	opts := util.Options{
		Logger: util.InitLogger(0),
		IPv4:   true,
		Port:   5301,
		Request: util.Request{
			Server: "madns.binarystar.systems",
			Type:   dns.TypeTXT,
			Name:   "limit.txt.example.",
		},
	}
	resolver, err := query.LoadResolver(opts)
	assert.NilError(t, err)

	msg := new(dns.Msg)
	msg.SetQuestion(opts.Request.Name, opts.Request.Type)
	res, err := resolver.LookUp(msg)

	assert.NilError(t, err)
	assert.Assert(t, res != util.Response{})
}

func TestResolveAgain(t *testing.T) {
	t.Parallel()

	tests := []struct {
		opt util.Options
	}{
		{
			util.Options{
				Logger: util.InitLogger(0),
				TCP:    true,
				Port:   53,
				Request: util.Request{
					Server: "8.8.4.4",
					Type:   dns.TypeA,
					Name:   "example.com.",
				},
			},
		},
		{
			util.Options{
				Logger: util.InitLogger(0),
				Port:   53,
				Request: util.Request{
					Server: "8.8.4.4",
					Type:   dns.TypeAAAA,
					Name:   "example.com.",
				},
			},
		},
		{
			util.Options{
				Logger: util.InitLogger(0),
				TLS:    true,
				Port:   853,
				Request: util.Request{
					Server: "dns.google",
					Type:   dns.TypeAAAA,
					Name:   "example.com.",
				},
			},
		},
	}

	for _, test := range tests {
		test := test

		t.Run("", func(t *testing.T) {
			t.Parallel()
			res, err := query.CreateQuery(test.opt)
			assert.NilError(t, err)
			assert.Assert(t, res != util.Response{})
		})
	}
}