Fix Error Indication during Deregistration

Author: linouxis9

internal/control_test_engine/gnb/ngap/handler/handler.go

@@ -713,9 +713,7 @@ func HandlerErrorIndication(gnb *context.GNBContext, message *ngapType.NGAPPDU)
 		}
 	}
 
-	ue := getUeFromContext(gnb, ranUeId, amfUeId)
-
-	log.Warn("[GNB][AMF] Received an Error Indication for UE with AMF UE ID: ", ue.GetAmfUeId(), ", RAN UE ID: ", ue.GetRanUeId())
+	log.Warn("[GNB][AMF] Received an Error Indication for UE with AMF UE ID: ", amfUeId, " RAN UE ID: ", ranUeId)
 }
 
 func getUeFromContext(gnb *context.GNBContext, ranUeId int64, amfUeId int64) *context.GNBUe {

internal/control_test_engine/gnb/ngap/service/service.go

@@ -80,7 +80,7 @@ func GnbListen(amf *context.GNBAmf, gnb *context.GNBContext) {
 		copy(forwardData, buf[:n])
 
 		// handling NGAP message.
-		ngap.Dispatch(amf, gnb, forwardData)
+		go ngap.Dispatch(amf, gnb, forwardData)
 
 	}
 

internal/control_test_engine/ue/ue.go

@@ -137,8 +137,8 @@ func ueMgrHandler(msg procedures.UeTesterMessage, ue *context.UEContext) bool {
 					trigger.InitPduSessionRelease(ue, pduSession)
 					select {
 					case <-pduSession.Wait:
-					case <-time.After(5 * time.Millisecond):
-						// If still unregistered after 5 ms, continue
+					case <-time.After(500 * time.Millisecond):
+						// If still unregistered after 500 ms, continue
 					}
 				}
 			}