1-click_test.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)
}