sessiontoken_test.go

package signalfx

import (
	"context"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/signalfx/signalfx-go/sessiontoken"

	"github.com/stretchr/testify/assert"
)

func TestCreateSessionToken(t *testing.T) {
	mux = http.NewServeMux()
	server = httptest.NewServer(mux)

	client, _ = NewClient("", APIUrl(server.URL))
	defer server.Close()

	mux.HandleFunc("/v2/session", verifyRequest(t, "POST", false, http.StatusOK, nil, "sessiontoken/create_success.json"))

	result, err := client.CreateSessionToken(context.Background(), &sessiontoken.CreateTokenRequest{
		Email:    "testemail@test.com",
		Password: "testpassword",
	})
	assert.NoError(t, err, "Unexpected error creating orgtoken")
	assert.Equal(t, "testemail@test.com", result.Email, "Email does not match")
	assert.Equal(t, "mytokenvalue", result.AccessToken, "Access token does not match")
}

func TestCreateBadCredentials(t *testing.T) {
	mux = http.NewServeMux()
	server = httptest.NewServer(mux)
	client, _ = NewClient("", APIUrl(server.URL))
	defer server.Close()

	mux.HandleFunc("/v2/session", verifyRequest(t, "POST", false, http.StatusBadRequest, nil, ""))

	result, err := client.CreateSessionToken(context.Background(), &sessiontoken.CreateTokenRequest{
		Email: "email",
	})
	assert.Error(t, err, "Should have gotten an error from a bad create")
	assert.Nil(t, result, "Should have a null token on bad create")
}

func TestDeleteSessionToken(t *testing.T) {
	teardown := setup()
	defer teardown()

	mux.HandleFunc("/v2/session", verifyRequest(t, "DELETE", false, http.StatusNoContent, nil, ""))

	err := client.DeleteSessionToken(context.Background(), TestToken)
	assert.NoError(t, err, "Unexpected error deleting token")
}

func TestDeleteMissingSessionToken(t *testing.T) {
	teardown := setup()
	defer teardown()

	mux.HandleFunc("/v2/session", verifyRequest(t, "DELETE", false, http.StatusNotFound, nil, ""))

	err := client.DeleteSessionToken(context.Background(), TestToken)
	assert.Error(t, err, "Should have gotten an error from a missing delete")
}