Copy latest code

[ric-plt/e2mgr.git] / E2Manager / mocks / e2t_shutdown_manager_mock.go

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 (file)
--- 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