X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fmocks%2FrnibWriterMock.go;h=b380d3a414960a8e3126890f6609ebd791ca6a1a;hb=refs%2Fchanges%2F12%2F4512%2F1;hp=53ec47a25cccd25f61d24f120d68d58633f57f03;hpb=2c9d450c93e4f0b312a68d39429efb685ec02022;p=ric-plt%2Fe2mgr.git diff --git a/E2Manager/mocks/rnibWriterMock.go b/E2Manager/mocks/rnibWriterMock.go index 53ec47a..b380d3a 100644 --- a/E2Manager/mocks/rnibWriterMock.go +++ b/E2Manager/mocks/rnibWriterMock.go @@ -17,7 +17,6 @@ // This source code is part of the near-RT RIC (RAN Intelligent Controller) // platform project (RICP). - package mocks import ( @@ -29,8 +28,8 @@ type RnibWriterMock struct { mock.Mock } -func (rnibWriterMock *RnibWriterMock) SaveNodeb(nbIdentity *entities.NbIdentity, nb *entities.NodebInfo) error { - args := rnibWriterMock.Called(nbIdentity, nb) +func (rnibWriterMock *RnibWriterMock) SaveNodeb(nodebInfo *entities.NodebInfo) error { + args := rnibWriterMock.Called(nodebInfo) errArg := args.Get(0) @@ -53,6 +52,18 @@ func (rnibWriterMock *RnibWriterMock) UpdateNodebInfo(nodebInfo *entities.NodebI return nil } +func (rnibWriterMock *RnibWriterMock) UpdateNodebInfoAndPublish(nodebInfo *entities.NodebInfo) error { + args := rnibWriterMock.Called(nodebInfo) + + errArg := args.Get(0) + + if errArg != nil { + return errArg.(error) + } + + return nil +} + func (rnibWriterMock *RnibWriterMock) SaveRanLoadInformation(inventoryName string, ranLoadInformation *entities.RanLoadInformation) error { args := rnibWriterMock.Called(inventoryName, ranLoadInformation) @@ -64,3 +75,80 @@ func (rnibWriterMock *RnibWriterMock) SaveRanLoadInformation(inventoryName strin return nil } + +func (rnibWriterMock *RnibWriterMock) SaveE2TInstance(e2tInstance *entities.E2TInstance) error { + args := rnibWriterMock.Called(e2tInstance) + + return args.Error(0) +} + +func (rnibWriterMock *RnibWriterMock) SaveE2TAddresses(addresses []string) error { + args := rnibWriterMock.Called(addresses) + + return args.Error(0) +} + +func (rnibWriterMock *RnibWriterMock) RemoveE2TInstance(address string) error { + args := rnibWriterMock.Called(address) + + return args.Error(0) +} + +func (rnibWriterMock *RnibWriterMock) UpdateGnbCells(nodebInfo *entities.NodebInfo, servedNrCells []*entities.ServedNRCell) error { + args := rnibWriterMock.Called(nodebInfo, servedNrCells) + return args.Error(0) +} + +func (rnibWriterMock *RnibWriterMock) RemoveServedNrCells(inventoryName string, servedNrCells []*entities.ServedNRCell) error { + args := rnibWriterMock.Called(inventoryName, servedNrCells) + return args.Error(0) +} + +func (rnibWriterMock *RnibWriterMock) UpdateNodebInfoOnConnectionStatusInversion(nodebInfo *entities.NodebInfo, event string) error { + args := rnibWriterMock.Called(nodebInfo, event) + + return args.Error(0) +} + +func (rnibWriterMock *RnibWriterMock) SaveGeneralConfiguration(config *entities.GeneralConfiguration) error { + args := rnibWriterMock.Called(config) + + return args.Error(0) +} + +func (rnibWriterMock *RnibWriterMock) RemoveEnb(nodebInfo *entities.NodebInfo) error { + args := rnibWriterMock.Called(nodebInfo) + + return args.Error(0) +} + +func (rnibWriterMock *RnibWriterMock) RemoveServedCells(inventoryName string, servedCells []*entities.ServedCellInfo) error { + args := rnibWriterMock.Called(inventoryName, servedCells) + + return args.Error(0) +} + +func (rnibWriterMock *RnibWriterMock) UpdateEnb(nodebInfo *entities.NodebInfo, servedCells []*entities.ServedCellInfo) error { + args := rnibWriterMock.Called(nodebInfo, servedCells) + + return args.Error(0) +} + +func (rnibWriterMock *RnibWriterMock) AddNbIdentity(nodeType entities.Node_Type, nbIdentity *entities.NbIdentity) error { + args := rnibWriterMock.Called(nodeType, nbIdentity) + return args.Error(0) +} +func (rnibWriterMock *RnibWriterMock) RemoveNbIdentity(nodeType entities.Node_Type, nbIdentity *entities.NbIdentity) error { + args := rnibWriterMock.Called(nodeType, nbIdentity) + return args.Error(0) +} + +func (rnibWriterMock *RnibWriterMock) AddEnb(nodebInfo *entities.NodebInfo) error { + args := rnibWriterMock.Called(nodebInfo) + return args.Error(0) +} + +func (rnibWriterMock *RnibWriterMock) UpdateNbIdentities(nodeType entities.Node_Type, oldNbIdentities []*entities.NbIdentity, newNbIdentities []*entities.NbIdentity) error { + args := rnibWriterMock.Called(nodeType, oldNbIdentities, newNbIdentities) + return args.Error(0) +}