package database

import (
	"testing"

	"modelRT/orm"

	"github.com/stretchr/testify/require"
)

func TestAppendBayDevCandidatesRequiresBayLinkedComponent(t *testing.T) {
	measurementSet := &orm.MeasurementSet{
		AllMeasTags: []string{"current"},
		CompTagToMeasTags: map[string][]string{
			"linked-component":   {"current"},
			"unlinked-component": {"voltage"},
		},
	}

	appendBayDevCandidates(
		measurementSet,
		[]string{"linked-component"},
		[]string{"dev_instruct", "dev_dyn_sense", "dev_fault_record"},
	)

	require.Equal(t,
		[]string{"current", "dev_instruct", "dev_dyn_sense", "dev_fault_record"},
		measurementSet.AllMeasTags,
	)
	require.Equal(t,
		[]string{"current", "dev_instruct", "dev_dyn_sense", "dev_fault_record"},
		measurementSet.CompTagToMeasTags["linked-component"],
	)
	require.Equal(t,
		[]string{"voltage"},
		measurementSet.CompTagToMeasTags["unlinked-component"],
	)
}

func TestAppendBayDevCandidatesWithoutBayLinkDoesNothing(t *testing.T) {
	measurementSet := &orm.MeasurementSet{
		AllMeasTags: []string{"current"},
		CompTagToMeasTags: map[string][]string{
			"component": {"current"},
		},
	}

	appendBayDevCandidates(measurementSet, nil, []string{"dev_instruct"})

	require.Equal(t, []string{"current"}, measurementSet.AllMeasTags)
	require.Equal(t, []string{"current"}, measurementSet.CompTagToMeasTags["component"])
}