Merge "Automation of nodeb state API"

[ric-plt/e2mgr.git] / E2Manager / handlers / httpmsghandlers / health_check_handler_test.go

diff --git a/E2Manager/handlers/httpmsghandlers/health_check_handler_test.go b/E2Manager/handlers/httpmsghandlers/health_check_handler_test.go
index bbd7acd..4bce41f 100644 (file)
--- a/E2Manager/handlers/httpmsghandlers/health_check_handler_test.go
+++ b/E2Manager/handlers/httpmsghandlers/health_check_handler_test.go
@@ -75,8 +75,9 @@ func TestHealthCheckRequestHandlerArguementHasRanNameSuccess(t *testing.T) {
        ranListManagerMock.On("UpdateHealthcheckTimeStampSent",nb1.RanName).Return(oldnbIdentity, newnbIdentity)
        ranListManagerMock.On("UpdateNbIdentities",nb1.NodeType, []*entities.NbIdentity{oldnbIdentity}, []*entities.NbIdentity{newnbIdentity}).Return(nil)
 
-       _, err := handler.Handle(models.HealthCheckRequest{ranNames})
+       resp, err := handler.Handle(models.HealthCheckRequest{ranNames})
 
+       assert.IsType(t, &models.HealthCheckSuccessResponse{}, resp)
        assert.Nil(t, err)
        readerMock.AssertExpectations(t)
 }
@@ -103,9 +104,10 @@ func TestHealthCheckRequestHandlerArguementHasNoRanNameSuccess(t *testing.T) {
        nb2 := &entities.NodebInfo{RanName: "RanName_2", ConnectionStatus: entities.ConnectionStatus_DISCONNECTED}
        readerMock.On("GetNodeb", "RanName_2").Return(nb2, nil)
 
-       _, err := handler.Handle(models.HealthCheckRequest{[]string{}})
+       resp, err := handler.Handle(models.HealthCheckRequest{[]string{}})
 
        assert.Nil(t, err)
+       assert.IsType(t, &models.HealthCheckSuccessResponse{}, resp)
 
 }