126 lines
2.6 KiB
Go
126 lines
2.6 KiB
Go
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
package main
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"git.froth.zone/sam/awl/query"
|
|
"github.com/miekg/dns"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestApp(t *testing.T) {
|
|
app := prepareCLI()
|
|
// What more can even be done lmao
|
|
require.NotNil(t, app)
|
|
}
|
|
|
|
func TestArgParse(t *testing.T) {
|
|
tests := []struct {
|
|
in []string
|
|
want query.Answers
|
|
}{
|
|
{
|
|
[]string{"@::1", "localhost", "AAAA"},
|
|
query.Answers{Server: "::1", Request: dns.TypeAAAA, Name: "localhost"},
|
|
},
|
|
{
|
|
[]string{"@1.0.0.1", "google.com"},
|
|
query.Answers{Server: "1.0.0.1", Request: dns.TypeA, Name: "google.com"},
|
|
},
|
|
{
|
|
[]string{"@8.8.4.4"},
|
|
query.Answers{Server: "8.8.4.4", Request: dns.TypeNS, Name: "."},
|
|
},
|
|
}
|
|
for _, test := range tests {
|
|
act, err := parseArgs(test.in, query.Options{})
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, test.want, act)
|
|
}
|
|
}
|
|
|
|
func TestQuery(t *testing.T) {
|
|
app := prepareCLI()
|
|
args := os.Args[0:1]
|
|
args = append(args, "--Treebug")
|
|
err := app.Run(args)
|
|
assert.NotNil(t, err)
|
|
}
|
|
|
|
func TestNoArgs(t *testing.T) {
|
|
app := prepareCLI()
|
|
args := os.Args[0:1]
|
|
args = append(args, "--no-truncate")
|
|
err := app.Run(args)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestFlags(t *testing.T) {
|
|
app := prepareCLI()
|
|
args := os.Args[0:1]
|
|
args = append(args, "--debug")
|
|
args = append(args, "--short")
|
|
args = append(args, "-4")
|
|
err := app.Run(args)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestHTTPS(t *testing.T) {
|
|
app := prepareCLI()
|
|
args := os.Args[0:1]
|
|
args = append(args, "-H")
|
|
args = append(args, "@https://cloudflare-dns.com/dns-query")
|
|
args = append(args, "git.froth.zone")
|
|
err := app.Run(args)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestJSON(t *testing.T) {
|
|
app := prepareCLI()
|
|
args := os.Args[0:1]
|
|
args = append(args, "-j")
|
|
args = append(args, "git.froth.zone")
|
|
err := app.Run(args)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestQUIC(t *testing.T) {
|
|
app := prepareCLI()
|
|
args := os.Args[0:1]
|
|
args = append(args, "-Q")
|
|
args = append(args, "@dns.adguard.com")
|
|
args = append(args, "git.froth.zone")
|
|
err := app.Run(args)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestReverse(t *testing.T) {
|
|
app := prepareCLI()
|
|
args := os.Args[0:1]
|
|
args = append(args, "-x")
|
|
args = append(args, "8.8.8.8")
|
|
err := app.Run(args)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func FuzzCli(f *testing.F) {
|
|
testcases := []string{"git.froth.zone", "", "!12345", "google.com.edu.org.fr"}
|
|
for _, tc := range testcases {
|
|
f.Add(tc)
|
|
}
|
|
|
|
f.Fuzz(func(t *testing.T, orig string) {
|
|
app := prepareCLI()
|
|
args := os.Args[0:1]
|
|
args = append(args, orig)
|
|
err := app.Run(args)
|
|
if err != nil {
|
|
assert.ErrorContains(t, err, "domain must be fully qualified")
|
|
}
|
|
assert.Nil(t, err)
|
|
})
|
|
}
|