//
// 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.
//

//  This source code is part of the near-RT RIC (RAN Intelligent Controller)
//  platform project (RICP).

package managers

import (
	"bytes"
	"e2mgr/clients"
	"e2mgr/configuration"
	"e2mgr/e2managererrors"
	"e2mgr/mocks"
	"e2mgr/models"
	"e2mgr/services"
	"encoding/json"
	"fmt"
	"io/ioutil"

	"gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
	"gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"

	//"k8s.io/apimachinery/pkg/runtime"
	//"k8s.io/client-go/kubernetes/fake"
	"net/http"
	"testing"
	"time"
)

const E2TAddress3 = "10.10.2.17:9800"

func initE2TShutdownManagerTest(t *testing.T) (*E2TShutdownManager, *mocks.RnibReaderMock, *mocks.RnibWriterMock, *mocks.HttpClientMock) {
	log := initLog(t)
	config := &configuration.Configuration{RnibRetryIntervalMs: 10, MaxRnibConnectionAttempts: 3, E2TInstanceDeletionTimeoutMs: 15000, RnibWriter: configuration.RnibWriterConfig{StateChangeMessageChannel: "RAN_CONNECTION_STATUS_CHANGE"}}

	readerMock := &mocks.RnibReaderMock{}
	writerMock := &mocks.RnibWriterMock{}
	rnibDataService := services.NewRnibDataService(log, config, readerMock, writerMock)

	e2tInstancesManager := NewE2TInstancesManager(rnibDataService, log)
	httpClientMock := &mocks.HttpClientMock{}
	rmClient := clients.NewRoutingManagerClient(log, config, httpClientMock)

	ranListManager := NewRanListManager(log, rnibDataService)
	ranAlarmService := services.NewRanAlarmService(log, config)
	ranConnectStatusChangeManager := NewRanConnectStatusChangeManager(log, rnibDataService, ranListManager, ranAlarmService)
	associationManager := NewE2TAssociationManager(log, rnibDataService, e2tInstancesManager, rmClient, ranConnectStatusChangeManager)
	shutdownManager := NewE2TShutdownManager(log, config, rnibDataService, e2tInstancesManager, associationManager, ranConnectStatusChangeManager)

	return shutdownManager, readerMock, writerMock, httpClientMock
}

