porter/server/api/integration_handler_test.go

package api_test

import (
	"encoding/json"
	"net/http"
	"strings"
	"testing"

	"github.com/go-test/deep"
	"github.com/porter-dev/porter/internal/forms"
	ints "github.com/porter-dev/porter/internal/models/integrations"
)

// ------------------------- TEST TYPES AND MAIN LOOP ------------------------- //

type publicIntTest struct {
	initializers []func(t *tester)
	msg          string
	method       string
	endpoint     string
	body         string
	expStatus    int
	expBody      string
	useCookie    bool
	validators   []func(c *publicIntTest, tester *tester, t *testing.T)
}

func testPublicIntegrationRequests(t *testing.T, tests []*publicIntTest, canQuery bool) {
	for _, c := range tests {
		// create a new tester
		tester := newTester(canQuery)

		// if there's an initializer, call it
		for _, init := range c.initializers {
			init(tester)
		}

		req, err := http.NewRequest(
			c.method,
			c.endpoint,
			strings.NewReader(c.body),
		)

		tester.req = req

		if c.useCookie {
			req.AddCookie(tester.cookie)
		}

		if err != nil {
			t.Fatal(err)
		}

		tester.execute()
		rr := tester.rr

		// first, check that the status matches
		if status := rr.Code; status != c.expStatus {
			t.Errorf("%s, handler returned wrong status code: got %v want %v",
				c.msg, status, c.expStatus)
		}

		// if there's a validator, call it
		for _, validate := range c.validators {
			validate(c, tester, t)
		}
	}
}

// ------------------------- TEST FIXTURES AND FUNCTIONS  ------------------------- //

var listClusterIntegrationsTests = []*publicIntTest{
	&publicIntTest{
		initializers: []func(t *tester){
			initUserDefault,
		},
		msg:       "List cluster integrations",
		method:    "GET",
		endpoint:  "/api/integrations/cluster",
		body:      ``,
		expStatus: http.StatusOK,
		expBody:   `[{"auth_mechanism":"gcp","category":"cluster","service":"gke"},{"auth_mechanism":"aws","category":"cluster","service":"eks"},{"auth_mechanism":"kube","category":"cluster","service":"kube"}]`,
		useCookie: true,
		validators: []func(c *publicIntTest, tester *tester, t *testing.T){
			publicIntBodyValidator,
		},
	},
}

func TestHandleListClusterIntegrations(t *testing.T) {
	testPublicIntegrationRequests(t, listClusterIntegrationsTests, true)
}

var listRegistryIntegrationsTests = []*publicIntTest{
	&publicIntTest{
		initializers: []func(t *tester){
			initUserDefault,
		},
		msg:       "List registry integrations",
		method:    "GET",
		endpoint:  "/api/integrations/registry",
		body:      ``,
		expStatus: http.StatusOK,
		expBody:   `[{"auth_mechanism":"gcp","category":"registry","service":"gcr"},{"auth_mechanism":"aws","category":"registry","service":"ecr"},{"auth_mechanism":"oauth","category":"registry","service":"docker"}]`,
		useCookie: true,
		validators: []func(c *publicIntTest, tester *tester, t *testing.T){
			publicIntBodyValidator,
		},
	},
}

func TestHandleListRegistryIntegrations(t *testing.T) {
	testPublicIntegrationRequests(t, listRegistryIntegrationsTests, true)
}

var listHelmRepoIntegrationsTest = []*publicIntTest{
	&publicIntTest{
		initializers: []func(t *tester){
			initUserDefault,
		},
		msg:       "List Helm repo integrations",
		method:    "GET",
		endpoint:  "/api/integrations/helm",
		body:      ``,
		expStatus: http.StatusOK,
		expBody:   `[{"auth_mechanism":"basic","category":"helm","service":"helm"},{"auth_mechanism":"gcp","category":"helm","service":"gcs"},{"auth_mechanism":"aws","category":"helm","service":"s3"}]`,
		useCookie: true,
		validators: []func(c *publicIntTest, tester *tester, t *testing.T){
			publicIntBodyValidator,
		},
	},
}

func TestHandleListHelmRepoIntegrations(t *testing.T) {
	testPublicIntegrationRequests(t, listHelmRepoIntegrationsTest, true)
}

var listRepoIntegrationsTests = []*publicIntTest{
	&publicIntTest{
		initializers: []func(t *tester){
			initUserDefault,
		},
		msg:       "List repo integrations",
		method:    "GET",
		endpoint:  "/api/integrations/repo",
		body:      ``,
		expStatus: http.StatusOK,
		expBody:   `[{"auth_mechanism":"oauth","category":"repo","service":"github"}]`,
		useCookie: true,
		validators: []func(c *publicIntTest, tester *tester, t *testing.T){
			publicIntBodyValidator,
		},
	},
}

func TestHandleListRepoIntegrations(t *testing.T) {
	testPublicIntegrationRequests(t, listRepoIntegrationsTests, true)
}

