[RICPLT-2585] Add UTs to init flow

[ric-plt/e2mgr.git] / E2Manager / rnibBuilders / node_info_builder_test.go

diff --git a/E2Manager/rnibBuilders/node_info_builder_test.go b/E2Manager/rnibBuilders/node_info_builder_test.go
index 40c0d1b..fc55935 100644 (file)
--- a/E2Manager/rnibBuilders/node_info_builder_test.go
+++ b/E2Manager/rnibBuilders/node_info_builder_test.go
@@ -29,14 +29,15 @@ const ranIP = "ip"
 const ranPort = uint16(30000)
 
 func TestCreateInitialNodeInfo(t *testing.T) {
-       requestDetails :=  &models.RequestDetails{
+       requestDetails :=  &models.SetupRequest{
                RanName: ranName,
                RanPort:ranPort,
                RanIp:ranIP,
        }
-       nodebInfo, identity := CreateInitialNodeInfo(requestDetails)
+       nodebInfo, identity := CreateInitialNodeInfo(requestDetails, entities.E2ApplicationProtocol_X2_SETUP_REQUEST)
        assert.Equal(t, identity.InventoryName, ranName)
        assert.Equal(t, nodebInfo.Ip, ranIP)
        assert.Equal(t, nodebInfo.ConnectionStatus, entities.ConnectionStatus_CONNECTING)
+       assert.Equal(t, nodebInfo.E2ApplicationProtocol, entities.E2ApplicationProtocol_X2_SETUP_REQUEST)
        assert.Equal(t, nodebInfo.Port, uint32(ranPort))
 }
\ No newline at end of file