X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=a1-go%2Fpkg%2Fresthooks%2Fresthooks_test.go;h=8c6b59ad846520549773beeb2e7942b7a6463f53;hb=159ddb7dd65305d364743d4728af35f84a6e1557;hp=deffc4b717c5cecfbf6e20d0bcd5e7f9437b4d2f;hpb=52ef1de13a1c245638264c3025eea0bf2fb8eb42;p=ric-plt%2Fa1.git

diff --git a/a1-go/pkg/resthooks/resthooks_test.go b/a1-go/pkg/resthooks/resthooks_test.go
index deffc4b..8c6b59a 100644
--- a/a1-go/pkg/resthooks/resthooks_test.go
+++ b/a1-go/pkg/resthooks/resthooks_test.go
@@ -144,7 +144,7 @@ func TestCreatePolicyTypeInstance(t *testing.T) {
 	a1.Logger.Debug("metadatainstancekey   : %+v", metadatainstancekey)
 	metadatainstancearr := []interface{}{metadatainstancekey, string(metadata)}
 	sdlInst.On("Set", "A1m_ns", metadatainstancearr).Return(nil)
-	rmrSenderInst.On("RmrSendToXapp", "httpBodyString").Return(true)
+	rmrSenderInst.On("RmrSendToXapp", "httpBodyString", 20010).Return(true)
 
 	errresp := rh.CreatePolicyInstance(policyTypeId, policyInstanceID, instancedata)
 
@@ -287,13 +287,31 @@ func TestDeletePolicyInstance(t *testing.T) {
 
 	httpBodyString := `{"operation":"DELETE","payload":"","policy_instance_id":"123456","policy_type_id":"20001"}`
 
-	rmrSenderInst.On("RmrSendToXapp", httpBodyString).Return(true)
+	rmrSenderInst.On("RmrSendToXapp", httpBodyString, 20010).Return(true)
 
 	errresp := rh.DeletePolicyInstance(policyTypeId, policyInstanceID)
 
 	assert.Nil(t, errresp)
 	sdlInst.AssertExpectations(t)
 }
+func TestDataDelivery(t *testing.T) {
+
+	httpBody := `{
+		"job":"1",
+		"payload":"payload"
+		}
+		`
+	var instancedata interface{}
+
+	json.Unmarshal([]byte(httpBody), &instancedata)
+	a1.Logger.Debug("Marshaled data : %+v", (instancedata))
+	httpBodyString := `{"ei_job_id":"1","payload":"payload"}`
+	rmrSenderInst.On("RmrSendToXapp", httpBodyString, 20017).Return(true)
+	errresp := rh.DataDelivery(instancedata)
+
+	assert.Nil(t, errresp)
+	sdlInst.AssertExpectations(t)
+}
 
 type SdlMock struct {
 	mock.Mock
@@ -353,9 +371,15 @@ func (s *SdlMock) SetIf(ns string, key string, oldData, newData interface{}) (bo
 	return args.Bool(0), args.Error(1)
 }
 
-func (rmr *RmrSenderMock) RmrSendToXapp(httpBodyString string) bool {
-	args := rmr.MethodCalled("RmrSendToXapp", httpBodyString)
-	return args.Bool(0)
+func (rmr *RmrSenderMock) RmrSendToXapp(httpBodyString string, mtype int) bool {
+	if httpBodyString == `{"blocking_rate":20,"enforce":true,"trigger_threshold":10,"window_length":20}` {
+		args := rmr.MethodCalled("RmrSendToXapp", httpBodyString, mtype)
+		return args.Bool(0)
+	} else if httpBodyString == `{"ei_job_id":"1","payload":"payload"}` {
+		args := rmr.MethodCalled("RmrSendToXapp", httpBodyString, mtype)
+		return args.Bool(0)
+	}
+	return true
 }
 
 func (s *SdlMock) Remove(ns string, keys []string) error {