Move capifcore code to separate folder

[nonrtric/plt/sme.git] / capifcore / internal / discoverservice / discoverservice_test.go

diff --git a/capifcore/internal/discoverservice/discoverservice_test.go b/capifcore/internal/discoverservice/discoverservice_test.go
new file mode 100644 (file)
index 0000000..5ac0041
--- /dev/null
+++ b/capifcore/internal/discoverservice/discoverservice_test.go
@@ -0,0 +1,144 @@
+// -
+//   ========================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 discoverservice
+
+import (
+       "fmt"
+       "net/http"
+       "os"
+       "testing"
+
+       "oransc.org/nonrtric/capifcore/internal/discoverserviceapi"
+
+       "oransc.org/nonrtric/capifcore/internal/publishservice"
+
+       "github.com/labstack/echo/v4"
+
+       publishapi "oransc.org/nonrtric/capifcore/internal/publishserviceapi"
+
+       "oransc.org/nonrtric/capifcore/internal/publishservice/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"
+)
+
+func TestGetAllServiceAPIs(t *testing.T) {
+       var err error
+
+       apiList := []publishapi.ServiceAPIDescription{
+               getAPI("apiName1", "v1"),
+               getAPI("apiName2", "v1"),
+       }
+       apiRegisterMock := mocks.APIRegister{}
+       apiRegisterMock.On("GetAPIs").Return(&apiList)
+       requestHandler := getEcho(&apiRegisterMock)
+
+       // Get all APIs, without any filter
+       result := testutil.NewRequest().Get("/allServiceAPIs?api-invoker-id=api_invoker_id").Go(t, requestHandler)
+
+       assert.Equal(t, http.StatusOK, result.Code())
+       var resultInvoker discoverserviceapi.DiscoveredAPIs
+       err = result.UnmarshalBodyToObject(&resultInvoker)
+       assert.NoError(t, err, "error unmarshaling response")
+       assert.Equal(t, 2, len(*resultInvoker.ServiceAPIDescriptions))
+       assert.Equal(t, "apiName1", (*resultInvoker.ServiceAPIDescriptions)[0].ApiName)
+       assert.Equal(t, "apiName2", (*resultInvoker.ServiceAPIDescriptions)[1].ApiName)
+       apiRegisterMock.AssertCalled(t, "GetAPIs")
+
+       // Get APIs with filter
+       result = testutil.NewRequest().Get("/allServiceAPIs?api-name=apiName1&api-version=v1&api-invoker-id=api_invoker_id").Go(t, requestHandler)
+
+       assert.Equal(t, http.StatusOK, result.Code())
+       err = result.UnmarshalBodyToObject(&resultInvoker)
+       assert.NoError(t, err, "error unmarshaling response")
+       assert.Equal(t, 1, len(*resultInvoker.ServiceAPIDescriptions))
+       assert.Equal(t, "apiName1", (*resultInvoker.ServiceAPIDescriptions)[0].ApiName)
+       apiRegisterMock.AssertCalled(t, "GetAPIs")
+}
+
+func getEcho(apiRegister publishservice.APIRegister) *echo.Echo {
+       swagger, err := discoverserviceapi.GetSwagger()
+       if err != nil {
+               fmt.Fprintf(os.Stderr, "Error loading swagger spec\n: %s", err)
+               os.Exit(1)
+       }
+
+       swagger.Servers = nil
+
+       ds := NewDiscoverService(apiRegister)
+
+       e := echo.New()
+       e.Use(echomiddleware.Logger())
+       e.Use(middleware.OapiRequestValidator(swagger))
+
+       discoverserviceapi.RegisterHandlers(e, ds)
+       return e
+}
+
+func getAPI(apiName, apiVersion string) publishapi.ServiceAPIDescription {
+       apiId := "apiId_" + apiName
+       aefId := "aefId"
+       description := "description"
+       domainName := "domain"
+       var protocol publishapi.Protocol = "HTTP_1_1"
+       return publishapi.ServiceAPIDescription{
+               ApiId:       &apiId,
+               ApiName:     apiName,
+               Description: &description,
+               AefProfiles: &[]publishapi.AefProfile{
+                       {
+                               AefId:      aefId,
+                               DomainName: &domainName,
+                               Protocol:   &protocol,
+                               Versions: []publishapi.Version{
+                                       {
+                                               ApiVersion: apiVersion,
+                                               Resources: &[]publishapi.Resource{
+                                                       {
+                                                               ResourceName: "app",
+                                                               CommType:     "REQUEST_RESPONSE",
+                                                               Uri:          "uri",
+                                                               Operations: &[]publishapi.Operation{
+                                                                       "POST",
+                                                               },
+                                                       },
+                                               },
+                                       },
+                                       {
+                                               ApiVersion: "v2",
+                                               Resources: &[]publishapi.Resource{
+                                                       {
+                                                               ResourceName: "app",
+                                                               CommType:     "REQUEST_RESPONSE",
+                                                               Uri:          "uri",
+                                                               Operations: &[]publishapi.Operation{
+                                                                       "POST",
+                                                               },
+                                                       },
+                                               },
+                                       },
+                               },
+                       },
+               },
+       }
+}