--- /dev/null
+// -
+// ========================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
+}