X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fmocks%2Fe2t_shutdown_manager_mock.go;fp=E2Manager%2FrnibBuilders%2Fnode_info_builder.go;h=c560f1d79b5853c64909f32738c6fe0ca6fa0d64;hb=efcb4528362460aa2249d319c9752b63bb720fe2;hp=bfee4f66044f90178ca5b90423cfb1403f4d8a55;hpb=b6c5a882f986ed2d6440cf9be83d69740179c53d;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/rnibBuilders/node_info_builder.go b/E2Manager/mocks/e2t_shutdown_manager_mock.go
similarity index 56%
rename from E2Manager/rnibBuilders/node_info_builder.go
rename to E2Manager/mocks/e2t_shutdown_manager_mock.go
index bfee4f6..c560f1d 100644
--- a/E2Manager/rnibBuilders/node_info_builder.go
+++ b/E2Manager/mocks/e2t_shutdown_manager_mock.go
@@ -18,23 +18,18 @@
 //  platform project (RICP).
 
 
-package rnibBuilders
+package mocks
 
 import (
 	"gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
-	"e2mgr/models"
+	"github.com/stretchr/testify/mock"
 )
 
-func CreateInitialNodeInfo(requestDetails *models.SetupRequest, protocol entities.E2ApplicationProtocol) (*entities.NodebInfo, *entities.NbIdentity) {
-	nodebInfo := &entities.NodebInfo{}
-	nodebInfo.Ip = requestDetails.RanIp
-	nodebInfo.Port = uint32(requestDetails.RanPort)
-	nodebInfo.ConnectionStatus = entities.ConnectionStatus_CONNECTING
-	nodebInfo.E2ApplicationProtocol = protocol
-	nodebInfo.RanName = requestDetails.RanName
-	nodebInfo.ConnectionAttempts = 0
+type E2TShutdownManagerMock struct {
+	mock.Mock
+}
 
-	nodebIdentity := &entities.NbIdentity{}
-	nodebIdentity.InventoryName = requestDetails.RanName
-	return nodebInfo, nodebIdentity
+func (m *E2TShutdownManagerMock) Shutdown(e2tInstance *entities.E2TInstance) error {
+	args := m.Called(e2tInstance)
+	return args.Error(0)
 }
\ No newline at end of file