var createGCPIntegrationTests = []*publicIntTest{
	&publicIntTest{
		initializers: []func(t *tester){
			initUserDefault,
			initProject,
		},
		msg:      "Create GCP Integration",
		method:   "POST",
		endpoint: "/api/projects/1/integrations/gcp",
		body: `{
			"gcp_key_data": "yoooo"
		}`,
		expStatus: http.StatusCreated,
		expBody:   `{"id":1,"user_id":1,"project_id":1,"gcp-project-id":"","gcp-user-email":""}`,
		useCookie: true,
		validators: []func(c *publicIntTest, tester *tester, t *testing.T){
			gcpIntBodyValidator,
		},
	},
}

func TestHandleCreateGCPIntegration(t *testing.T) {
	testPublicIntegrationRequests(t, createGCPIntegrationTests, true)
}

var createAWSIntegrationTests = []*publicIntTest{
	&publicIntTest{
		initializers: []func(t *tester){
			initUserDefault,
			initProject,
		},
		msg:      "Create AWS Integration",
		method:   "POST",
		endpoint: "/api/projects/1/integrations/aws",
		body: `{
			"aws_cluster_id": "cluster-id-0",
			"aws_access_key_id": "accesskey",
			"aws_secret_access_key": "secretkey"
		}`,
		expStatus: http.StatusCreated,
		expBody:   `{"id":1,"user_id":1,"project_id":1,"aws-entity-id":"","aws-caller-id":""}`,
		useCookie: true,
		validators: []func(c *publicIntTest, tester *tester, t *testing.T){
			awsIntBodyValidator,
		},
	},
}

func TestHandleCreateAWSIntegration(t *testing.T) {
	testPublicIntegrationRequests(t, createGCPIntegrationTests, true)
}

var createBasicIntegrationTests = []*publicIntTest{
	&publicIntTest{
		initializers: []func(t *tester){
			initUserDefault,
			initProject,
		},
		msg:      "Create basic integration",
		method:   "POST",
		endpoint: "/api/projects/1/integrations/basic",
		body: `{
			"username": "username",
			"password": "password"
		}`,
		expStatus: http.StatusCreated,
		expBody:   `{"id":1,"user_id":1,"project_id":1}`,
		useCookie: true,
		validators: []func(c *publicIntTest, tester *tester, t *testing.T){
			basicIntBodyValidator,
		},
	},
}

func TestHandleCreateBasicIntegration(t *testing.T) {
	testPublicIntegrationRequests(t, createBasicIntegrationTests, true)
}

// ------------------------- INITIALIZERS AND VALIDATORS ------------------------- //

func initAWSIntegration(tester *tester) {
	proj, _ := tester.repo.Project.ReadProject(1)

	form := &forms.CreateAWSIntegrationForm{
		ProjectID: proj.ID,
		UserID:    1,
	}

	// convert the form to a ServiceAccountCandidate
	awsInt, _ := form.ToAWSIntegration()

	tester.repo.AWSIntegration.CreateAWSIntegration(awsInt)
}

func initBasicIntegration(tester *tester) {
	proj, _ := tester.repo.Project.ReadProject(1)

	basicInt := &ints.BasicIntegration{
		ProjectID: proj.ID,
		UserID:    1,
	}

	tester.repo.BasicIntegration.CreateBasicIntegration(basicInt)
}

func publicIntBodyValidator(c *publicIntTest, tester *tester, t *testing.T) {
	gotBody := make([]*ints.PorterIntegration, 0)
	expBody := make([]*ints.PorterIntegration, 0)

	bytes := tester.rr.Body.Bytes()

	json.Unmarshal(bytes, &gotBody)
	json.Unmarshal([]byte(c.expBody), &expBody)

	if diff := deep.Equal(gotBody, expBody); diff != nil {
		t.Errorf("handler returned wrong body:\n")
		t.Error(diff)
	}
}

func gcpIntBodyValidator(c *publicIntTest, tester *tester, t *testing.T) {
	gotBody := &ints.GCPIntegration{}
	expBody := &ints.GCPIntegration{}

	json.Unmarshal(tester.rr.Body.Bytes(), &gotBody)
	json.Unmarshal([]byte(c.expBody), &expBody)

	if diff := deep.Equal(gotBody, expBody); diff != nil {
		t.Errorf("handler returned wrong body:\n")
		t.Error(diff)
	}
}

func awsIntBodyValidator(c *publicIntTest, tester *tester, t *testing.T) {
	gotBody := &ints.AWSIntegration{}
	expBody := &ints.AWSIntegration{}

	json.Unmarshal(tester.rr.Body.Bytes(), &gotBody)
	json.Unmarshal([]byte(c.expBody), &expBody)

	if diff := deep.Equal(gotBody, expBody); diff != nil {
		t.Errorf("handler returned wrong body:\n")
		t.Error(diff)
	}
}

func basicIntBodyValidator(c *publicIntTest, tester *tester, t *testing.T) {
	gotBody := &ints.BasicIntegration{}
	expBody := &ints.BasicIntegration{}

	bytes := tester.rr.Body.Bytes()

	json.Unmarshal(bytes, &gotBody)
	json.Unmarshal([]byte(c.expBody), &expBody)

	if diff := deep.Equal(gotBody, expBody); diff != nil {
		t.Errorf("handler returned wrong body:\n")
		t.Error(diff)
	}
}