package resolver

/*
  TODO: TO BE DEPRECATED IN 0.25 @RS185722
*/
import (
	"context"
	"testing"

	"github.com/golang/mock/gomock"
	"github.com/stretchr/testify/assert"

	"edge-infra.dev/pkg/edge/api/graph/model"
	"edge-infra.dev/pkg/edge/api/mocks"
)

var (
	testClassificationBannerEdgeID = "9c478a9c-b536-4ea9-9f1f-144c87094f6b"

	testID          = "de4f477e-1f68-40f7-9f1f-f9c9dc08c250"
	testDescription = "test-description-1"
	testPod         = "test-pod1"
	testContainer   = "test-container1"
	testType        = "test-type1"
	testClass       = model.LogClassSelectionAudit
	testPattern     = "test-pattern1"
)

func TestCreateClassification(t *testing.T) {
	mock := gomock.NewController(t)
	logClassificationService := mocks.NewMockLogClassificationService(mock)

	createClassificationInput := model.CreateClassificationInput{
		Description: testDescription,
		Pod:         testPod,
		Container:   testContainer,
		Type:        testType,
		Class:       testClass,
		Pattern:     testPattern,
	}

	logClassificationService.EXPECT().CreateLogClassification(context.Background(), testClassificationBannerEdgeID, "", createClassificationInput).
		Return(&model.LogClassification{}, nil)

	classifcation, err := logClassificationService.CreateLogClassification(context.Background(), testClassificationBannerEdgeID, "", createClassificationInput)

	assert.NoError(t, err)
	assert.Equal(t, &model.LogClassification{}, classifcation)
}
func TestUpdateClassification(t *testing.T) {
	mock := gomock.NewController(t)
	logClassificationService := mocks.NewMockLogClassificationService(mock)

	updateClassificationInput := &model.UpdateClassificationInput{
		Description: &testDescription,
		Pod:         &testPod,
		Container:   &testContainer,
		Type:        &testType,
		Class:       &testClass,
		Pattern:     &testPattern,
	}

	logClassificationService.EXPECT().UpdateLogClassification(context.Background(), testClassificationBannerEdgeID, testID, updateClassificationInput).
		Return(&model.LogClassification{}, nil)

	classifcation, err := logClassificationService.UpdateLogClassification(context.Background(), testClassificationBannerEdgeID, testID, updateClassificationInput)

	assert.NoError(t, err)
	assert.Equal(t, &model.LogClassification{}, classifcation)
}

func TestGetClassification(t *testing.T) {
	mock := gomock.NewController(t)
	logClassificationService := mocks.NewMockLogClassificationService(mock)

	logClassificationService.EXPECT().GetLogClassification(context.Background(), testClassificationBannerEdgeID, testID).
		Return(&model.LogClassification{}, nil)

	classifcation, err := logClassificationService.GetLogClassification(context.Background(), testClassificationBannerEdgeID, testID)

	assert.NoError(t, err)
	assert.Equal(t, &model.LogClassification{}, classifcation)
}

func TestGetClassificationByBanner(t *testing.T) {
	mock := gomock.NewController(t)
	logClassificationService := mocks.NewMockLogClassificationService(mock)

	logClassificationService.EXPECT().GetLogClassificationsByBanner(context.Background(), testClassificationBannerEdgeID).
		Return([]*model.LogClassification{}, nil)

	classifcation, err := logClassificationService.GetLogClassificationsByBanner(context.Background(), testClassificationBannerEdgeID)

	assert.NoError(t, err)
	assert.Equal(t, []*model.LogClassification{}, classifcation)
}
func TestDeleteClassification(t *testing.T) {
	mock := gomock.NewController(t)
	logClassificationService := mocks.NewMockLogClassificationService(mock)

	logClassificationService.EXPECT().DeleteLogClassification(context.Background(), testClassificationBannerEdgeID, testID).
		Return(true, nil)

	classifcation, err := logClassificationService.DeleteLogClassification(context.Background(), testClassificationBannerEdgeID, testID)

	assert.NoError(t, err)
	assert.True(t, classifcation)
}