// 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") } // FIXME: See when this is fixed and fix it. func TestTruncate(t *testing.T) { t.Skip("Sadly this is broken.") 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{}) }) } }