github.com/Axway/agent-sdk@v1.1.101/pkg/apic/servicebuilder_test.go

package apic

import (
	"testing"

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

const longDescription = `This is a sample Pet Store Server based on the OpenAPI 3.0 specification.  You can find out more about
Swagger at [http://swagger.io](http://swagger.io). In the third iteration of the pet store, we've switched to the design first approach!
You can now help us improve the API whether it's by making changes to the definition itself or to the code.
That way, with time, we can improve the API in general, and expose some of the new features in OAS3.

Some useful links:
- [The Pet Store repository](https://github.com/swagger-api/swagger-petstore)
- [The source API definition for the Pet Store](https://github.com/swagger-api/swagger-petstore/blob/master/src/main/resources/openapi.yaml)`

func TestNewServiceBodyBuilder(t *testing.T) {
	svcBody, err := NewServiceBodyBuilder().Build()
	assert.NotNil(t, err) // service body does not build any spec for json or yaml, and no resource type
	assert.NotNil(t, svcBody)

	// test all the default values
	assert.Equal(t, Passthrough, svcBody.AuthPolicy)
	assert.Equal(t, PublishedState, svcBody.State)
	assert.Equal(t, PublishedStatus, svcBody.Status)
}

func TestServiceBodySetters(t *testing.T) {
	tags := map[string]interface{}{"tag1": "t1", "tag2": "t2"}
	attribs := map[string]string{"attrib1": "a1", "attrib2": "a2"}
	revAttr := map[string]string{"svc": "attr"}
	instAttr := map[string]string{"inst": "attr"}
	svcDetails := map[string]interface{}{"svc": "details"}
	instDetails := map[string]interface{}{"inst": "details"}
	revDetails := map[string]interface{}{"rev": "details"}
	ep := []EndpointDefinition{
		{
			Protocol: "https",
			Host:     "test.com",
			Port:     443,
			BasePath: "/test",
		},
	}

	serviceBuilder := NewServiceBodyBuilder()
	serviceBuilder = serviceBuilder.
		SetTitle("sbtitle").
		SetAPIName("testAPI").
		SetID("1234").
		SetPrimaryKey("PrimaryKey").
		SetRequestDefinitionsAllowed(true).
		SetStage("stageID").
		SetStageDisplayName("teststage").
		SetURL("https://1234.execute-api.us-region.amazonaws.com/teststage").
		SetDescription(longDescription).
		SetVersion("1.0.0").
		SetAuthPolicy("Oauth").
		SetAPISpec([]byte{}).
		SetDocumentation([]byte("documentation")).
		SetAPIUpdateSeverity("MAJOR").
		SetStatus(UnpublishedStatus).
		SetState(PublishedStatus).
		SetSubscriptionName("testsubscription").
		AddServiceEndpoint("https", "test.com", 443, "/test").
		SetImage("image").
		SetImageContentType("image/jpeg").
		SetResourceType("foobar").
		SetTags(tags).
		SetServiceAttribute(attribs).
		SetRevisionAttribute(revAttr).
		SetInstanceAttribute(instAttr).
		SetUnstructuredContentType("application/zip").
		SetUnstructuredFilename("test.zip").
		SetUnstructuredLabel("Label").
		SetUnstructuredType("Type").
		SetTeamName("00000").
		SetServiceAgentDetails(svcDetails).
		SetInstanceAgentDetails(instDetails).
		SetRevisionAgentDetails(revDetails).
		SetReferenceServiceName("refSvc", "refEnv").
		SetReferenceInstanceName("refInstance", "refEnv").
		SetIgnoreSpecBasedCreds(true)

	sb, err := serviceBuilder.
		SetServiceEndpoints(ep).
		Build()

	assert.Nil(t, err)
	assert.NotNil(t, sb)
	assert.Equal(t, "sbtitle", sb.NameToPush)
	assert.Equal(t, "testAPI", sb.APIName)
	assert.Equal(t, "1234", sb.RestAPIID)
	assert.Equal(t, "PrimaryKey", sb.PrimaryKey)
	assert.Equal(t, "stageID", sb.Stage)
	assert.Equal(t, "teststage", sb.StageDisplayName)
	assert.Equal(t, "https://1234.execute-api.us-region.amazonaws.com/teststage", sb.URL)
	assert.Equal(t, 350, len(sb.Description))

	description := longDescription[0:maxDescriptionLength-len(strEllipsis)] + strEllipsis
	assert.Equal(t, description, sb.Description)
	assert.Equal(t, "1.0.0", sb.Version)
	assert.Equal(t, "Oauth", sb.AuthPolicy)
	assert.Equal(t, UnpublishedStatus, sb.Status)
	assert.Equal(t, PublishedStatus, sb.State)

	assert.Equal(t, []byte{}, sb.SpecDefinition)
	assert.Equal(t, []byte("documentation"), sb.Documentation)

	assert.Equal(t, "image/jpeg", sb.ImageContentType)
	assert.Equal(t, "image", sb.Image)
	assert.Equal(t, Unstructured, sb.ResourceType)

	assert.Equal(t, "MAJOR", sb.APIUpdateSeverity)
	assert.Len(t, sb.Tags, 2)
	assert.Equal(t, "t2", sb.Tags["tag2"])
	assert.Len(t, sb.ServiceAttributes, 2)
	assert.Equal(t, "a2", sb.ServiceAttributes["attrib2"])
	assert.Equal(t, "application/zip", sb.UnstructuredProps.ContentType)
	assert.Equal(t, "test.zip", sb.UnstructuredProps.Filename)
	assert.Equal(t, "Label", sb.UnstructuredProps.Label)
	assert.Equal(t, "Type", sb.UnstructuredProps.AssetType)
	assert.Equal(t, "00000", sb.TeamName)
	assert.Equal(t, ep, sb.Endpoints)
	assert.Equal(t, revAttr, sb.RevisionAttributes)
	assert.Equal(t, instAttr, sb.InstanceAttributes)
	assert.Equal(t, svcDetails, sb.ServiceAgentDetails)
	assert.Equal(t, instDetails, sb.InstanceAgentDetails)
	assert.Equal(t, revDetails, sb.RevisionAgentDetails)
	assert.Equal(t, Unidentified, sb.dataplaneType)
	assert.Equal(t, false, sb.isDesignDataplane)
	assert.Equal(t, "refEnv/refSvc", sb.GetReferencedServiceName())
	assert.Equal(t, "refEnv/refInstance", sb.GetReferenceInstanceName())
	assert.Equal(t, true, sb.ignoreSpecBasesCreds)

	sb, err = serviceBuilder.
		SetSourceDataplaneType(GitHub, true).
		Build()

	assert.Nil(t, err)
	assert.NotNil(t, sb)
	assert.Equal(t, GitHub, sb.dataplaneType)
	assert.Equal(t, true, sb.isDesignDataplane)

	// invalid service body path
	ep[0].BasePath = "#/basepath"
	sb, err = serviceBuilder.
		SetServiceEndpoints(ep).
		Build()

	assert.NotNil(t, err)
	assert.NotNil(t, sb)

	// valid service body with empty path
	ep[0].BasePath = ""
	sb, err = serviceBuilder.
		SetServiceEndpoints(ep).
		Build()

	assert.Nil(t, err)
	assert.NotNil(t, sb)
}

func TestServiceBodyWithParseError(t *testing.T) {
	serviceBuilder := NewServiceBodyBuilder()
	_, err := serviceBuilder.SetResourceType(Oas3).SetAPISpec([]byte("{\"test\":\"123\"}")).Build()
	assert.NotNil(t, err)

	_, err = serviceBuilder.SetResourceType(Oas2).SetAPISpec([]byte("{\"test\":\"123\"}")).Build()
	assert.NotNil(t, err)

	_, err = serviceBuilder.SetResourceType(Wsdl).SetAPISpec([]byte("{\"test\":\"123\"}")).Build()
	assert.NotNil(t, err)

	_, err = serviceBuilder.SetResourceType(Protobuf).SetAPISpec([]byte("{\"test\":\"123\"}")).Build()
	assert.NotNil(t, err)

	_, err = serviceBuilder.SetResourceType(AsyncAPI).SetAPISpec([]byte("{\"test\":\"123\"}")).Build()
	assert.NotNil(t, err)

	_, err = serviceBuilder.SetResourceType(Unstructured).SetAPISpec([]byte("{\"test\":\"123\"}")).Build()
	assert.Nil(t, err)
}