81 lines
1.4 KiB
Go
81 lines
1.4 KiB
Go
package godo
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
var testOneClick = &OneClick{
|
|
Slug: "test-slug",
|
|
Type: "droplet",
|
|
}
|
|
|
|
var testOneClickJSON = `
|
|
{
|
|
"slug":"test-slug",
|
|
"type":"droplet"
|
|
}
|
|
`
|
|
var testMessage = &InstallKubernetesAppsResponse{
|
|
Message: "test message",
|
|
}
|
|
|
|
var testMessageJSON = `
|
|
{
|
|
"message" : "test message"
|
|
}
|
|
`
|
|
|
|
var kubernetesPayload = &InstallKubernetesAppsRequest{
|
|
ClusterUUID: "123",
|
|
Slugs: []string{"slug1", "slug2"},
|
|
}
|
|
|
|
func TestOneClick_List(t *testing.T) {
|
|
setup()
|
|
defer teardown()
|
|
|
|
svc := client.OneClick
|
|
path := "/v2/1-clicks"
|
|
want := []*OneClick{
|
|
testOneClick,
|
|
}
|
|
|
|
jsonBlob := `
|
|
{
|
|
"1_clicks": [
|
|
` + testOneClickJSON + `
|
|
]
|
|
}
|
|
`
|
|
mux.HandleFunc(path, func(w http.ResponseWriter, r *http.Request) {
|
|
testMethod(t, r, http.MethodGet)
|
|
fmt.Fprint(w, jsonBlob)
|
|
})
|
|
|
|
got, _, err := svc.List(ctx, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, want, got)
|
|
}
|
|
|
|
func TestOneClick_InstallKubernetes(t *testing.T) {
|
|
setup()
|
|
defer teardown()
|
|
|
|
svc := client.OneClick
|
|
path := "/v2/1-clicks/kubernetes"
|
|
|
|
mux.HandleFunc(path, func(w http.ResponseWriter, r *http.Request) {
|
|
testMethod(t, r, http.MethodPost)
|
|
fmt.Fprint(w, testMessageJSON)
|
|
})
|
|
|
|
got, _, err := svc.InstallKubernetes(ctx, kubernetesPayload)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, testMessage, got)
|
|
}
|