forked from TrueCloudLab/lego
143 lines
3.3 KiB
Go
143 lines
3.3 KiB
Go
package internal
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func setupTest(t *testing.T, filename string) *Client {
|
|
t.Helper()
|
|
mux := http.NewServeMux()
|
|
server := httptest.NewServer(mux)
|
|
t.Cleanup(server.Close)
|
|
|
|
serverURL, err := url.Parse(server.URL)
|
|
require.NoError(t, err)
|
|
|
|
client := NewClient(serverURL, "user", "secret")
|
|
client.HTTPClient = server.Client()
|
|
|
|
mux.HandleFunc("/enterprise/control/agent.php", func(rw http.ResponseWriter, req *http.Request) {
|
|
if req.Method != http.MethodPost {
|
|
http.Error(rw, fmt.Sprintf("unsupported method: %s", req.Method), http.StatusMethodNotAllowed)
|
|
return
|
|
}
|
|
|
|
login := req.Header.Get("Http_auth_login")
|
|
if login != "user" {
|
|
http.Error(rw, fmt.Sprintf("invalid login: %s", login), http.StatusUnauthorized)
|
|
return
|
|
}
|
|
|
|
password := req.Header.Get("Http_auth_passwd")
|
|
if password != "secret" {
|
|
http.Error(rw, fmt.Sprintf("invalid password: %s", password), http.StatusUnauthorized)
|
|
return
|
|
}
|
|
|
|
file, err := os.Open(filepath.Join("fixtures", filename))
|
|
if err != nil {
|
|
http.Error(rw, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
defer func() { _ = file.Close() }()
|
|
|
|
_, err = io.Copy(rw, file)
|
|
if err != nil {
|
|
http.Error(rw, err.Error(), http.StatusInternalServerError)
|
|
return
|
|
}
|
|
})
|
|
|
|
return client
|
|
}
|
|
|
|
func TestClient_GetSite(t *testing.T) {
|
|
client := setupTest(t, "get-site.xml")
|
|
|
|
siteID, err := client.GetSite("example.com")
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, 82, siteID)
|
|
}
|
|
|
|
func TestClient_GetSite_error(t *testing.T) {
|
|
client := setupTest(t, "get-site-error.xml")
|
|
|
|
siteID, err := client.GetSite("example.com")
|
|
require.Error(t, err)
|
|
|
|
assert.Equal(t, 0, siteID)
|
|
}
|
|
|
|
func TestClient_GetSite_system_error(t *testing.T) {
|
|
client := setupTest(t, "global-error.xml")
|
|
|
|
siteID, err := client.GetSite("example.com")
|
|
require.Error(t, err)
|
|
|
|
assert.Equal(t, 0, siteID)
|
|
}
|
|
|
|
func TestClient_AddRecord(t *testing.T) {
|
|
client := setupTest(t, "add-record.xml")
|
|
|
|
recordID, err := client.AddRecord(123, "_acme-challenge.example.com", "txtTXTtxt")
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, 4537, recordID)
|
|
}
|
|
|
|
func TestClient_AddRecord_error(t *testing.T) {
|
|
client := setupTest(t, "add-record-error.xml")
|
|
|
|
recordID, err := client.AddRecord(123, "_acme-challenge.example.com", "txtTXTtxt")
|
|
require.Error(t, err)
|
|
|
|
assert.Equal(t, 0, recordID)
|
|
}
|
|
|
|
func TestClient_AddRecord_system_error(t *testing.T) {
|
|
client := setupTest(t, "global-error.xml")
|
|
|
|
recordID, err := client.AddRecord(123, "_acme-challenge.example.com", "txtTXTtxt")
|
|
require.Error(t, err)
|
|
|
|
assert.Equal(t, 0, recordID)
|
|
}
|
|
|
|
func TestClient_DeleteRecord(t *testing.T) {
|
|
client := setupTest(t, "delete-record.xml")
|
|
|
|
recordID, err := client.DeleteRecord(4537)
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, 4537, recordID)
|
|
}
|
|
|
|
func TestClient_DeleteRecord_error(t *testing.T) {
|
|
client := setupTest(t, "delete-record-error.xml")
|
|
|
|
recordID, err := client.DeleteRecord(4537)
|
|
require.Error(t, err)
|
|
|
|
assert.Equal(t, 0, recordID)
|
|
}
|
|
|
|
func TestClient_DeleteRecord_system_error(t *testing.T) {
|
|
client := setupTest(t, "global-error.xml")
|
|
|
|
recordID, err := client.DeleteRecord(4537)
|
|
require.Error(t, err)
|
|
|
|
assert.Equal(t, 0, recordID)
|
|
}
|