+func TestPutTrustedInvokerSuccessfully(t *testing.T) {
+ invokerRegisterMock := invokermocks.InvokerRegister{}
+ invokerRegisterMock.On("IsInvokerRegistered", mock.AnythingOfType("string")).Return(true)
+ aefId := "aefId"
+ aefProfile := getAefProfile(aefId)
+ aefProfile.SecurityMethods = &[]publishserviceapi.SecurityMethod{
+ publishserviceapi.SecurityMethodPKI,
+ }
+ aefProfiles := []publishserviceapi.AefProfile{
+ aefProfile,
+ }
+ apiId := "apiId"
+ publishedServices := []publishserviceapi.ServiceAPIDescription{
+ {
+ ApiId: &apiId,
+ AefProfiles: &aefProfiles,
+ },
+ }
+ publishRegisterMock := publishmocks.PublishRegister{}
+ publishRegisterMock.On("GetAllPublishedServices").Return(publishedServices)
+
+ requestHandler, _ := getEcho(nil, &publishRegisterMock, &invokerRegisterMock, nil)
+
+ invokerId := "invokerId"
+ serviceSecurityUnderTest := getServiceSecurity(aefId, apiId)
+ serviceSecurityUnderTest.SecurityInfo[0].ApiId = &apiId
+
+ result := testutil.NewRequest().Put("/trustedInvokers/"+invokerId).WithJsonBody(serviceSecurityUnderTest).Go(t, requestHandler)
+
+ assert.Equal(t, http.StatusCreated, result.Code())
+ var resultResponse securityapi.ServiceSecurity
+ err := result.UnmarshalBodyToObject(&resultResponse)
+ assert.NoError(t, err, "error unmarshaling response")
+ assert.NotEmpty(t, resultResponse.NotificationDestination)
+
+ for _, security := range resultResponse.SecurityInfo {
+ assert.Equal(t, *security.ApiId, apiId)
+ assert.Equal(t, *security.SelSecurityMethod, publishserviceapi.SecurityMethodPKI)
+ }
+ invokerRegisterMock.AssertCalled(t, "IsInvokerRegistered", invokerId)
+}
+
+func TestPutTrustedInkoverNotRegistered(t *testing.T) {
+ invokerRegisterMock := invokermocks.InvokerRegister{}
+ invokerRegisterMock.On("IsInvokerRegistered", mock.AnythingOfType("string")).Return(false)
+
+ requestHandler, _ := getEcho(nil, nil, &invokerRegisterMock, nil)
+
+ invokerId := "invokerId"
+ serviceSecurityUnderTest := getServiceSecurity("aefId", "apiId")
+
+ result := testutil.NewRequest().Put("/trustedInvokers/"+invokerId).WithJsonBody(serviceSecurityUnderTest).Go(t, requestHandler)
+
+ badRequest := http.StatusBadRequest
+ assert.Equal(t, badRequest, result.Code())
+ var problemDetails common29122.ProblemDetails
+ err := result.UnmarshalBodyToObject(&problemDetails)
+ assert.NoError(t, err, "error unmarshaling response")
+ assert.Equal(t, &badRequest, problemDetails.Status)
+ assert.Contains(t, *problemDetails.Cause, "Invoker not registered")
+ invokerRegisterMock.AssertCalled(t, "IsInvokerRegistered", invokerId)
+}
+
+func TestPutTrustedInkoverInvalidInputServiceSecurity(t *testing.T) {
+ invokerRegisterMock := invokermocks.InvokerRegister{}
+ invokerRegisterMock.On("IsInvokerRegistered", mock.AnythingOfType("string")).Return(true)
+
+ requestHandler, _ := getEcho(nil, nil, &invokerRegisterMock, nil)
+
+ invokerId := "invokerId"
+ notificationUrl := "url"
+ serviceSecurityUnderTest := getServiceSecurity("aefId", "apiId")
+ serviceSecurityUnderTest.NotificationDestination = common29122.Uri(notificationUrl)
+
+ result := testutil.NewRequest().Put("/trustedInvokers/"+invokerId).WithJsonBody(serviceSecurityUnderTest).Go(t, requestHandler)
+
+ badRequest := http.StatusBadRequest
+ assert.Equal(t, badRequest, result.Code())
+ var problemDetails common29122.ProblemDetails
+ err := result.UnmarshalBodyToObject(&problemDetails)
+ assert.NoError(t, err, "error unmarshaling response")
+ assert.Equal(t, &badRequest, problemDetails.Status)
+ assert.Contains(t, *problemDetails.Cause, "ServiceSecurity has invalid notificationDestination")
+ invokerRegisterMock.AssertCalled(t, "IsInvokerRegistered", invokerId)
+}
+
+func TestPutTrustedInvokerInterfaceDetailsNotNil(t *testing.T) {
+ invokerRegisterMock := invokermocks.InvokerRegister{}
+ invokerRegisterMock.On("IsInvokerRegistered", mock.AnythingOfType("string")).Return(true)
+ aefId := "aefId"
+ aefProfile := getAefProfile(aefId)
+ aefProfile.SecurityMethods = &[]publishserviceapi.SecurityMethod{
+ publishserviceapi.SecurityMethodPKI,
+ }
+ aefProfiles := []publishserviceapi.AefProfile{
+ aefProfile,
+ }
+ apiId := "apiId"
+ publishedServices := []publishserviceapi.ServiceAPIDescription{
+ {
+ ApiId: &apiId,
+ AefProfiles: &aefProfiles,
+ },
+ }
+ publishRegisterMock := publishmocks.PublishRegister{}
+ publishRegisterMock.On("GetAllPublishedServices").Return(publishedServices)
+
+ requestHandler, _ := getEcho(nil, &publishRegisterMock, &invokerRegisterMock, nil)
+
+ invokerId := "invokerId"
+ serviceSecurityUnderTest := getServiceSecurity(aefId, apiId)
+ serviceSecurityUnderTest.SecurityInfo[0] = securityapi.SecurityInformation{
+ ApiId: &apiId,
+ PrefSecurityMethods: []publishserviceapi.SecurityMethod{
+ publishserviceapi.SecurityMethodOAUTH,
+ },
+ InterfaceDetails: &publishserviceapi.InterfaceDescription{
+ SecurityMethods: &[]publishserviceapi.SecurityMethod{
+ publishserviceapi.SecurityMethodPSK,
+ },
+ },
+ }
+
+ result := testutil.NewRequest().Put("/trustedInvokers/"+invokerId).WithJsonBody(serviceSecurityUnderTest).Go(t, requestHandler)
+
+ assert.Equal(t, http.StatusCreated, result.Code())
+ var resultResponse securityapi.ServiceSecurity
+ err := result.UnmarshalBodyToObject(&resultResponse)
+ assert.NoError(t, err, "error unmarshaling response")
+ assert.NotEmpty(t, resultResponse.NotificationDestination)
+
+ for _, security := range resultResponse.SecurityInfo {
+ assert.Equal(t, apiId, *security.ApiId)
+ assert.Equal(t, publishserviceapi.SecurityMethodPSK, *security.SelSecurityMethod)
+ }
+ invokerRegisterMock.AssertCalled(t, "IsInvokerRegistered", invokerId)
+}
+
+func TestPutTrustedInvokerNotFoundSecurityMethod(t *testing.T) {
+ invokerRegisterMock := invokermocks.InvokerRegister{}
+ invokerRegisterMock.On("IsInvokerRegistered", mock.AnythingOfType("string")).Return(true)
+
+ aefProfiles := []publishserviceapi.AefProfile{
+ getAefProfile("aefId"),
+ }
+ apiId := "apiId"
+ publishedServices := []publishserviceapi.ServiceAPIDescription{
+ {
+ ApiId: &apiId,
+ AefProfiles: &aefProfiles,
+ },
+ }
+ publishRegisterMock := publishmocks.PublishRegister{}
+ publishRegisterMock.On("GetAllPublishedServices").Return(publishedServices)
+
+ accessMgmMock := keycloackmocks.AccessManagement{}
+ accessMgmMock.On("AddClient", mock.AnythingOfType("string"), mock.AnythingOfType("string")).Return(nil)
+
+ requestHandler, _ := getEcho(nil, &publishRegisterMock, &invokerRegisterMock, &accessMgmMock)
+
+ invokerId := "invokerId"
+ serviceSecurityUnderTest := getServiceSecurity("aefId", "apiId")
+
+ result := testutil.NewRequest().Put("/trustedInvokers/"+invokerId).WithJsonBody(serviceSecurityUnderTest).Go(t, requestHandler)
+
+ badRequest := http.StatusBadRequest
+ assert.Equal(t, badRequest, result.Code())
+ var problemDetails common29122.ProblemDetails
+ err := result.UnmarshalBodyToObject(&problemDetails)
+ assert.NoError(t, err, "error unmarshaling response")
+ assert.Equal(t, &badRequest, problemDetails.Status)
+ assert.Contains(t, *problemDetails.Cause, "not found")
+ assert.Contains(t, *problemDetails.Cause, "security method")
+ invokerRegisterMock.AssertCalled(t, "IsInvokerRegistered", invokerId)
+}
+
+func TestDeleteSecurityContext(t *testing.T) {
+
+ requestHandler, securityUnderTest := getEcho(nil, nil, nil, nil)
+
+ aefId := "aefId"
+ apiId := "apiId"
+ serviceSecurityUnderTest := getServiceSecurity(aefId, apiId)
+ serviceSecurityUnderTest.SecurityInfo[0].ApiId = &apiId
+
+ invokerId := "invokerId"
+ securityUnderTest.trustedInvokers[invokerId] = serviceSecurityUnderTest
+
+ // Delete the security context
+ result := testutil.NewRequest().Delete("/trustedInvokers/"+invokerId).Go(t, requestHandler)
+
+ assert.Equal(t, http.StatusNoContent, result.Code())
+ _, ok := securityUnderTest.trustedInvokers[invokerId]
+ assert.False(t, ok)
+}
+
+func TestGetSecurityContextByInvokerId(t *testing.T) {
+
+ requestHandler, securityUnderTest := getEcho(nil, nil, nil, nil)
+
+ aefId := "aefId"
+ apiId := "apiId"
+ authenticationInfo := "authenticationInfo"
+ authorizationInfo := "authorizationInfo"
+ serviceSecurityUnderTest := getServiceSecurity(aefId, apiId)
+ serviceSecurityUnderTest.SecurityInfo[0].AuthenticationInfo = &authenticationInfo
+ serviceSecurityUnderTest.SecurityInfo[0].AuthorizationInfo = &authorizationInfo
+
+ invokerId := "invokerId"
+ securityUnderTest.trustedInvokers[invokerId] = serviceSecurityUnderTest
+
+ // Get security context
+ result := testutil.NewRequest().Get("/trustedInvokers/"+invokerId).Go(t, requestHandler)
+
+ assert.Equal(t, http.StatusOK, result.Code())
+ var resultService securityapi.ServiceSecurity
+ err := result.UnmarshalBodyToObject(&resultService)
+ assert.NoError(t, err, "error unmarshaling response")
+
+ for _, secInfo := range resultService.SecurityInfo {
+ assert.Equal(t, apiId, *secInfo.ApiId)
+ assert.Equal(t, aefId, *secInfo.AefId)
+ assert.Equal(t, "", *secInfo.AuthenticationInfo)
+ assert.Equal(t, "", *secInfo.AuthorizationInfo)
+ }
+
+ result = testutil.NewRequest().Get("/trustedInvokers/"+invokerId+"?authenticationInfo=true&authorizationInfo=false").Go(t, requestHandler)
+ assert.Equal(t, http.StatusOK, result.Code())
+ err = result.UnmarshalBodyToObject(&resultService)
+ assert.NoError(t, err, "error unmarshaling response")
+
+ for _, secInfo := range resultService.SecurityInfo {
+ assert.Equal(t, authenticationInfo, *secInfo.AuthenticationInfo)
+ assert.Equal(t, "", *secInfo.AuthorizationInfo)
+ }
+
+ result = testutil.NewRequest().Get("/trustedInvokers/"+invokerId+"?authenticationInfo=true&authorizationInfo=true").Go(t, requestHandler)
+ assert.Equal(t, http.StatusOK, result.Code())
+ err = result.UnmarshalBodyToObject(&resultService)
+ assert.NoError(t, err, "error unmarshaling response")
+
+ for _, secInfo := range resultService.SecurityInfo {
+ assert.Equal(t, authenticationInfo, *secInfo.AuthenticationInfo)
+ assert.Equal(t, authorizationInfo, *secInfo.AuthorizationInfo)
+ }
+}
+
+func TestUpdateTrustedInvoker(t *testing.T) {
+
+ requestHandler, securityUnderTest := getEcho(nil, nil, nil, nil)
+
+ aefId := "aefId"
+ apiId := "apiId"
+ invokerId := "invokerId"
+ serviceSecurityTest := getServiceSecurity(aefId, apiId)
+ serviceSecurityTest.SecurityInfo[0].ApiId = &apiId
+ securityUnderTest.trustedInvokers[invokerId] = serviceSecurityTest
+
+ // Update the service security with valid invoker, should return 200 with updated service security
+ newNotifURL := "http://golang.org/"
+ serviceSecurityTest.NotificationDestination = common29122.Uri(newNotifURL)
+ result := testutil.NewRequest().Post("/trustedInvokers/"+invokerId+"/update").WithJsonBody(serviceSecurityTest).Go(t, requestHandler)
+
+ var resultResponse securityapi.ServiceSecurity
+ assert.Equal(t, http.StatusOK, result.Code())
+ err := result.UnmarshalBodyToObject(&resultResponse)
+ assert.NoError(t, err, "error unmarshaling response")
+ assert.Equal(t, newNotifURL, string(resultResponse.NotificationDestination))
+
+ // Update with a service security missing required NotificationDestination, should get 400 with problem details
+ invalidServiceSecurity := securityapi.ServiceSecurity{
+ SecurityInfo: []securityapi.SecurityInformation{
+ {
+ AefId: &aefId,
+ ApiId: &apiId,
+ PrefSecurityMethods: []publishserviceapi.SecurityMethod{
+ publishserviceapi.SecurityMethodOAUTH,
+ },
+ },
+ },
+ }
+
+ result = testutil.NewRequest().Post("/trustedInvokers/"+invokerId+"/update").WithJsonBody(invalidServiceSecurity).Go(t, requestHandler)
+
+ assert.Equal(t, http.StatusBadRequest, result.Code())
+ var problemDetails common29122.ProblemDetails
+ err = result.UnmarshalBodyToObject(&problemDetails)
+ assert.NoError(t, err, "error unmarshaling response")
+ assert.Equal(t, http.StatusBadRequest, *problemDetails.Status)
+ assert.Contains(t, *problemDetails.Cause, "missing")
+ assert.Contains(t, *problemDetails.Cause, "notificationDestination")
+
+ // Update a service security that has not been registered, should get 404 with problem details
+ missingId := "1"
+ result = testutil.NewRequest().Post("/trustedInvokers/"+missingId+"/update").WithJsonBody(serviceSecurityTest).Go(t, requestHandler)
+
+ assert.Equal(t, http.StatusNotFound, result.Code())
+ err = result.UnmarshalBodyToObject(&problemDetails)
+ assert.NoError(t, err, "error unmarshaling response")
+ assert.Equal(t, http.StatusNotFound, *problemDetails.Status)
+ assert.Contains(t, *problemDetails.Cause, "not register")
+ assert.Contains(t, *problemDetails.Cause, "trusted invoker")
+}
+
+func TestRevokeAuthorizationToInvoker(t *testing.T) {
+ aefId := "aefId"
+ apiId := "apiId"
+ invokerId := "invokerId"
+
+ notification := securityapi.SecurityNotification{
+ AefId: &aefId,
+ ApiInvokerId: invokerId,
+ ApiIds: []string{apiId},
+ Cause: securityapi.CauseUNEXPECTEDREASON,
+ }
+
+ requestHandler, securityUnderTest := getEcho(nil, nil, nil, nil)
+
+ serviceSecurityTest := getServiceSecurity(aefId, apiId)
+ serviceSecurityTest.SecurityInfo[0].ApiId = &apiId
+
+ apiIdTwo := "apiIdTwo"
+ secInfo := securityapi.SecurityInformation{
+ AefId: &aefId,
+ ApiId: &apiIdTwo,
+ PrefSecurityMethods: []publishserviceapi.SecurityMethod{
+ publishserviceapi.SecurityMethodPKI,
+ },
+ }
+
+ serviceSecurityTest.SecurityInfo = append(serviceSecurityTest.SecurityInfo, secInfo)
+
+ securityUnderTest.trustedInvokers[invokerId] = serviceSecurityTest
+
+ // Revoke apiId
+ result := testutil.NewRequest().Post("/trustedInvokers/"+invokerId+"/delete").WithJsonBody(notification).Go(t, requestHandler)
+
+ assert.Equal(t, http.StatusNoContent, result.Code())
+ assert.Equal(t, 1, len(securityUnderTest.trustedInvokers[invokerId].SecurityInfo))
+
+ notification.ApiIds = []string{apiIdTwo}
+ // Revoke apiIdTwo
+ result = testutil.NewRequest().Post("/trustedInvokers/"+invokerId+"/delete").WithJsonBody(notification).Go(t, requestHandler)
+
+ assert.Equal(t, http.StatusNoContent, result.Code())
+ _, ok := securityUnderTest.trustedInvokers[invokerId]
+ assert.False(t, ok)
+}
+
+func getEcho(serviceRegister providermanagement.ServiceRegister, publishRegister publishservice.PublishRegister, invokerRegister invokermanagement.InvokerRegister, keycloakMgm keycloak.AccessManagement) (*echo.Echo, *Security) {