forked from TrueCloudLab/restic
87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
// Package testutils contains some test utilities for the Azure SDK
|
|
package testutils
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/Azure/azure-sdk-for-go/management"
|
|
)
|
|
|
|
// GetTestClient returns a management Client for testing. Expects
|
|
// AZSUBSCRIPTIONID and AZCERTDATA to be present in the environment. AZCERTDATA
|
|
// is the base64encoded binary representation of the PEM certificate data.
|
|
func GetTestClient(t *testing.T) management.Client {
|
|
subid := os.Getenv("AZSUBSCRIPTIONID")
|
|
certdata := os.Getenv("AZCERTDATA")
|
|
if subid == "" || certdata == "" {
|
|
t.Skip("AZSUBSCRIPTIONID or AZCERTDATA not set, skipping test")
|
|
}
|
|
cert, err := base64.StdEncoding.DecodeString(certdata)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
client, err := management.NewClient(subid, cert)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return testClient{client, t}
|
|
}
|
|
|
|
type testClient struct {
|
|
management.Client
|
|
t *testing.T
|
|
}
|
|
|
|
func chop(d []byte) string {
|
|
const maxlen = 5000
|
|
|
|
s := string(d)
|
|
|
|
if len(s) > maxlen {
|
|
return s[:maxlen] + "..."
|
|
}
|
|
return s
|
|
}
|
|
|
|
func (l testClient) SendAzureGetRequest(url string) ([]byte, error) {
|
|
d, err := l.Client.SendAzureGetRequest(url)
|
|
logOperation(l.t, "GET", url, nil, d, "", err)
|
|
return d, err
|
|
}
|
|
|
|
func (l testClient) SendAzurePostRequest(url string, data []byte) (management.OperationID, error) {
|
|
oid, err := l.Client.SendAzurePostRequest(url, data)
|
|
logOperation(l.t, "POST", url, data, nil, oid, err)
|
|
return oid, err
|
|
}
|
|
|
|
func (l testClient) SendAzurePutRequest(url string, contentType string, data []byte) (management.OperationID, error) {
|
|
oid, err := l.Client.SendAzurePutRequest(url, contentType, data)
|
|
logOperation(l.t, "PUT", url, data, nil, oid, err)
|
|
return oid, err
|
|
}
|
|
|
|
func (l testClient) SendAzureDeleteRequest(url string) (management.OperationID, error) {
|
|
oid, err := l.Client.SendAzureDeleteRequest(url)
|
|
logOperation(l.t, "DELETE", url, nil, nil, oid, err)
|
|
return oid, err
|
|
}
|
|
|
|
func logOperation(t *testing.T, method, url string, requestData, responseData []byte, oid management.OperationID, err error) {
|
|
t.Logf("AZURE> %s %s\n", method, url)
|
|
if requestData != nil {
|
|
t.Logf(" >>> %s\n", chop(requestData))
|
|
}
|
|
if err != nil {
|
|
t.Logf(" <<< ERROR: %+v\n", err)
|
|
} else {
|
|
if responseData != nil {
|
|
t.Logf(" <<< %s\n", chop(responseData))
|
|
} else {
|
|
t.Logf(" <<< OperationID: %s\n", oid)
|
|
}
|
|
}
|
|
}
|