func TestShutdownSuccess1OutOf3Instances(t *testing.T) {
	shutdownManager, readerMock, writerMock, httpClientMock := initE2TShutdownManagerTest(t)

	e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
	e2tInstance1.State = entities.Active
	e2tInstance1.AssociatedRanList = []string{"test1", "test2", "test5"}
	e2tInstance2 := entities.NewE2TInstance(E2TAddress2, PodName)
	e2tInstance2.State = entities.Active
	e2tInstance2.AssociatedRanList = []string{"test3"}
	e2tInstance3 := entities.NewE2TInstance(E2TAddress3, PodName)
	e2tInstance3.State = entities.Active
	e2tInstance3.AssociatedRanList = []string{"test4"}
	writerMock.On("SaveE2TInstance", mock.MatchedBy(func(e2tInstance *entities.E2TInstance) bool {
		return e2tInstance.Address == E2TAddress && e2tInstance.State == entities.ToBeDeleted
	})).Return(nil)

	nodeb1 := &entities.NodebInfo{RanName: "test1", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test1").Return(nodeb1, nil)
	nodeb2 := &entities.NodebInfo{RanName: "test2", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test2").Return(nodeb2, nil)
	nodeb5 := &entities.NodebInfo{RanName: "test5", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test5").Return(nodeb5, nil)

	e2tAddresses := []string{E2TAddress, E2TAddress2, E2TAddress3}
	readerMock.On("GetE2TAddresses").Return(e2tAddresses, nil)

	data := models.NewRoutingManagerDeleteRequestModel(E2TAddress, e2tInstance1.AssociatedRanList, nil)
	marshaled, _ := json.Marshal(data)
	body := bytes.NewBuffer(marshaled)
	respBody := ioutil.NopCloser(bytes.NewBufferString(""))
	httpClientMock.On("Delete", "e2t", "application/json", body).Return(&http.Response{StatusCode: http.StatusCreated, Body: respBody}, nil)

	writerMock.On("RemoveE2TInstance", E2TAddress).Return(nil)
	writerMock.On("SaveE2TAddresses", []string{E2TAddress2, E2TAddress3}).Return(nil)

	/*** nodeb 1 ***/
	nodeb1connected := *nodeb1
	nodeb1connected.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, "test1_DISCONNECTED").Return(nil)

	nodeb1NotAssociated := *nodeb1
	nodeb1NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb1NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	/*** nodeb 2 ***/
	nodeb2connected := *nodeb2
	nodeb2connected.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	nodeb2NotAssociated := *nodeb2
	nodeb2NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb2NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	/*** nodeb 5 ***/
	nodeb5connected := *nodeb5
	nodeb5connected.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, "test5_DISCONNECTED").Return(nil)

	nodeb5NotAssociated := *nodeb5
	nodeb5NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb5NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	err := shutdownManager.Shutdown(e2tInstance1)

	assert.Nil(t, err)
	readerMock.AssertExpectations(t)
	writerMock.AssertExpectations(t)
	httpClientMock.AssertExpectations(t)
}

func TestShutdownSuccess1InstanceWithoutRans(t *testing.T) {
	shutdownManager, readerMock, writerMock, httpClientMock := initE2TShutdownManagerTest(t)

	e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
	e2tInstance1.State = entities.Active
	e2tInstance1.AssociatedRanList = []string{}
	writerMock.On("SaveE2TInstance", mock.MatchedBy(func(e2tInstance *entities.E2TInstance) bool {
		return e2tInstance.Address == E2TAddress && e2tInstance.State == entities.ToBeDeleted
	})).Return(nil)

	data := models.NewRoutingManagerDeleteRequestModel(E2TAddress, nil, nil)
	marshaled, _ := json.Marshal(data)
	body := bytes.NewBuffer(marshaled)
	respBody := ioutil.NopCloser(bytes.NewBufferString(""))
	httpClientMock.On("Delete", "e2t", "application/json", body).Return(&http.Response{StatusCode: http.StatusCreated, Body: respBody}, nil)

	writerMock.On("RemoveE2TInstance", E2TAddress).Return(nil)
	readerMock.On("GetE2TAddresses").Return([]string{E2TAddress}, nil)
	writerMock.On("SaveE2TAddresses", []string{}).Return(nil)

	err := shutdownManager.Shutdown(e2tInstance1)

	assert.Nil(t, err)
	readerMock.AssertExpectations(t)
	writerMock.AssertExpectations(t)
	httpClientMock.AssertExpectations(t)
}

func TestShutdownSuccess1Instance2Rans(t *testing.T) {
	shutdownManager, readerMock, writerMock, httpClientMock := initE2TShutdownManagerTest(t)

	e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
	e2tInstance1.State = entities.Active
	e2tInstance1.AssociatedRanList = []string{"test1", "test2"}
	writerMock.On("SaveE2TInstance", mock.MatchedBy(func(e2tInstance *entities.E2TInstance) bool {
		return e2tInstance.Address == E2TAddress && e2tInstance.State == entities.ToBeDeleted
	})).Return(nil)

	nodeb1 := &entities.NodebInfo{RanName: "test1", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test1").Return(nodeb1, nil)
	nodeb2 := &entities.NodebInfo{RanName: "test2", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test2").Return(nodeb2, nil)

	data := models.NewRoutingManagerDeleteRequestModel(E2TAddress, []string{"test1", "test2"}, nil)
	marshaled, _ := json.Marshal(data)
	body := bytes.NewBuffer(marshaled)
	respBody := ioutil.NopCloser(bytes.NewBufferString(""))
	httpClientMock.On("Delete", "e2t", "application/json", body).Return(&http.Response{StatusCode: http.StatusCreated, Body: respBody}, nil)

	writerMock.On("RemoveE2TInstance", E2TAddress).Return(nil)
	readerMock.On("GetE2TAddresses").Return([]string{E2TAddress}, nil)
	writerMock.On("SaveE2TAddresses", []string{}).Return(nil)

	/*** nodeb 1 connected ***/
	nodeb1new := *nodeb1
	nodeb1new.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, "test1_DISCONNECTED").Return(nil)

	nodeb1NotAssociated := *nodeb1
	nodeb1NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb1NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	/*** nodeb 2 disconnected ***/
	nodeb2new := *nodeb2
	nodeb2new.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	nodeb2NotAssociated := *nodeb2
	nodeb2NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb2NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	err := shutdownManager.Shutdown(e2tInstance1)

	assert.Nil(t, err)
	readerMock.AssertExpectations(t)
	writerMock.AssertExpectations(t)
	httpClientMock.AssertExpectations(t)
}

func TestShutdownE2tInstanceAlreadyBeingDeleted(t *testing.T) {
	shutdownManager, readerMock, writerMock, httpClientMock := initE2TShutdownManagerTest(t)

	e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
	e2tInstance1.State = entities.ToBeDeleted
	e2tInstance1.AssociatedRanList = []string{"test1"}
	e2tInstance1.DeletionTimestamp = time.Now().UnixNano()

	err := shutdownManager.Shutdown(e2tInstance1)

	assert.Nil(t, err)
	readerMock.AssertExpectations(t)
	writerMock.AssertExpectations(t)
	httpClientMock.AssertExpectations(t)
}

func TestShutdownFailureMarkInstanceAsToBeDeleted(t *testing.T) {
	shutdownManager, readerMock, writerMock, httpClientMock := initE2TShutdownManagerTest(t)

	e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
	e2tInstance1.State = entities.Active
	e2tInstance1.AssociatedRanList = []string{"test1", "test2", "test5"}
	writerMock.On("SaveE2TInstance", mock.MatchedBy(func(e2tInstance *entities.E2TInstance) bool {
		return e2tInstance.Address == E2TAddress && e2tInstance.State == entities.ToBeDeleted
	})).Return(e2managererrors.NewRnibDbError())

	err := shutdownManager.Shutdown(e2tInstance1)

	assert.NotNil(t, err)
	readerMock.AssertExpectations(t)
	writerMock.AssertExpectations(t)
	httpClientMock.AssertExpectations(t)
}

func TestShutdownFailureRoutingManagerError(t *testing.T) {
	shutdownManager, readerMock, writerMock, httpClientMock := initE2TShutdownManagerTest(t)

	e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
	e2tInstance1.State = entities.Active
	e2tInstance1.AssociatedRanList = []string{"test1", "test2", "test5"}
	e2tInstance2 := entities.NewE2TInstance(E2TAddress2, PodName)
	e2tInstance2.State = entities.Active
	e2tInstance2.AssociatedRanList = []string{"test3"}
	e2tInstance3 := entities.NewE2TInstance(E2TAddress3, PodName)
	e2tInstance3.State = entities.Active
	e2tInstance3.AssociatedRanList = []string{"test4"}
	writerMock.On("SaveE2TInstance", mock.MatchedBy(func(e2tInstance *entities.E2TInstance) bool {
		return e2tInstance.Address == E2TAddress && e2tInstance.State == entities.ToBeDeleted
	})).Return(nil)

	nodeb1 := &entities.NodebInfo{RanName: "test1", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test1").Return(nodeb1, nil)
	nodeb2 := &entities.NodebInfo{RanName: "test2", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test2").Return(nodeb2, nil)
	nodeb5 := &entities.NodebInfo{RanName: "test5", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test5").Return(nodeb5, nil)

	e2tAddresses := []string{E2TAddress, E2TAddress2, E2TAddress3}
	readerMock.On("GetE2TAddresses").Return(e2tAddresses, nil)

	data := models.NewRoutingManagerDeleteRequestModel(E2TAddress, e2tInstance1.AssociatedRanList, nil)
	marshaled, _ := json.Marshal(data)
	body := bytes.NewBuffer(marshaled)
	respBody := ioutil.NopCloser(bytes.NewBufferString(""))
	httpClientMock.On("Delete", "e2t", "application/json", body).Return(&http.Response{StatusCode: http.StatusBadRequest, Body: respBody}, nil)

	writerMock.On("RemoveE2TInstance", E2TAddress).Return(nil)
	writerMock.On("SaveE2TAddresses", []string{E2TAddress2, E2TAddress3}).Return(nil)

	/*** nodeb 1 connected ***/
	nodeb1connected := *nodeb1
	nodeb1connected.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, "test1_DISCONNECTED").Return(nil)

	nodeb1NotAssociated := *nodeb1
	nodeb1NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb1NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	/*** nodeb 2 shutting down ***/
	nodeb2connected := *nodeb2
	nodeb2connected.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	nodeb2NotAssociated := *nodeb2
	nodeb2NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb2NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	/*** nodeb 5 connected ***/
	nodeb5connected := *nodeb5
	nodeb5connected.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, "test5_DISCONNECTED").Return(nil)

	nodeb5NotAssociated := *nodeb5
	nodeb5NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb5NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	err := shutdownManager.Shutdown(e2tInstance1)

	assert.Nil(t, err)
	readerMock.AssertExpectations(t)
	writerMock.AssertExpectations(t)
	httpClientMock.AssertExpectations(t)
}

func TestShutdownFailureInClearNodebsAssociation(t *testing.T) {
	shutdownManager, readerMock, writerMock, httpClientMock := initE2TShutdownManagerTest(t)

	e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
	e2tInstance1.State = entities.Active
	e2tInstance1.AssociatedRanList = []string{"test1", "test2"}
	writerMock.On("SaveE2TInstance", mock.MatchedBy(func(e2tInstance *entities.E2TInstance) bool {
		return e2tInstance.Address == E2TAddress && e2tInstance.State == entities.ToBeDeleted
	})).Return(nil)

	nodeb1 := &entities.NodebInfo{RanName: "test1", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test1").Return(nodeb1, nil)

	nodeb1new := *nodeb1
	nodeb1new.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, "test1_DISCONNECTED").Return(nil)

	nodeb1NotAssociated := *nodeb1
	nodeb1NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb1NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(common.NewInternalError(fmt.Errorf("for tests")))

	err := shutdownManager.Shutdown(e2tInstance1)

	assert.NotNil(t, err)
	readerMock.AssertExpectations(t)
	writerMock.AssertExpectations(t)
	httpClientMock.AssertExpectations(t)
}

func TestShutdownFailureInClearNodebsAssociation_UpdateConnectionStatus(t *testing.T) {
	shutdownManager, readerMock, writerMock, httpClientMock := initE2TShutdownManagerTest(t)

	e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
	e2tInstance1.State = entities.Active
	e2tInstance1.AssociatedRanList = []string{"test1", "test2"}
	writerMock.On("SaveE2TInstance", mock.MatchedBy(func(e2tInstance *entities.E2TInstance) bool {
		return e2tInstance.Address == E2TAddress && e2tInstance.State == entities.ToBeDeleted
	})).Return(nil)

	nodeb1 := &entities.NodebInfo{RanName: "test1", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test1").Return(nodeb1, nil)

	nodeb1new := *nodeb1
	nodeb1new.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, "test1_DISCONNECTED").Return(common.NewInternalError(fmt.Errorf("for tests")))

	err := shutdownManager.Shutdown(e2tInstance1)

	assert.NotNil(t, err)
	readerMock.AssertExpectations(t)
	writerMock.AssertExpectations(t)
	httpClientMock.AssertExpectations(t)
}

func TestShutdownResourceNotFoundErrorInGetNodeb(t *testing.T) {
	shutdownManager, readerMock, writerMock, httpClientMock := initE2TShutdownManagerTest(t)

	e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
	e2tInstance1.State = entities.Active
	e2tInstance1.AssociatedRanList = []string{"test1", "test2"}
	writerMock.On("SaveE2TInstance", mock.MatchedBy(func(e2tInstance *entities.E2TInstance) bool {
		return e2tInstance.Address == E2TAddress && e2tInstance.State == entities.ToBeDeleted
	})).Return(nil)

	nodeb1 := &entities.NodebInfo{RanName: "test1", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test1").Return(nodeb1, nil)
	var nodeb2 *entities.NodebInfo
	readerMock.On("GetNodeb", "test2").Return(nodeb2, common.NewResourceNotFoundError("for testing"))

	nodeb1new := *nodeb1
	nodeb1new.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, "test1_DISCONNECTED").Return(nil)

	nodeb1NotAssociated := *nodeb1
	nodeb1NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb1NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	err := shutdownManager.Shutdown(e2tInstance1)

	assert.NotNil(t, err)
	readerMock.AssertExpectations(t)
	writerMock.AssertExpectations(t)
	httpClientMock.AssertExpectations(t)
}

func TestShutdownResourceGeneralErrorInGetNodeb(t *testing.T) {
	shutdownManager, readerMock, writerMock, httpClientMock := initE2TShutdownManagerTest(t)

	e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
	e2tInstance1.State = entities.Active
	e2tInstance1.AssociatedRanList = []string{"test1", "test2"}
	writerMock.On("SaveE2TInstance", mock.MatchedBy(func(e2tInstance *entities.E2TInstance) bool {
		return e2tInstance.Address == E2TAddress && e2tInstance.State == entities.ToBeDeleted
	})).Return(nil)

	var nodeb1 *entities.NodebInfo
	readerMock.On("GetNodeb", "test1").Return(nodeb1, common.NewInternalError(fmt.Errorf("for testing")))
	nodeb2 := &entities.NodebInfo{RanName: "test2", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_DISCONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test2").Return(nodeb2, nil)

	data := models.NewRoutingManagerDeleteRequestModel(E2TAddress, []string{"test1", "test2"}, nil)
	marshaled, _ := json.Marshal(data)
	body := bytes.NewBuffer(marshaled)
	respBody := ioutil.NopCloser(bytes.NewBufferString(""))
	httpClientMock.On("Delete", "e2t", "application/json", body).Return(&http.Response{StatusCode: http.StatusCreated, Body: respBody}, nil)

	writerMock.On("RemoveE2TInstance", E2TAddress).Return(nil)
	readerMock.On("GetE2TAddresses").Return([]string{E2TAddress}, nil)
	writerMock.On("SaveE2TAddresses", []string{}).Return(nil)

	nodeb2new := *nodeb2
	nodeb2new.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	nodeb2NotAssociated := *nodeb2
	nodeb2NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb2NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	err := shutdownManager.Shutdown(e2tInstance1)

	assert.Nil(t, err)
	readerMock.AssertExpectations(t)
	writerMock.AssertExpectations(t)
	httpClientMock.AssertExpectations(t)

}

func TestShutdownFailureInRemoveE2TInstance(t *testing.T) {
	shutdownManager, readerMock, writerMock, httpClientMock := initE2TShutdownManagerTest(t)

	e2tInstance1 := entities.NewE2TInstance(E2TAddress, PodName)
	e2tInstance1.State = entities.Active
	e2tInstance1.AssociatedRanList = []string{"test1", "test2", "test5"}
	e2tInstance2 := entities.NewE2TInstance(E2TAddress2, PodName)
	e2tInstance2.State = entities.Active
	e2tInstance2.AssociatedRanList = []string{"test3"}
	e2tInstance3 := entities.NewE2TInstance(E2TAddress3, PodName)
	e2tInstance3.State = entities.Active
	e2tInstance3.AssociatedRanList = []string{"test4"}
	writerMock.On("SaveE2TInstance", mock.MatchedBy(func(e2tInstance *entities.E2TInstance) bool {
		return e2tInstance.Address == E2TAddress && e2tInstance.State == entities.ToBeDeleted
	})).Return(nil)

	nodeb1 := &entities.NodebInfo{RanName: "test1", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test1").Return(nodeb1, nil)
	nodeb2 := &entities.NodebInfo{RanName: "test2", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_SHUTTING_DOWN, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test2").Return(nodeb2, nil)
	nodeb5 := &entities.NodebInfo{RanName: "test5", AssociatedE2TInstanceAddress: E2TAddress, ConnectionStatus: entities.ConnectionStatus_CONNECTED, E2ApplicationProtocol: entities.E2ApplicationProtocol_X2_SETUP_REQUEST}
	readerMock.On("GetNodeb", "test5").Return(nodeb5, nil)

	data := models.NewRoutingManagerDeleteRequestModel(E2TAddress, e2tInstance1.AssociatedRanList, nil)
	marshaled, _ := json.Marshal(data)
	body := bytes.NewBuffer(marshaled)
	respBody := ioutil.NopCloser(bytes.NewBufferString(""))
	httpClientMock.On("Delete", "e2t", "application/json", body).Return(&http.Response{StatusCode: http.StatusCreated, Body: respBody}, nil)

	writerMock.On("RemoveE2TInstance", E2TAddress).Return(common.NewInternalError(fmt.Errorf("for tests")))

	/*** nodeb 1 connected ***/
	nodeb1connected := *nodeb1
	nodeb1connected.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, "test1_DISCONNECTED").Return(nil)

	nodeb1NotAssociated := *nodeb1
	nodeb1NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb1NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	/*** nodeb 2 shutting down ***/
	nodeb2connected := *nodeb2
	nodeb2connected.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	nodeb2NotAssociated := *nodeb2
	nodeb2NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb2NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	/*** nodeb 5 connected ***/
	nodeb5connected := *nodeb5
	nodeb5connected.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfoOnConnectionStatusInversion", mock.Anything, "test5_DISCONNECTED").Return(nil)

	nodeb5NotAssociated := *nodeb5
	nodeb5NotAssociated.AssociatedE2TInstanceAddress = ""
	nodeb5NotAssociated.ConnectionStatus = entities.ConnectionStatus_DISCONNECTED
	writerMock.On("UpdateNodebInfo", mock.Anything).Return(nil)

	err := shutdownManager.Shutdown(e2tInstance1)

	assert.IsType(t, &e2managererrors.RnibDbError{}, err)
	readerMock.AssertExpectations(t)
	writerMock.AssertExpectations(t)
	httpClientMock.AssertExpectations(t)
}