[RIC-587, RIC-584] Add Automation for RIC-587 | Add UTs for RIC-584

[ric-plt/e2mgr.git] / E2Manager / managers / update_enb_manager.go

diff --git a/E2Manager/managers/update_enb_manager.go b/E2Manager/managers/update_enb_manager.go
index cb3e4bc..6cb5414 100755 (executable)
--- a/E2Manager/managers/update_enb_manager.go
+++ b/E2Manager/managers/update_enb_manager.go
@@ -82,7 +82,11 @@ func (h *UpdateEnbManager) RemoveNodebCells(nodeb *entities.NodebInfo) error {
 
 func (h *UpdateEnbManager) SetNodeb(nodeb *entities.NodebInfo, request models.Request) {
        updateEnbRequest := request.(*models.UpdateEnbRequest)
-       updateEnbRequest.Enb.EnbType = nodeb.GetEnb().GetEnbType()
+
+       if nodeb.SetupFromNetwork {
+               updateEnbRequest.Enb.EnbType = nodeb.GetEnb().GetEnbType()
+       }
+
        nodeb.Configuration = &entities.NodebInfo_Enb{Enb: updateEnbRequest.Enb}
 }
 
@@ -108,7 +112,7 @@ func (h *UpdateEnbManager) validateRequestBody(request *models.UpdateEnbRequest)
                return err
        }
 
-       if h.nodebValidator.IsNgEnbType(request.Enb.GetEnbType()){
+       if h.nodebValidator.IsNgEnbType(request.Enb.GetEnbType()) {
                return errors.New("enb.enbType")
        }