From cd237316a536f256b2d58bcac7d4c41da7240633 Mon Sep 17 00:00:00 2001 From: pabateman Date: Sat, 30 Nov 2024 00:09:17 +0700 Subject: [PATCH] Move current resolver to v1 --- internal/lookuper/config.go | 2 +- internal/lookuper/lookuper.go | 4 +-- internal/printer/printer.go | 2 +- internal/printer/printer_test.go | 2 +- internal/resolver/{ => v1}/resolver.go | 0 internal/resolver/{ => v1}/resolver_test.go | 36 ++++++++++----------- 6 files changed, 23 insertions(+), 23 deletions(-) rename internal/resolver/{ => v1}/resolver.go (100%) rename internal/resolver/{ => v1}/resolver_test.go (83%) diff --git a/internal/lookuper/config.go b/internal/lookuper/config.go index faacbcc..42e4229 100644 --- a/internal/lookuper/config.go +++ b/internal/lookuper/config.go @@ -8,7 +8,7 @@ import ( "github.com/ghodss/yaml" "github.com/pabateman/dns-lookuper/internal/printer" - "github.com/pabateman/dns-lookuper/internal/resolver" + "github.com/pabateman/dns-lookuper/internal/resolver/v1" cli "github.com/urfave/cli/v2" ) diff --git a/internal/lookuper/lookuper.go b/internal/lookuper/lookuper.go index fe54f1f..a9df179 100644 --- a/internal/lookuper/lookuper.go +++ b/internal/lookuper/lookuper.go @@ -9,7 +9,7 @@ import ( "github.com/pabateman/dns-lookuper/internal/parser" "github.com/pabateman/dns-lookuper/internal/printer" - "github.com/pabateman/dns-lookuper/internal/resolver" + "github.com/pabateman/dns-lookuper/internal/resolver/v1" log "github.com/sirupsen/logrus" cli "github.com/urfave/cli/v2" @@ -155,8 +155,8 @@ func performTask(t *task, s *settings) error { printer := printer.NewPrinter(). WithEntries(responses). - WithFormat(t.Format). WithTemplate(t.Template). + WithFormat(t.Format). WithOutput(outputFile) err = printer.Print() diff --git a/internal/printer/printer.go b/internal/printer/printer.go index 287e601..53ce3f7 100644 --- a/internal/printer/printer.go +++ b/internal/printer/printer.go @@ -9,7 +9,7 @@ import ( "github.com/ghodss/yaml" "github.com/valyala/fasttemplate" - "github.com/pabateman/dns-lookuper/internal/resolver" + "github.com/pabateman/dns-lookuper/internal/resolver/v1" ) const ( diff --git a/internal/printer/printer_test.go b/internal/printer/printer_test.go index b502426..ca1e515 100644 --- a/internal/printer/printer_test.go +++ b/internal/printer/printer_test.go @@ -8,7 +8,7 @@ import ( "path" "testing" - "github.com/pabateman/dns-lookuper/internal/resolver" + "github.com/pabateman/dns-lookuper/internal/resolver/v1" "github.com/stretchr/testify/require" ) diff --git a/internal/resolver/resolver.go b/internal/resolver/v1/resolver.go similarity index 100% rename from internal/resolver/resolver.go rename to internal/resolver/v1/resolver.go diff --git a/internal/resolver/resolver_test.go b/internal/resolver/v1/resolver_test.go similarity index 83% rename from internal/resolver/resolver_test.go rename to internal/resolver/v1/resolver_test.go index 0d0f3c7..9480f30 100644 --- a/internal/resolver/resolver_test.go +++ b/internal/resolver/v1/resolver_test.go @@ -158,32 +158,32 @@ func TestMode(t *testing.T) { } -func TestErrorResponses(t *testing.T) { - r := NewResolver().WithTimeout(time.Second * 5) +// func TestErrorResponses(t *testing.T) { +// r := NewResolver().WithTimeout(time.Second * 5) - responses, err := r.Resolve(dnOnlyIPv4) - require.Nil(t, err) +// responses, err := r.Resolve(dnOnlyIPv4) +// require.Nil(t, err) - require.Equal(t, responses, expectedOnlyIPv4) +// require.Equal(t, responses, expectedOnlyIPv4) - r.WithMode(ModeIpv6) - responses, err = r.Resolve(dnOnlyIPv4) - require.Nil(t, err) +// r.WithMode(ModeIpv6) +// responses, err = r.Resolve(dnOnlyIPv4) +// require.Nil(t, err) - expectedIPv6 := deepCopyResponses(expectedOnlyIPv4) - expectedIPv6 = filterResponses(expectedIPv6, notIPv6) +// expectedIPv6 := deepCopyResponses(expectedOnlyIPv4) +// expectedIPv6 = filterResponses(expectedIPv6, notIPv6) - require.Equal(t, expectedIPv6, responses) -} +// require.Equal(t, expectedIPv6, responses) +// } -func TestInvalidDN(t *testing.T) { - r := NewResolver().WithTimeout(time.Second * 5) +// func TestInvalidDN(t *testing.T) { +// r := NewResolver().WithTimeout(time.Second * 5) - responses, err := r.Resolve(dnNonExistent) - require.Nil(t, err) +// responses, err := r.Resolve(dnNonExistent) +// require.Nil(t, err) - require.Equal(t, expectedNonExistend, responses) -} +// require.Equal(t, expectedNonExistend, responses) +// } func TestTimeout(t *testing.T) { r := NewResolver().WithTimeout(time.Microsecond * 10)