lego/providers/dns/sonic/internal/client_test.go

package internal

import (
	"context"
	"fmt"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/stretchr/testify/require"
)

func setupTest(t *testing.T, body string) *Client {
	t.Helper()

	mux := http.NewServeMux()
	server := httptest.NewServer(mux)
	t.Cleanup(server.Close)

	mux.HandleFunc("/host", func(rw http.ResponseWriter, req *http.Request) {
		_, _ = fmt.Fprintln(rw, body)
	})

	client, err := NewClient("foo", "secret")
	require.NoError(t, err)

	client.baseURL = server.URL

	return client
}

func TestClient_SetRecord(t *testing.T) {
	testCases := []struct {
		desc     string
		response string
		assert   require.ErrorAssertionFunc
	}{
		{
			desc:     "success",
			response: `{"message":"OK","result":200}`,
			assert:   require.NoError,
		},
		{
			desc:     "failure",
			response: `{"message":"Not Found :  the information you requested was not found.","result":404}`,
			assert:   require.Error,
		},
	}

	for _, test := range testCases {
		t.Run(test.desc, func(t *testing.T) {
			t.Parallel()

			client := setupTest(t, test.response)

			err := client.SetRecord(context.Background(), "example.com", "txttxttxt", 10)
			test.assert(t, err)
		})
	}
}