account_test.go

package integration

import (
	"fmt"
	"net/http"
	"net/http/httptest"
	"net/http/httputil"
	"os/exec"
	"strings"
	"testing"
	"time"

	"github.com/sclevine/spec"
	"github.com/stretchr/testify/require"
)

var _ = suite("account/get", func(t *testing.T, when spec.G, it spec.S) {
	var (
		expect *require.Assertions
		server *httptest.Server
	)

	it.Before(func() {
		expect = require.New(t)

		server = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
			w.Header().Add("content-type", "application/json")

			switch req.URL.Path {
			case "/v2/account":
				auth := req.Header.Get("Authorization")
				if auth == "Bearer user-context-token" {
					w.Write([]byte(userContextGetResponse))
					return
				}
				if auth != "Bearer some-magic-token" {
					w.WriteHeader(http.StatusUnauthorized)
					return
				}

				if req.Method != http.MethodGet {
					w.WriteHeader(http.StatusMethodNotAllowed)
					return
				}

				w.Write([]byte(accountGetResponse))
			default:
				dump, err := httputil.DumpRequest(req, true)
				if err != nil {
					t.Fatal("failed to dump request")
				}

				t.Fatalf("received unknown request: %s", dump)
			}
		}))
	})

	it("returns the details of my account", func() {
		cmd := exec.Command(builtBinaryPath,
			"-t", "some-magic-token",
			"-u", server.URL,
			"account",
			"get",
		)

		output, err := cmd.CombinedOutput()
		expect.NoError(err)

		expect.Equal(strings.TrimSpace(accountOutput), strings.TrimSpace(string(output)))
	})

	when("format flags are passed", func() {
		it("only displays the correct fields", func() {
			cmd := exec.Command(builtBinaryPath,
				"-t", "some-magic-token",
				"-u", server.URL,
				"account",
				"get",
				"--format", "Email,UUID,TeamUUID",
			)

			output, err := cmd.CombinedOutput()
			expect.NoError(err, string(output))

			expect.Equal(strings.TrimSpace(formattedAccountOutput), strings.TrimSpace(string(output)))
		})
	})

	it("doesn't panic in a user context", func() {
		cmd := exec.Command(builtBinaryPath,
			"-t", "user-context-token",
			"-u", server.URL,
			"account",
			"get",
		)

		output, err := cmd.CombinedOutput()
		expect.NoError(err)

		expect.Equal(strings.TrimSpace(userContextOutput), strings.TrimSpace(string(output)))
	})
})

var _ = suite("account/ratelimit", func(t *testing.T, when spec.G, it spec.S) {
	var (
		expect *require.Assertions
		server *httptest.Server
	)

	it.Before(func() {
		expect = require.New(t)

		server = httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
			switch req.URL.Path {
			case "/v2/account":
				if req.Method != "GET" {
					w.WriteHeader(http.StatusMethodNotAllowed)
					return
				}

				auth := req.Header.Get("Authorization")
				if auth == "Bearer some-magic-token" {
					w.Header().Add("RateLimit-Limit", "200")
					w.Header().Add("RateLimit-Remaining", "199")
					w.Header().Add("RateLimit-Reset", "1565385881")

					w.Write([]byte(`{ "account":{}}`))
					return
				}

				if auth == "Bearer token-with-ratelimit-exhausted" {
					w.Header().Add("RateLimit-Limit", "200")
					w.Header().Add("RateLimit-Remaining", "0")
					w.Header().Add("RateLimit-Reset", "1565385881")
					w.WriteHeader(http.StatusTooManyRequests)

					w.Write([]byte(`{ "id":"too_many_requests", "message":"Too many requests"}`))
					return
				}

				w.WriteHeader(http.StatusUnauthorized)
				return
			default:
				dump, err := httputil.DumpRequest(req, true)
				if err != nil {
					t.Fatal("failed to dump request")
				}

				t.Fatalf("received unknown request: %s", dump)
			}
		}))
	})

	it("returns the current rate limits for my account", func() {
		cmd := exec.Command(builtBinaryPath,
			"-t", "some-magic-token",
			"-u", server.URL,
			"account",
			"ratelimit",
		)

		output, err := cmd.CombinedOutput()
		expect.NoError(err)

		t := time.Unix(1565385881, 0)
		expectedOutput := strings.TrimSpace(fmt.Sprintf(ratelimitOutput, t))
		expect.Equal(expectedOutput, strings.TrimSpace(string(output)))
	})

	it("doesn't return an error when rate-limited", func() {
		cmd := exec.Command(builtBinaryPath,
			"-t", "token-with-ratelimit-exhausted",
			"-u", server.URL,
			"account",
			"ratelimit",
		)

		output, err := cmd.CombinedOutput()
		expect.NoError(err, string(output))

		t := time.Unix(1565385881, 0)
		expectedOutput := strings.TrimSpace(fmt.Sprintf(ratelimitExhaustedOutput, t))
		expect.Equal(expectedOutput, strings.TrimSpace(string(output)))
	})

	it("doesn't retry when rate-limited", func() {
		cmd := exec.Command(builtBinaryPath,
			"-t", "token-with-ratelimit-exhausted",
			"-u", server.URL,
			"account",
			"ratelimit", "--trace",
		)

		output, err := cmd.CombinedOutput()
		expect.NoError(err, string(output))

		expect.NotContains(strings.TrimSpace(string(output)), "retrying in")
	})
})

const (
	accountGetResponse = `
{
  "account": {
    "droplet_limit": 25,
    "floating_ip_limit": 5,
    "email": "sammy@digitalocean.com",
    "uuid": "b6fr89dbf6d9156cace5f3c78dc9851d957381ef",
    "email_verified": true,
    "status": "active",
    "status_message": "",
    "team": {
      "uuid": "e8566708-f6fd-11ec-aac1-7f9bcd99de41",
      "name": "My Team"
    }
  }
}`

	userContextGetResponse = `
{
  "account": {
    "droplet_limit": 25,
    "floating_ip_limit": 5,
    "email": "sammy@digitalocean.com",
    "uuid": "b6fr89dbf6d9156cace5f3c78dc9851d957381ef",
    "email_verified": true,
    "status": "active",
    "status_message": ""
  }
}`

	accountOutput = `
User Email                Team       Droplet Limit    Email Verified    User UUID                                   Status
sammy@digitalocean.com    My Team    25               true              b6fr89dbf6d9156cace5f3c78dc9851d957381ef    active
`

	userContextOutput = `
User Email                Team     Droplet Limit    Email Verified    User UUID                                   Status
sammy@digitalocean.com    <nil>    25               true              b6fr89dbf6d9156cace5f3c78dc9851d957381ef    active
`

	formattedAccountOutput = `
User Email                User UUID                                   Team UUID
sammy@digitalocean.com    b6fr89dbf6d9156cace5f3c78dc9851d957381ef    e8566708-f6fd-11ec-aac1-7f9bcd99de41
`

	ratelimitOutput = `
Limit    Remaining    Reset
200      199          %s
`

	ratelimitExhaustedOutput = `
Limit    Remaining    Reset
200      0            %s
`
)