Small fixes

This commit is contained in:
Joona Hoikkala 2022-12-25 13:30:38 +02:00
parent 157241994f
commit f90ef442a3
No known key found for this signature in database
GPG Key ID: 1708DAE66E87A524
3 changed files with 25 additions and 26 deletions

View File

@ -31,7 +31,8 @@ func main() {
fmt.Printf("Could not set up logging: %s\n", err) fmt.Printf("Could not set up logging: %s\n", err)
os.Exit(1) os.Exit(1)
} }
defer logger.Sync() // Make sure to flush the zap logger buffer before exiting
defer logger.Sync() //nolint:all
sugar := logger.Sugar() sugar := logger.Sugar()
sugar.Infow("Using config file", sugar.Infow("Using config file",

View File

@ -3,7 +3,6 @@ package acmedns
import ( import (
"go.uber.org/zap" "go.uber.org/zap"
"go.uber.org/zap/zapcore" "go.uber.org/zap/zapcore"
"io/ioutil"
"os" "os"
"syscall" "syscall"
"testing" "testing"
@ -111,7 +110,7 @@ func TestReadConfig(t *testing.T) {
} }
func TestFileCheckPermissionDenied(t *testing.T) { func TestFileCheckPermissionDenied(t *testing.T) {
tmpfile, err := ioutil.TempFile("", "acmedns") tmpfile, err := os.CreateTemp("", "acmedns")
if err != nil { if err != nil {
t.Error("Could not create temporary file") t.Error("Could not create temporary file")
} }
@ -130,7 +129,7 @@ func TestFileCheckNotExists(t *testing.T) {
} }
func TestFileCheckOK(t *testing.T) { func TestFileCheckOK(t *testing.T) {
tmpfile, err := ioutil.TempFile("", "acmedns") tmpfile, err := os.CreateTemp("", "acmedns")
if err != nil { if err != nil {
t.Error("Could not create temporary file") t.Error("Could not create temporary file")
} }

View File

@ -28,10 +28,7 @@ var records = []string{
} }
func loggerHasEntryWithMessage(message string, logObserver *observer.ObservedLogs) bool { func loggerHasEntryWithMessage(message string, logObserver *observer.ObservedLogs) bool {
if len(logObserver.FilterMessage(message).All()) > 0 { return len(logObserver.FilterMessage(message).All()) > 0
return true
}
return false
} }
func fakeConfigAndLogger() (acmedns.AcmeDnsConfig, *zap.SugaredLogger, *observer.ObservedLogs) { func fakeConfigAndLogger() (acmedns.AcmeDnsConfig, *zap.SugaredLogger, *observer.ObservedLogs) {
@ -76,23 +73,6 @@ func (r *resolver) lookup(host string, qtype uint16) (*dns.Msg, error) {
return in, nil return in, nil
} }
func hasExpectedTXTAnswer(answer []dns.RR, cmpTXT string) error {
for _, record := range answer {
// We expect only one answer, so no need to loop through the answer slice
if rec, ok := record.(*dns.TXT); ok {
for _, txtValue := range rec.Txt {
if txtValue == cmpTXT {
return nil
}
}
} else {
errmsg := fmt.Sprintf("Got answer of unexpected type [%q]", answer[0])
return errors.New(errmsg)
}
}
return errors.New("Expected answer not found")
}
func TestQuestionDBError(t *testing.T) { func TestQuestionDBError(t *testing.T) {
config, logger, _ := fakeConfigAndLogger() config, logger, _ := fakeConfigAndLogger()
config.General.Listen = "127.0.0.1:15353" config.General.Listen = "127.0.0.1:15353"
@ -327,7 +307,26 @@ func TestResolveTXT(t *testing.T) {
} }
} }
} }
}*/ }
func hasExpectedTXTAnswer(answer []dns.RR, cmpTXT string) error {
for _, record := range answer {
// We expect only one answer, so no need to loop through the answer slice
if rec, ok := record.(*dns.TXT); ok {
for _, txtValue := range rec.Txt {
if txtValue == cmpTXT {
return nil
}
}
} else {
errmsg := fmt.Sprintf("Got answer of unexpected type [%q]", answer[0])
return errors.New(errmsg)
}
}
return errors.New("Expected answer not found")
}
*/
func TestCaseInsensitiveResolveA(t *testing.T) { func TestCaseInsensitiveResolveA(t *testing.T) {
resolv := resolver{server: "127.0.0.1:15353"} resolv := resolver{server: "127.0.0.1:15353"}