fix: set UPF association context correctly

free5gc · Aug 20, 2024 · ee395c5 · ee395c5
1 parent 900f987
commit ee395c5
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/internal/context/user_plane_information_test.go b/internal/context/user_plane_information_test.go
@@ -504,6 +504,7 @@ func TestGetUEIPPool(t *testing.T) {
 	userplaneInformation := smf_context.NewUserPlaneInformation(configForIPPoolAllocate)
 	for _, upf := range userplaneInformation.UPFs {
 		upf.UPFStatus = smf_context.AssociatedSetUpSuccess
+		upf.Association, upf.AssociationCancelFunc = context.WithCancel(context.Background())
 	}
 
 	for ci, tc := range testCasesOfGetUEIPPool {