X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=E2Manager%2Fmanagers%2Fe2t_instances_manager_test.go;fp=E2Manager%2Fmanagers%2Fe2t_instances_manager_test.go;h=006c5a24d80a52915145bc8a28e7f735b5cac682;hb=fba870af318f9c7061fb53bf5ae398f600069b6e;hp=2c71cd3e66a8e12b1e1fa9e7174d37546aaf4953;hpb=d36bbc4b8f5a0798b46d9631af1a264979f4a28b;p=ric-plt%2Fe2mgr.git

diff --git a/E2Manager/managers/e2t_instances_manager_test.go b/E2Manager/managers/e2t_instances_manager_test.go
index 2c71cd3..006c5a2 100644
--- a/E2Manager/managers/e2t_instances_manager_test.go
+++ b/E2Manager/managers/e2t_instances_manager_test.go
@@ -1,3 +1,19 @@
+//
+// Copyright 2019 AT&T Intellectual Property
+// Copyright 2019 Nokia
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+//
 package managers
 
 import (
@@ -276,6 +292,73 @@ func TestSelectE2TInstancesSuccess(t *testing.T) {
 	rnibWriterMock.AssertExpectations(t)
 }
 
+func TestResetKeepAliveTimestampGetInternalFailure(t *testing.T) {
+	rnibReaderMock, rnibWriterMock, e2tInstancesManager := initE2TInstancesManagerTest(t)
+
+	address := "10.10.2.15:9800"
+	e2tInstance := entities.NewE2TInstance(address)
+	rnibReaderMock.On("GetE2TInstance", address).Return(e2tInstance, common.NewInternalError(errors.New("Error")))
+	rnibWriterMock.On("SaveE2TInstance", mock.Anything).Return(nil)
+
+	err := e2tInstancesManager.ResetKeepAliveTimestamp(address)
+	assert.NotNil(t, err)
+	rnibReaderMock.AssertNotCalled(t, "SaveE2TInstance")
+}
+
+func TestAResetKeepAliveTimestampSaveInternalFailure(t *testing.T) {
+	rnibReaderMock, rnibWriterMock, e2tInstancesManager := initE2TInstancesManagerTest(t)
+
+	address := "10.10.2.15:9800"
+	e2tInstance := entities.NewE2TInstance(address)
+	rnibReaderMock.On("GetE2TInstance", address).Return(e2tInstance, nil)
+	rnibWriterMock.On("SaveE2TInstance", mock.Anything).Return(common.NewInternalError(errors.New("Error")))
+
+	err := e2tInstancesManager.ResetKeepAliveTimestamp(address)
+	assert.NotNil(t, err)
+}
+
+func TestResetKeepAliveTimestampSuccess(t *testing.T) {
+	rnibReaderMock, rnibWriterMock, e2tInstancesManager := initE2TInstancesManagerTest(t)
+
+	address := "10.10.2.15:9800"
+	e2tInstance := entities.NewE2TInstance(address)
+	rnibReaderMock.On("GetE2TInstance", address).Return(e2tInstance, nil)
+	rnibWriterMock.On("SaveE2TInstance", mock.Anything).Return(nil)
+
+	err := e2tInstancesManager.ResetKeepAliveTimestamp(address)
+	assert.Nil(t, err)
+	rnibReaderMock.AssertCalled(t, "GetE2TInstance", address)
+	rnibWriterMock.AssertNumberOfCalls(t, "SaveE2TInstance", 1)
+}
+
+func TestResetKeepAliveTimestampToBeDeleted(t *testing.T) {
+	rnibReaderMock, rnibWriterMock, e2tInstancesManager := initE2TInstancesManagerTest(t)
+
+	address := "10.10.2.15:9800"
+	e2tInstance := entities.NewE2TInstance(address)
+	e2tInstance.State = entities.ToBeDeleted
+	rnibReaderMock.On("GetE2TInstance", address).Return(e2tInstance, nil)
+
+	err := e2tInstancesManager.ResetKeepAliveTimestamp(address)
+	assert.Nil(t, err)
+	rnibReaderMock.AssertCalled(t, "GetE2TInstance", address)
+	rnibWriterMock.AssertNotCalled(t, "SaveE2TInstance")
+}
+
+func TestResetKeepAliveTimestampRoutingManagerFailure(t *testing.T) {
+	rnibReaderMock, rnibWriterMock, e2tInstancesManager := initE2TInstancesManagerTest(t)
+
+	address := "10.10.2.15:9800"
+	e2tInstance := entities.NewE2TInstance(address)
+	e2tInstance.State = entities.RoutingManagerFailure
+	rnibReaderMock.On("GetE2TInstance", address).Return(e2tInstance, nil)
+
+	err := e2tInstancesManager.ResetKeepAliveTimestamp(address)
+	assert.Nil(t, err)
+	rnibReaderMock.AssertCalled(t, "GetE2TInstance", address)
+	rnibWriterMock.AssertNotCalled(t, "SaveE2TInstance")
+}
+
 func TestRemoveE2TInstance(t *testing.T) {
 	_, _, e2tInstancesManager := initE2TInstancesManagerTest(t)
 	e2tInstance1  := entities.NewE2TInstance(E2TAddress)