UT for SetPolicyInstanceStatus

[ric-plt/a1.git] / a1-go / pkg / restful / restful.go

diff --git a/a1-go/pkg/restful/restful.go b/a1-go/pkg/restful/restful.go
index b0663dd..44b3422 100644 (file)
--- a/a1-go/pkg/restful/restful.go
+++ b/a1-go/pkg/restful/restful.go
@@ -28,6 +28,7 @@ import (
        "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/models"
        "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/restapi"
        "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/restapi/operations"
+       "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/restapi/operations/a1_e_i_data_delivery"
        "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/restapi/operations/a1_mediator"
        "gerrit.o-ran-sc.org/r/ric-plt/a1/pkg/resthooks"
        "github.com/go-openapi/loads"
@@ -49,6 +50,16 @@ func (r *Restful) setupHandler() *operations.A1API {
        }
 
        api := operations.NewA1API(swaggerSpec)
+
+       api.A1MediatorA1ControllerGetHealthcheckHandler = a1_mediator.A1ControllerGetHealthcheckHandlerFunc(func(param a1_mediator.A1ControllerGetHealthcheckParams) middleware.Responder {
+               a1.Logger.Debug("handler for get Health Check of A1")
+               resp := r.rh.GetA1Health()
+               if resp == false {
+                       return a1_mediator.NewA1ControllerGetHealthcheckInternalServerError()
+               }
+               return a1_mediator.NewA1ControllerGetHealthcheckOK()
+       })
+
        api.A1MediatorA1ControllerGetAllPolicyTypesHandler = a1_mediator.A1ControllerGetAllPolicyTypesHandlerFunc(func(param a1_mediator.A1ControllerGetAllPolicyTypesParams) middleware.Responder {
                a1.Logger.Debug("handler for get all policy type")
                return a1_mediator.NewA1ControllerGetAllPolicyTypesOK().WithPayload(r.rh.GetAllPolicyType())
@@ -145,7 +156,15 @@ func (r *Restful) setupHandler() *operations.A1API {
                return a1_mediator.NewA1ControllerDeletePolicyInstanceAccepted()
 
        })
-       
+
+       api.A1eiDataDeliveryA1ControllerDataDeliveryHandler = a1_e_i_data_delivery.A1ControllerDataDeliveryHandlerFunc(func(params a1_e_i_data_delivery.A1ControllerDataDeliveryParams) middleware.Responder {
+               a1.Logger.Debug("handler for EI data delivery")
+               if err = r.rh.DataDelivery(params.Body); err != nil {
+                       return a1_e_i_data_delivery.NewA1ControllerDataDeliveryNotFound()
+               }
+               return a1_e_i_data_delivery.NewA1ControllerDataDeliveryOK()
+       })
+
        return api
 
 }