mirror of
https://github.com/grafana/grafana.git
synced 2025-02-25 18:55:37 -06:00
67 lines
2.7 KiB
Go
67 lines
2.7 KiB
Go
package rules
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana/pkg/infra/log"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const (
|
|
testFileBrokenYAML = "./testdata/broken-yaml"
|
|
testFileCorrectProperties = "./testdata/correct-properties"
|
|
testFileCorrectPropertiesWithOrg = "./testdata/correct-properties-with-org"
|
|
testFileEmptyFile = "./testdata/empty-file"
|
|
testFileEmptyFolder = "./testdata/empty-folder"
|
|
testFileMultipleRules = "./testdata/multiple-rules"
|
|
testFileMultipleFiles = "./testdata/multiple-files"
|
|
testFileSupportedFiletypes = "./testdata/supported-filetypes"
|
|
)
|
|
|
|
func TestConfigReader(t *testing.T) {
|
|
configReader := newRulesConfigReader(log.NewNopLogger())
|
|
ctx := context.Background()
|
|
t.Run("a broken YAML file should error", func(t *testing.T) {
|
|
_, err := configReader.readConfig(ctx, testFileBrokenYAML)
|
|
require.Error(t, err)
|
|
})
|
|
t.Run("a rule file with correct properties should not error", func(t *testing.T) {
|
|
ruleFiles, err := configReader.readConfig(ctx, testFileCorrectProperties)
|
|
require.NoError(t, err)
|
|
t.Run("when no organization is present it should be set to 1", func(t *testing.T) {
|
|
require.Equal(t, int64(1), ruleFiles[0].Groups[0].Rules[0].OrgID)
|
|
})
|
|
})
|
|
t.Run("a rule file with correct properties and specific org should not error", func(t *testing.T) {
|
|
ruleFiles, err := configReader.readConfig(ctx, testFileCorrectPropertiesWithOrg)
|
|
require.NoError(t, err)
|
|
t.Run("when an organization is set it should not overwrite if with the default of 1", func(t *testing.T) {
|
|
require.Equal(t, int64(1337), ruleFiles[0].Groups[0].Rules[0].OrgID)
|
|
})
|
|
})
|
|
t.Run("an empty rule file should not make the config reader error", func(t *testing.T) {
|
|
_, err := configReader.readConfig(ctx, testFileEmptyFile)
|
|
require.NoError(t, err)
|
|
})
|
|
t.Run("an empty folder should not make the config reader error", func(t *testing.T) {
|
|
_, err := configReader.readConfig(ctx, testFileEmptyFolder)
|
|
require.NoError(t, err)
|
|
})
|
|
t.Run("the config reader should be able to read multiple files in the folder", func(t *testing.T) {
|
|
ruleFiles, err := configReader.readConfig(ctx, testFileMultipleFiles)
|
|
require.NoError(t, err)
|
|
require.Len(t, ruleFiles, 2)
|
|
})
|
|
t.Run("the config reader should be able to read multiple rule groups", func(t *testing.T) {
|
|
ruleFiles, err := configReader.readConfig(ctx, testFileMultipleRules)
|
|
require.NoError(t, err)
|
|
require.Len(t, ruleFiles[0].Groups, 2)
|
|
})
|
|
t.Run("the config reader should support .yaml,.yml and .json files", func(t *testing.T) {
|
|
ruleFiles, err := configReader.readConfig(ctx, testFileSupportedFiletypes)
|
|
require.NoError(t, err)
|
|
require.Len(t, ruleFiles, 3)
|
|
})
|
|
}
|