service_instances.go

// Copyright (C) 2016-Present Pivotal Software, Inc. All rights reserved.
// This program and the accompanying materials are made available under the terms of the under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
// http://www.apache.org/licenses/LICENSE-2.0
// Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.

package mockcfapi

import (
	"encoding/json"
	"fmt"

	"github.com/amansingh066/on-demand-service-broker/mockhttp"
	. "github.com/onsi/gomega"
)

type Operation string

const (
	Delete Operation = "delete"
)

type OperationState string

const (
	InProgress = "in progress"
	Succeeded  = "succeeded"
	Failed     = "failed"
)

type getServiceInstanceMock struct {
	*mockhttp.Handler
	instanceGUID string
	planId       string
}

func GetServiceInstance(instanceGUID string) *getServiceInstanceMock {
	return &getServiceInstanceMock{
		Handler:      mockhttp.NewMockedHttpRequest("GET", "/v2/service_instances/"+instanceGUID),
		instanceGUID: instanceGUID,
	}
}

func (m *getServiceInstanceMock) RespondsWithSucceeded() *mockhttp.Handler {
	body := fmt.Sprintf(instanceResponseBody, m.instanceGUID, "create", Succeeded)
	return m.RespondsOKWith(body)
}

func (m *getServiceInstanceMock) RespondsWithInProgress(operation Operation) *mockhttp.Handler {
	body := fmt.Sprintf(instanceResponseBody, m.instanceGUID, operation, InProgress)
	return m.RespondsOKWith(body)
}

func (m *getServiceInstanceMock) RespondsWithPlanURL(planId string, operation Operation, operationState OperationState) *mockhttp.Handler {
	body := fmt.Sprintf(instanceResponseBodyWithPlanUrl, m.instanceGUID, operation, operationState, planId)
	return m.RespondsOKWith(body)
}

func (m *getServiceInstanceMock) RespondsWithFailed(operation Operation) *mockhttp.Handler {
	body := fmt.Sprintf(instanceResponseBody, m.instanceGUID, operation, Failed)
	return m.RespondsOKWith(body)
}

func DeleteServiceInstance(instanceGUID string) *mockhttp.Handler {
	path := fmt.Sprintf("/v2/service_instances/%s?accepts_incomplete=true", instanceGUID)
	return mockhttp.NewMockedHttpRequest("DELETE", path)
}

type listServiceInstancesMock struct {
	*mockhttp.Handler
}

func ListServiceInstances(servicePlanGUID string) *listServiceInstancesMock {
	return &listServiceInstancesMock{

		mockhttp.NewMockedHttpRequest(
			"GET",
			"/v2/service_plans/"+servicePlanGUID+"/service_instances?results-per-page=100",
		),
	}
}

func ListServiceInstancesBySpace(servicePlanGUID, spaceGUID string) *listServiceInstancesMock {
	return &listServiceInstancesMock{

		mockhttp.NewMockedHttpRequest(
			"GET",
			"/v2/service_plans/"+servicePlanGUID+"/service_instances?results-per-page=100&q=space_guid:"+spaceGUID,
		),
	}
}

func ListServiceInstancesForPage(servicePlanGUID string, page int) *listServiceInstancesMock {
	return &listServiceInstancesMock{
		mockhttp.NewMockedHttpRequest(
			"GET",
			fmt.Sprintf(
				"/v2/service_plans/%s/service_instances?order-direction=asc&page=%d&results-per-page=100",
				servicePlanGUID,
				page),
		),
	}
}

func ListServiceInstancesBySpaceForPage(servicePlanGUID, spaceGUID string, page int) *listServiceInstancesMock {
	return &listServiceInstancesMock{
		mockhttp.NewMockedHttpRequest(
			"GET",
			fmt.Sprintf(
				"/v2/service_plans/%s/service_instances?order-direction=asc&page=%d&results-per-page=100&q=space_guid:%s",
				servicePlanGUID,
				page,
				spaceGUID,
			),
		),
	}
}

func (m *listServiceInstancesMock) RespondsWithNoServiceInstances() *mockhttp.Handler {
	return m.RespondsOKWith(`{
			"total_results": 1,
			"total_pages": 1,
			"prev_url": null,
			"next_url": null,
			"resources": []
		}`)
}

func (m *listServiceInstancesMock) RespondsWithServiceInstances(instanceIDs ...string) *mockhttp.Handler {
	return m.RespondsOKWith(listServiceInstancesResponse(instanceIDs...))
}

func (m *listServiceInstancesMock) RespondsWithPaginatedServiceInstances(
	planID string,
	page,
	resultsPerPage,
	totalPages int,
	instanceIDs ...string,
) *mockhttp.Handler {
	return m.RespondsOKWith(paginatedListServiceInstanceResponse(
		planID,
		page,
		resultsPerPage,
		totalPages,
		instanceIDs...,
	))
}

func listServiceInstancesResponse(instanceIDs ...string) string {
	var resources []serviceInstanceResource
	for _, instanceID := range instanceIDs {
		resources = append(resources, serviceInstanceResource{Metadata: serviceInstanceMetadata{GUID: instanceID}})
	}

	responseBytes, err := json.Marshal(serviceInstances{Resources: resources})
	Expect(err).NotTo(HaveOccurred())
	return string(responseBytes)
}

func paginatedListServiceInstanceResponse(
	planID string,
	page,
	resultsPerPage,
	totalPages int,
	instanceIDs ...string,
) string {
	Expect(len(instanceIDs)).To(BeNumerically("<=", resultsPerPage))
	var nextURL, prevURL string

	if page < totalPages {
		nextURL = fmt.Sprintf(
			"/v2/service_plans/%s/service_instances?order-direction=asc&page=%d&results-per-page=%d",
			planID,
			page+1,
			resultsPerPage,
		)
	}

	if page > 1 {
		prevURL = fmt.Sprintf(
			"/v2/service_plans/%s/service_instances?order-direction=asc&page=%d&results-per-page=%d",
			planID,
			page-1,
			resultsPerPage,
		)
	}

	var resources []serviceInstanceResource
	for _, instanceID := range instanceIDs {
		resources = append(resources, serviceInstanceResource{Metadata: serviceInstanceMetadata{GUID: instanceID}})
	}

	responseBytes, err := json.Marshal(serviceInstances{Resources: resources, NextURL: &nextURL, PrevURL: &prevURL})
	Expect(err).NotTo(HaveOccurred())

	return string(responseBytes)
}

type serviceInstances struct {
	NextURL   *string                   `json:"next_url"`
	PrevURL   *string                   `json:"prev_url"`
	Resources []serviceInstanceResource `json:"resources"`
}

type serviceInstanceResource struct {
	Metadata serviceInstanceMetadata `json:"metadata"`
}

type serviceInstanceMetadata struct {
	GUID string `json:"guid"`
}

const instanceResponseBody = `{
	"metadata": {
		"guid": "%s"
	},
	"entity": {
		"last_operation": {
			"type": "%s",
			"state": "%s"
		}
	}
}`

const instanceResponseBodyWithPlanUrl = `{
	"metadata": {
		"guid": "%s"
	},
	"entity": {
		"last_operation": {
			"type": "%s",
			"state": "%s"
		},
		"service_plan_url": "/v2/service_plans/%s"
	}
}`