// Copyright The OpenTelemetry Authors
//
// 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 testdata

import (
	"testing"

	"github.com/stretchr/testify/assert"

	"go.opentelemetry.io/collector/consumer/pdata"
	otlplogs "go.opentelemetry.io/collector/internal/data/opentelemetry-proto-gen/logs/v1"
)

type logTestCase struct {
	name string
	ld   pdata.Logs
	otlp []*otlplogs.ResourceLogs
}

func generateAllLogTestCases() []logTestCase {
	return []logTestCase{
		{
			name: "empty",
			ld:   GenerateLogDataEmpty(),
			otlp: generateLogOtlpEmpty(),
		},
		{
			name: "one-empty-resource-logs",
			ld:   GenerateLogDataOneEmptyResourceLogs(),
			otlp: generateLogOtlpOneEmptyResourceLogs(),
		},
		{
			name: "one-empty-one-nil-resource-logs",
			ld:   GenerateLogDataOneEmptyOneNilResourceLogs(),
			otlp: generateLogOtlpOneEmptyOneNilResourceLogs(),
		},
		{
			name: "no-log-records",
			ld:   GenerateLogDataNoLogRecords(),
			otlp: generateLogOtlpNoLogRecords(),
		},
		{
			name: "one-empty-log-record",
			ld:   GenerateLogDataOneEmptyLogs(),
			otlp: generateLogOtlpOneEmptyLogs(),
		},
		{
			name: "one-empty-one-nil-log-record",
			ld:   GenerateLogDataOneEmptyOneNilLogRecord(),
			otlp: generateLogOtlpOneEmptyOneNilLogRecord(),
		},
		{
			name: "one-log-record-no-resource",
			ld:   GenerateLogDataOneLogNoResource(),
			otlp: generateLogOtlpOneLogNoResource(),
		},
		{
			name: "one-log-record",
			ld:   GenerateLogDataOneLog(),
			otlp: generateLogOtlpOneLog(),
		},
		{
			name: "one-log-record-one-nil",
			ld:   GenerateLogDataOneLogOneNil(),
			otlp: generateLogOtlpOneLogOneNil(),
		},
		{
			name: "two-records-same-resource",
			ld:   GenerateLogDataTwoLogsSameResource(),
			otlp: GenerateLogOtlpSameResourceTwoLogs(),
		},
		{
			name: "two-records-same-resource-one-different",
			ld:   GenerateLogDataTwoLogsSameResourceOneDifferent(),
			otlp: generateLogOtlpTwoLogsSameResourceOneDifferent(),
		},
	}
}

func TestToFromOtlpLog(t *testing.T) {
	allTestCases := generateAllLogTestCases()
	// Ensure NumLogTests gets updated.
	assert.EqualValues(t, NumLogTests, len(allTestCases))
	for i := range allTestCases {
		test := allTestCases[i]
		t.Run(test.name, func(t *testing.T) {
			ld := pdata.LogsFromOtlp(test.otlp)
			assert.EqualValues(t, test.ld, ld)
			otlp := pdata.LogsToOtlp(ld)
			assert.EqualValues(t, test.otlp, otlp)
		})
	}
}

func TestToFromOtlpLogWithNils(t *testing.T) {
	md := GenerateLogDataOneEmptyOneNilResourceLogs()
	assert.EqualValues(t, 2, md.ResourceLogs().Len())
	assert.False(t, md.ResourceLogs().At(0).IsNil())
	assert.True(t, md.ResourceLogs().At(1).IsNil())

	md = GenerateLogDataOneEmptyOneNilLogRecord()
	rs := md.ResourceLogs().At(0)
	assert.EqualValues(t, 2, rs.InstrumentationLibraryLogs().At(0).Logs().Len())
	assert.False(t, rs.InstrumentationLibraryLogs().At(0).Logs().At(0).IsNil())
	assert.True(t, rs.InstrumentationLibraryLogs().At(0).Logs().At(1).IsNil())

	md = GenerateLogDataOneLogOneNil()
	rl0 := md.ResourceLogs().At(0)
	assert.EqualValues(t, 2, rl0.InstrumentationLibraryLogs().At(0).Logs().Len())
	assert.False(t, rl0.InstrumentationLibraryLogs().At(0).Logs().At(0).IsNil())
	assert.True(t, rl0.InstrumentationLibraryLogs().At(0).Logs().At(1).IsNil())
}
