1
0
mirror of https://github.com/go-acme/lego.git synced 2024-12-14 14:10:05 +02:00
lego/providers/dns/internetbs/internal/client_test.go
2021-06-14 02:44:13 +02:00

259 lines
5.6 KiB
Go

package internal
import (
"fmt"
"io"
"net/http"
"net/http/httptest"
"net/url"
"os"
"strconv"
"testing"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
const testBaseURL = "https://testapi.internet.bs"
const (
testAPIKey = "testapi"
testPassword = "testpass"
)
func TestClient_AddRecord(t *testing.T) {
client := setupTest(t, "/Domain/DnsRecord/Add", "./fixtures/Domain_DnsRecord_Add_SUCCESS.json")
query := RecordQuery{
FullRecordName: "www.example.com",
Type: "TXT",
Value: "xxx",
TTL: 36000,
}
err := client.AddRecord(query)
require.NoError(t, err)
}
func TestClient_AddRecord_error(t *testing.T) {
client := setupTest(t, "/Domain/DnsRecord/Add", "./fixtures/Domain_DnsRecord_Add_FAILURE.json")
query := RecordQuery{
FullRecordName: "www.example.com.",
Type: "TXT",
Value: "xxx",
TTL: 36000,
}
err := client.AddRecord(query)
require.Error(t, err)
}
func TestClient_AddRecord_integration(t *testing.T) {
env, ok := os.LookupEnv("INTERNET_BS_DEBUG")
if !ok {
t.Skip("skip integration test")
}
client := NewClient(testAPIKey, testPassword)
client.baseURL, _ = url.Parse(testBaseURL)
client.debug, _ = strconv.ParseBool(env)
query := RecordQuery{
FullRecordName: "www.example.com",
Type: "TXT",
Value: "xxx",
TTL: 36000,
}
err := client.AddRecord(query)
require.NoError(t, err)
query = RecordQuery{
FullRecordName: "www.example.com",
Type: "TXT",
Value: "yyy",
TTL: 36000,
}
err = client.AddRecord(query)
require.NoError(t, err)
}
func TestClient_RemoveRecord(t *testing.T) {
client := setupTest(t, "/Domain/DnsRecord/Remove", "./fixtures/Domain_DnsRecord_Remove_SUCCESS.json")
query := RecordQuery{
FullRecordName: "www.example.com",
Type: "TXT",
Value: "",
}
err := client.RemoveRecord(query)
require.NoError(t, err)
}
func TestClient_RemoveRecord_error(t *testing.T) {
client := setupTest(t, "/Domain/DnsRecord/Remove", "./fixtures/Domain_DnsRecord_Remove_FAILURE.json")
query := RecordQuery{
FullRecordName: "www.example.com.",
Type: "TXT",
Value: "",
}
err := client.RemoveRecord(query)
require.Error(t, err)
}
func TestClient_RemoveRecord_integration(t *testing.T) {
env, ok := os.LookupEnv("INTERNET_BS_DEBUG")
if !ok {
t.Skip("skip integration test")
}
client := NewClient(testAPIKey, testPassword)
client.baseURL, _ = url.Parse(testBaseURL)
client.debug, _ = strconv.ParseBool(env)
query := RecordQuery{
FullRecordName: "www.example.com",
Type: "TXT",
Value: "",
}
err := client.RemoveRecord(query)
require.NoError(t, err)
}
func TestClient_ListRecords(t *testing.T) {
client := setupTest(t, "/Domain/DnsRecord/List", "./fixtures/Domain_DnsRecord_List_SUCCESS.json")
query := ListRecordQuery{
Domain: "example.com",
}
records, err := client.ListRecords(query)
require.NoError(t, err)
expected := []Record{
{
Name: "example.com",
Value: "ns-hongkong.internet.bs",
TTL: 3600,
Type: "NS",
},
{
Name: "example.com",
Value: "ns-toronto.internet.bs",
TTL: 3600,
Type: "NS",
},
{
Name: "example.com",
Value: "ns-london.internet.bs",
TTL: 3600,
Type: "NS",
},
{
Name: "test.example.com",
Value: "example1.com",
TTL: 3600,
Type: "CNAME",
},
{
Name: "www.example.com",
Value: "xxx",
TTL: 36000,
Type: "TXT",
},
{
Name: "www.example.com",
Value: "yyy",
TTL: 36000,
Type: "TXT",
},
}
assert.Equal(t, expected, records)
}
func TestClient_ListRecords_error(t *testing.T) {
client := setupTest(t, "/Domain/DnsRecord/List", "./fixtures/Domain_DnsRecord_List_FAILURE.json")
query := ListRecordQuery{
Domain: "www.example.com",
}
_, err := client.ListRecords(query)
require.Error(t, err)
}
func TestClient_ListRecords_integration(t *testing.T) {
env, ok := os.LookupEnv("INTERNET_BS_DEBUG")
if !ok {
t.Skip("skip integration test")
}
client := NewClient(testAPIKey, testPassword)
client.baseURL, _ = url.Parse(testBaseURL)
client.debug, _ = strconv.ParseBool(env)
query := ListRecordQuery{
Domain: "example.com",
}
records, err := client.ListRecords(query)
require.NoError(t, err)
for _, record := range records {
fmt.Println(record)
}
}
func setupTest(t *testing.T, path, filename string) *Client {
t.Helper()
mux := http.NewServeMux()
server := httptest.NewServer(mux)
t.Cleanup(server.Close)
mux.HandleFunc(path, testHandler(filename))
client := NewClient(testAPIKey, testPassword)
client.baseURL, _ = url.Parse(server.URL)
return client
}
func testHandler(filename string) http.HandlerFunc {
return 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
}
if req.FormValue("apiKey") != testAPIKey {
http.Error(rw, `{"transactid":"d46d812569acdb8b39c3933ec4351e79","status":"FAILURE","message":"Invalid API key and\/or Password","code":107002}`, http.StatusOK)
return
}
if req.FormValue("password") != testPassword {
http.Error(rw, `{"transactid":"d46d812569acdb8b39c3933ec4351e79","status":"FAILURE","message":"Invalid API key and\/or Password","code":107002}`, http.StatusOK)
return
}
file, err := os.Open(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
}
}
}