X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=capifcore%2Finternal%2Fprovidermanagement%2Fprovidermanagement_test.go;fp=capifcore%2Finternal%2Fprovidermanagement%2Fprovidermanagement_test.go;h=a3bb7aa6304ff6ea7a22fff36dc7e1e91eee7cf0;hb=3ff31a181cad84bc2f5554f8440390bedb42711c;hp=b5c899b125cfabe03ce496bb58fbe7fe808ee856;hpb=c01d387b9ff3b5da35e6d9a635523e6747c5b72c;p=nonrtric%2Fplt%2Fsme.git

diff --git a/capifcore/internal/providermanagement/providermanagement_test.go b/capifcore/internal/providermanagement/providermanagement_test.go
index b5c899b..a3bb7aa 100644
--- a/capifcore/internal/providermanagement/providermanagement_test.go
+++ b/capifcore/internal/providermanagement/providermanagement_test.go
@@ -70,7 +70,7 @@ func TestRegisterValidProvider(t *testing.T) {
 	assert.True(t, managerUnderTest.IsFunctionRegistered("APF_id_rApp_as_APF"))
 }
 
-func TestUpdateValidProvider(t *testing.T) {
+func TestUpdateValidProviderWithNewFunction(t *testing.T) {
 	managerUnderTest, requestHandler := getEcho()
 
 	provider := getProvider()
@@ -111,6 +111,38 @@ func TestUpdateValidProvider(t *testing.T) {
 	assert.True(t, managerUnderTest.IsFunctionRegistered("AEF_id_new_func_as_AEF"))
 }
 
+func TestUpdateValidProviderWithDeletedFunction(t *testing.T) {
+	managerUnderTest, requestHandler := getEcho()
+
+	provider := getProvider()
+	provider.ApiProvDomId = &domainID
+	(*provider.ApiProvFuncs)[0].ApiProvFuncId = &funcIdAPF
+	(*provider.ApiProvFuncs)[1].ApiProvFuncId = &funcIdAMF
+	(*provider.ApiProvFuncs)[2].ApiProvFuncId = &funcIdAEF
+	managerUnderTest.onboardedProviders[domainID] = provider
+
+	// Modify the provider
+	updatedProvider := getProvider()
+	updatedProvider.ApiProvDomId = &domainID
+	(*updatedProvider.ApiProvFuncs)[0].ApiProvFuncId = &funcIdAPF
+	(*updatedProvider.ApiProvFuncs)[2].ApiProvFuncId = &funcIdAEF
+	testFuncs := []provapi.APIProviderFunctionDetails{
+		(*updatedProvider.ApiProvFuncs)[0],
+		(*updatedProvider.ApiProvFuncs)[2],
+	}
+	updatedProvider.ApiProvFuncs = &testFuncs
+
+	result := testutil.NewRequest().Put("/registrations/"+domainID).WithJsonBody(updatedProvider).Go(t, requestHandler)
+
+	var resultProvider provapi.APIProviderEnrolmentDetails
+	assert.Equal(t, http.StatusOK, result.Code())
+	err := result.UnmarshalBodyToObject(&resultProvider)
+	assert.NoError(t, err, "error unmarshaling response")
+	assert.Len(t, (*resultProvider.ApiProvFuncs), 2)
+	assert.Empty(t, resultProvider.FailReason)
+	assert.False(t, managerUnderTest.IsFunctionRegistered(funcIdAMF))
+}
+
 func TestUpdateMissingFunction(t *testing.T) {
 	managerUnderTest, requestHandler := getEcho()