1
0
Fork 0
forked from TrueCloudLab/lego
lego/providers/dns/exec/exec_test.go
2020-12-09 20:52:21 +01:00

159 lines
3 KiB
Go

package exec
import (
"fmt"
"os"
"strings"
"testing"
"github.com/go-acme/lego/v4/log"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/mock"
"github.com/stretchr/testify/require"
)
func TestDNSProvider_Present(t *testing.T) {
backupLogger := log.Logger
defer func() {
log.Logger = backupLogger
}()
logRecorder := &LogRecorder{}
log.Logger = logRecorder
type expected struct {
args string
error bool
}
testCases := []struct {
desc string
config *Config
expected expected
}{
{
desc: "Standard mode",
config: &Config{
Program: "echo",
Mode: "",
},
expected: expected{
args: "present _acme-challenge.domain. pW9ZKG0xz_PCriK-nCMOjADy9eJcgGWIzkkj2fN4uZM",
},
},
{
desc: "program error",
config: &Config{
Program: "ogellego",
Mode: "",
},
expected: expected{error: true},
},
{
desc: "Raw mode",
config: &Config{
Program: "echo",
Mode: "RAW",
},
expected: expected{
args: "present -- domain token keyAuth",
},
},
}
var message string
logRecorder.On("Println", mock.Anything).Run(func(args mock.Arguments) {
message = args.String(0)
fmt.Fprintln(os.Stdout, "XXX", message)
})
for _, test := range testCases {
t.Run(test.desc, func(t *testing.T) {
message = ""
provider, err := NewDNSProviderConfig(test.config)
require.NoError(t, err)
err = provider.Present("domain", "token", "keyAuth")
if test.expected.error {
require.Error(t, err)
} else {
require.NoError(t, err)
assert.Equal(t, test.expected.args, strings.TrimSpace(message))
}
})
}
}
func TestDNSProvider_CleanUp(t *testing.T) {
backupLogger := log.Logger
defer func() {
log.Logger = backupLogger
}()
logRecorder := &LogRecorder{}
log.Logger = logRecorder
type expected struct {
args string
error bool
}
testCases := []struct {
desc string
config *Config
expected expected
}{
{
desc: "Standard mode",
config: &Config{
Program: "echo",
Mode: "",
},
expected: expected{
args: "cleanup _acme-challenge.domain. pW9ZKG0xz_PCriK-nCMOjADy9eJcgGWIzkkj2fN4uZM",
},
},
{
desc: "program error",
config: &Config{
Program: "ogellego",
Mode: "",
},
expected: expected{error: true},
},
{
desc: "Raw mode",
config: &Config{
Program: "echo",
Mode: "RAW",
},
expected: expected{
args: "cleanup -- domain token keyAuth",
},
},
}
var message string
logRecorder.On("Println", mock.Anything).Run(func(args mock.Arguments) {
message = args.String(0)
fmt.Fprintln(os.Stdout, "XXX", message)
})
for _, test := range testCases {
t.Run(test.desc, func(t *testing.T) {
message = ""
provider, err := NewDNSProviderConfig(test.config)
require.NoError(t, err)
err = provider.CleanUp("domain", "token", "keyAuth")
if test.expected.error {
require.Error(t, err)
} else {
require.NoError(t, err)
assert.Equal(t, test.expected.args, strings.TrimSpace(message))
}
})
}
}