X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=internal%2Fpublishservice%2Fpublishservice_test.go;fp=internal%2Fpublishservice%2Fpublishservice_test.go;h=b5f601d9b203db80aacbd633220e7e42130b11bf;hb=3d37436267da1867cf4cf9a4e39a6b0da2271b38;hp=0000000000000000000000000000000000000000;hpb=96d4aabf009d42ed479fe49aad416495201f2b2c;p=nonrtric%2Fplt%2Fsme.git

diff --git a/internal/publishservice/publishservice_test.go b/internal/publishservice/publishservice_test.go
new file mode 100644
index 0000000..b5f601d
--- /dev/null
+++ b/internal/publishservice/publishservice_test.go
@@ -0,0 +1,146 @@
+// -
+//   ========================LICENSE_START=================================
+//   O-RAN-SC
+//   %%
+//   Copyright (C) 2022: Nordix Foundation
+//   %%
+//   Licensed 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.
+//   ========================LICENSE_END===================================
+//
+
+package publishservice
+
+import (
+	"fmt"
+	"net/http"
+	"os"
+	"testing"
+
+	"oransc.org/nonrtric/sme/internal/providermanagement"
+
+	"github.com/labstack/echo/v4"
+
+	publishapi "oransc.org/nonrtric/sme/internal/publishserviceapi"
+
+	"oransc.org/nonrtric/sme/internal/helmmanagement"
+	helmMocks "oransc.org/nonrtric/sme/internal/helmmanagement/mocks"
+	serviceMocks "oransc.org/nonrtric/sme/internal/providermanagement/mocks"
+
+	"github.com/deepmap/oapi-codegen/pkg/middleware"
+	"github.com/deepmap/oapi-codegen/pkg/testutil"
+	echomiddleware "github.com/labstack/echo/v4/middleware"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/mock"
+)
+
+func TestPublishUnpublishService(t *testing.T) {
+	aefId := "aefId"
+	newApiId := "api_id_app-management"
+	serviceRegisterMock := serviceMocks.ServiceRegister{}
+	serviceRegisterMock.On("IsFunctionRegistered", aefId).Return(true)
+	helmManagerMock := helmMocks.HelmManager{}
+	helmManagerMock.On("InstallHelmChart", mock.Anything, mock.Anything, mock.Anything, mock.Anything).Return(nil)
+	serviceUnderTest, requestHandler := getEcho(&serviceRegisterMock, &helmManagerMock)
+
+	// Check no services published
+	result := testutil.NewRequest().Get("/aefId/service-apis/"+newApiId).Go(t, requestHandler)
+
+	assert.Equal(t, http.StatusNotFound, result.Code())
+
+	domainName := "domain"
+	var protocol publishapi.Protocol = "HTTP_1_1"
+	description := "Description,namespace,repoName,chartName,releaseName"
+	newServiceDescription := publishapi.ServiceAPIDescription{
+		AefProfiles: &[]publishapi.AefProfile{
+			{
+				AefId:      aefId,
+				DomainName: &domainName,
+				Protocol:   &protocol,
+				Versions: []publishapi.Version{
+					{
+						ApiVersion: "v1",
+						Resources: &[]publishapi.Resource{
+							{
+								CommType: "REQUEST_RESPONSE",
+								Operations: &[]publishapi.Operation{
+									"POST",
+								},
+								ResourceName: "app",
+								Uri:          "app",
+							},
+						},
+					},
+				},
+			},
+		},
+		ApiName:     "app-management",
+		Description: &description,
+	}
+
+	// Publish a service
+	result = testutil.NewRequest().Post("/aefId/service-apis").WithJsonBody(newServiceDescription).Go(t, requestHandler)
+
+	assert.Equal(t, http.StatusCreated, result.Code())
+	var resultService publishapi.ServiceAPIDescription
+	err := result.UnmarshalBodyToObject(&resultService)
+	assert.NoError(t, err, "error unmarshaling response")
+	assert.Equal(t, *resultService.ApiId, newApiId)
+	assert.Equal(t, "http://example.com/"+aefId+"/service-apis/"+*resultService.ApiId, result.Recorder.Header().Get(echo.HeaderLocation))
+	newServiceDescription.ApiId = &newApiId
+	wantedAPILIst := []publishapi.ServiceAPIDescription{newServiceDescription}
+	assert.True(t, serviceUnderTest.AreAPIsRegistered(&wantedAPILIst))
+	assert.True(t, serviceUnderTest.IsAPIRegistered("aefId", "app-management"))
+	serviceRegisterMock.AssertCalled(t, "IsFunctionRegistered", aefId)
+	helmManagerMock.AssertCalled(t, "InstallHelmChart", "namespace", "repoName", "chartName", "releaseName")
+	assert.ElementsMatch(t, wantedAPILIst, *serviceUnderTest.GetAPIs())
+
+	// Check that service is published
+	result = testutil.NewRequest().Get("/aefId/service-apis/"+newApiId).Go(t, requestHandler)
+
+	assert.Equal(t, http.StatusOK, result.Code())
+	err = result.UnmarshalBodyToObject(&resultService)
+	assert.NoError(t, err, "error unmarshaling response")
+	assert.Equal(t, *resultService.ApiId, newApiId)
+
+	// Delete a service
+	helmManagerMock.On("UninstallHelmChart", mock.Anything, mock.Anything).Return(nil)
+	result = testutil.NewRequest().Delete("/aefId/service-apis/"+newApiId).Go(t, requestHandler)
+
+	assert.Equal(t, http.StatusNoContent, result.Code())
+	helmManagerMock.AssertCalled(t, "UninstallHelmChart", "namespace", "chartName")
+	assert.Empty(t, *serviceUnderTest.GetAPIs())
+
+	// Check no services published
+	result = testutil.NewRequest().Get("/aefId/service-apis/"+newApiId).Go(t, requestHandler)
+
+	assert.Equal(t, http.StatusNotFound, result.Code())
+}
+
+func getEcho(serviceRegister providermanagement.ServiceRegister, helmManager helmmanagement.HelmManager) (*PublishService, *echo.Echo) {
+	swagger, err := publishapi.GetSwagger()
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error loading swagger spec\n: %s", err)
+		os.Exit(1)
+	}
+
+	swagger.Servers = nil
+
+	ps := NewPublishService(serviceRegister, helmManager)
+
+	e := echo.New()
+	e.Use(echomiddleware.Logger())
+	e.Use(middleware.OapiRequestValidator(swagger))
+
+	publishapi.RegisterHandlers(e, ps)
+	return ps, e
+}