package handlers

import (
	"e2mgr/configuration"
	"e2mgr/e2managererrors"
	"e2mgr/mocks"
	"e2mgr/models"
	"e2mgr/rNibWriter"
	"e2mgr/rmrCgo"
	"fmt"
	"gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/common"
	"gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/entities"
	"gerrit.o-ran-sc.org/r/ric-plt/nodeb-rnib.git/reader"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"
	"testing"
	"time"
)


func TestHandleSuccessfulDefaultCause(t *testing.T){
	log := initLog(t)

	curTime := time.Now()
	ranName := "test1"

	readerMock := &mocks.RnibReaderMock{}
	readerProvider := func() reader.RNibReader {
		return readerMock
	}
	writerMock := &mocks.RnibWriterMock{}
	writerProvider := func() rNibWriter.RNibWriter {
		return writerMock
	}
	// o&m intervention
	payload:= []byte {0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x64}
	xaction := []byte(ranName)
	msg:= rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
	rmrMessengerMock := &mocks.RmrMessengerMock{}
	rmrMessengerMock.On("SendMsg",msg,mock.Anything).Return(msg,nil)

	config := configuration.ParseConfiguration()
	rmrService:=getRmrService(rmrMessengerMock, log)
	handler := NewX2ResetRequestHandler(rmrService, config, writerProvider, readerProvider)

	var nodeb = &entities.NodebInfo{ConnectionStatus:  entities.ConnectionStatus_CONNECTED }
	readerMock.On("GetNodeb",ranName).Return(nodeb, nil)

	actual := handler.Handle(log, models.ResetRequest{RanName: ranName , StartTime: curTime})

	assert.Nil(t, actual)

	v, ok := rmrService.E2sessions[ranName]
	assert.True(t, ok)

	assert.Equal(t, v.Request.RanName, ranName)

	assert.Equal(t, v.SessionStart ,curTime)
}

func TestHandleSuccessfulRequestedCause(t *testing.T){
	log := initLog(t)

	curTime := time.Now()
	ranName := "test1"

	readerMock := &mocks.RnibReaderMock{}
	readerProvider := func() reader.RNibReader {
		return readerMock
	}
	writerMock := &mocks.RnibWriterMock{}
	writerProvider := func() rNibWriter.RNibWriter {
		return writerMock
	}

	payload:= []byte {0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x40}
	xaction := []byte(ranName)
	msg:= rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
	rmrMessengerMock := &mocks.RmrMessengerMock{}
	rmrMessengerMock.On("SendMsg",msg,mock.Anything).Return(msg,nil)

	config := configuration.ParseConfiguration()
	rmrService:=getRmrService(rmrMessengerMock, log)
	handler := NewX2ResetRequestHandler(rmrService, config, writerProvider, readerProvider)

	var nodeb = &entities.NodebInfo{ConnectionStatus:  entities.ConnectionStatus_CONNECTED }
	readerMock.On("GetNodeb",ranName).Return(nodeb, nil)

	actual := handler.Handle(log, models.ResetRequest{RanName: ranName , Cause:"protocol:transfer-syntax-error", StartTime: curTime})

	assert.Nil(t, actual)

	v, ok := rmrService.E2sessions[ranName]
	assert.True(t, ok)

	assert.Equal(t, v.Request.RanName, ranName)

	assert.Equal(t, v.SessionStart ,curTime)
}

func TestHandleFailureUnknownCause(t *testing.T){
	log := initLog(t)

	curTime := time.Now()
	ranName := "test1"

	readerMock := &mocks.RnibReaderMock{}
	readerProvider := func() reader.RNibReader {
		return readerMock
	}
	writerMock := &mocks.RnibWriterMock{}
	writerProvider := func() rNibWriter.RNibWriter {
		return writerMock
	}


	rmrMessengerMock := &mocks.RmrMessengerMock{}


	config := configuration.ParseConfiguration()
	rmrService:=getRmrService(rmrMessengerMock, log)
	handler := NewX2ResetRequestHandler(rmrService, config, writerProvider, readerProvider)

	var nodeb = &entities.NodebInfo{ConnectionStatus:  entities.ConnectionStatus_CONNECTED }
	readerMock.On("GetNodeb",ranName).Return(nodeb, nil)

	actual := handler.Handle(log, models.ResetRequest{RanName: ranName , Cause:"XXX", StartTime: curTime})

	assert.IsType(t, e2managererrors.NewRequestValidationError(), actual)

	_, ok := rmrService.E2sessions[ranName]
	assert.False(t, ok)
}

