53 lines
1.3 KiB
Go
53 lines
1.3 KiB
Go
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"])
|
|
}
|