forked from TrueCloudLab/lego
89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
package internal
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func setupTest(t *testing.T, credentials map[string]string, handler http.HandlerFunc) *Client {
|
|
t.Helper()
|
|
|
|
mux := http.NewServeMux()
|
|
server := httptest.NewServer(mux)
|
|
t.Cleanup(server.Close)
|
|
|
|
mux.HandleFunc("/", handler)
|
|
|
|
client := NewClient(credentials)
|
|
client.HTTPClient = server.Client()
|
|
client.baseURL = server.URL
|
|
|
|
return client
|
|
}
|
|
|
|
func TestClient_Add(t *testing.T) {
|
|
txtValue := "123456789012"
|
|
|
|
client := setupTest(t, map[string]string{"example.org": "secret"}, handlerMock(addAction, txtValue))
|
|
|
|
err := client.Add(context.Background(), "example.org", txtValue)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestClient_Add_error(t *testing.T) {
|
|
txtValue := "123456789012"
|
|
|
|
client := setupTest(t, map[string]string{"example.com": "secret"}, handlerMock(addAction, txtValue))
|
|
|
|
err := client.Add(context.Background(), "example.org", txtValue)
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestClient_Remove(t *testing.T) {
|
|
txtValue := "ABCDEFGHIJKL"
|
|
|
|
client := setupTest(t, map[string]string{"example.org": "secret"}, handlerMock(removeAction, txtValue))
|
|
|
|
err := client.Remove(context.Background(), "example.org", txtValue)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestClient_Remove_error(t *testing.T) {
|
|
txtValue := "ABCDEFGHIJKL"
|
|
|
|
client := setupTest(t, map[string]string{"example.com": "secret"}, handlerMock(removeAction, txtValue))
|
|
|
|
err := client.Remove(context.Background(), "example.org", txtValue)
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func handlerMock(action, value string) http.HandlerFunc {
|
|
return func(rw http.ResponseWriter, req *http.Request) {
|
|
rw.WriteHeader(http.StatusOK)
|
|
|
|
query := req.URL.Query()
|
|
|
|
if query.Get("acme") != action {
|
|
_, _ = rw.Write([]byte("nochg 1234:1234:1234:1234:1234:1234:1234:1234"))
|
|
return
|
|
}
|
|
|
|
txtValue := query.Get("txt")
|
|
if len(txtValue) < 12 {
|
|
_, _ = rw.Write([]byte("error - no valid acme txt record"))
|
|
return
|
|
}
|
|
|
|
if txtValue != value {
|
|
http.Error(rw, fmt.Sprintf("got: %q, expected: %q", txtValue, value), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
_, _ = fmt.Fprintf(rw, "%s %s", successCode, txtValue)
|
|
}
|
|
}
|