func TestHandleFailureWrongState(t *testing.T){
	log := initLog(t)

	curTime := time.Now()
	ranName := "test1"

	readerMock := &mocks.RnibReaderMock{}
	readerProvider := func() reader.RNibReader {
		return readerMock
	}
	writerMock := &mocks.RnibWriterMock{}
	writerProvider := func() rNibWriter.RNibWriter {
		return writerMock
	}


	rmrMessengerMock := &mocks.RmrMessengerMock{}


	config := configuration.ParseConfiguration()
	rmrService:=getRmrService(rmrMessengerMock, log)
	handler := NewX2ResetRequestHandler(rmrService, config, writerProvider, readerProvider)

	var nodeb = &entities.NodebInfo{ConnectionStatus:  entities.ConnectionStatus_DISCONNECTED }
	readerMock.On("GetNodeb",ranName).Return(nodeb, nil)

	actual := handler.Handle(log, models.ResetRequest{RanName: ranName ,  StartTime: curTime})

	assert.IsType(t, e2managererrors.NewWrongStateError(entities.ConnectionStatus_name[int32(nodeb.ConnectionStatus)]), actual)

	_, ok := rmrService.E2sessions[ranName]
	assert.False(t, ok)
}



func TestHandleFailureRanNotFound(t *testing.T){
	log := initLog(t)

	curTime := time.Now()
	ranName := "test1"

	readerMock := &mocks.RnibReaderMock{}
	readerProvider := func() reader.RNibReader {
		return readerMock
	}
	writerMock := &mocks.RnibWriterMock{}
	writerProvider := func() rNibWriter.RNibWriter {
		return writerMock
	}


	rmrMessengerMock := &mocks.RmrMessengerMock{}


	config := configuration.ParseConfiguration()
	rmrService:=getRmrService(rmrMessengerMock, log)
	handler := NewX2ResetRequestHandler(rmrService, config, writerProvider, readerProvider)

	readerMock.On("GetNodeb",ranName).Return(&entities.NodebInfo{}, common.NewResourceNotFoundError(fmt.Errorf("nodeb not found")))

	actual := handler.Handle(log, models.ResetRequest{RanName: ranName ,  StartTime: curTime})

	assert.IsType(t, e2managererrors.NewResourceNotFoundError(), actual)

	_, ok := rmrService.E2sessions[ranName]
	assert.False(t, ok)
}


func TestHandleFailureRnibError(t *testing.T){
	log := initLog(t)

	curTime := time.Now()
	ranName := "test1"

	readerMock := &mocks.RnibReaderMock{}
	readerProvider := func() reader.RNibReader {
		return readerMock
	}
	writerMock := &mocks.RnibWriterMock{}
	writerProvider := func() rNibWriter.RNibWriter {
		return writerMock
	}


	rmrMessengerMock := &mocks.RmrMessengerMock{}


	config := configuration.ParseConfiguration()
	rmrService:=getRmrService(rmrMessengerMock, log)
	handler := NewX2ResetRequestHandler(rmrService, config, writerProvider, readerProvider)

	readerMock.On("GetNodeb",ranName).Return(&entities.NodebInfo{}, common.NewInternalError(fmt.Errorf("internal error")))

	actual := handler.Handle(log, models.ResetRequest{RanName: ranName ,  StartTime: curTime})

	assert.IsType(t, e2managererrors.NewRnibDbError(), actual)

	_, ok := rmrService.E2sessions[ranName]
	assert.False(t, ok)
}


func TestHandleFailureRmrError(t *testing.T){
	log := initLog(t)

	curTime := time.Now()
	ranName := "test1"

	readerMock := &mocks.RnibReaderMock{}
	readerProvider := func() reader.RNibReader {
		return readerMock
	}
	writerMock := &mocks.RnibWriterMock{}
	writerProvider := func() rNibWriter.RNibWriter {
		return writerMock
	}
	// o&m intervention
	payload:= []byte {0x00, 0x07, 0x00, 0x08, 0x00, 0x00, 0x01, 0x00, 0x05, 0x40, 0x01, 0x64}
	xaction := []byte(ranName)
	msg:= rmrCgo.NewMBuf(rmrCgo.RIC_X2_RESET, len(payload), ranName, &payload, &xaction)
	rmrMessengerMock := &mocks.RmrMessengerMock{}
	rmrMessengerMock.On("SendMsg",msg,mock.Anything).Return(&rmrCgo.MBuf{},fmt.Errorf("rmr error"))

	config := configuration.ParseConfiguration()
	rmrService:=getRmrService(rmrMessengerMock, log)
	handler := NewX2ResetRequestHandler(rmrService, config, writerProvider, readerProvider)

	var nodeb = &entities.NodebInfo{ConnectionStatus:  entities.ConnectionStatus_CONNECTED }
	readerMock.On("GetNodeb",ranName).Return(nodeb, nil)

	actual := handler.Handle(log, models.ResetRequest{RanName: ranName , StartTime: curTime})

	assert.IsType(t, e2managererrors.NewRmrError(), actual)

	_, ok := rmrService.E2sessions[ranName]
	assert.False(t, ok)
}