Provisioning: Validate that datasource access field equals to direct or proxy (#26440)

* Validate that datasource access field equals to allowed value (direct or proxy)
pull/26452/head
Maksim Nabokikh 5 years ago committed by GitHub
parent c298653623
commit a1f40195be
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
  1. 12
      pkg/services/provisioning/datasources/config_reader.go
  2. 8
      pkg/services/provisioning/datasources/config_reader_test.go
  3. 6
      pkg/services/provisioning/datasources/testdata/invalid-access/invalid-access.yaml

@ -7,6 +7,7 @@ import (
"strings" "strings"
"github.com/grafana/grafana/pkg/infra/log" "github.com/grafana/grafana/pkg/infra/log"
"github.com/grafana/grafana/pkg/models"
"gopkg.in/yaml.v2" "gopkg.in/yaml.v2"
) )
@ -36,7 +37,7 @@ func (cr *configReader) readConfig(path string) ([]*configs, error) {
} }
} }
err = validateDefaultUniqueness(datasources) err = cr.validateDefaultUniqueness(datasources)
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -82,7 +83,7 @@ func (cr *configReader) parseDatasourceConfig(path string, file os.FileInfo) (*c
return v0.mapToDatasourceFromConfig(apiVersion.APIVersion), nil return v0.mapToDatasourceFromConfig(apiVersion.APIVersion), nil
} }
func validateDefaultUniqueness(datasources []*configs) error { func (cr *configReader) validateDefaultUniqueness(datasources []*configs) error {
defaultCount := map[int64]int{} defaultCount := map[int64]int{}
for i := range datasources { for i := range datasources {
if datasources[i].Datasources == nil { if datasources[i].Datasources == nil {
@ -95,7 +96,12 @@ func validateDefaultUniqueness(datasources []*configs) error {
} }
if ds.Access == "" { if ds.Access == "" {
ds.Access = "proxy" ds.Access = models.DS_ACCESS_PROXY
}
if ds.Access != models.DS_ACCESS_DIRECT && ds.Access != models.DS_ACCESS_PROXY {
cr.log.Warn("invalid access value, will use 'proxy' instead", "value", ds.Access)
ds.Access = models.DS_ACCESS_PROXY
} }
if ds.IsDefault { if ds.IsDefault {

@ -22,6 +22,7 @@ var (
brokenYaml = "testdata/broken-yaml" brokenYaml = "testdata/broken-yaml"
multipleOrgsWithDefault = "testdata/multiple-org-default" multipleOrgsWithDefault = "testdata/multiple-org-default"
withoutDefaults = "testdata/appliedDefaults" withoutDefaults = "testdata/appliedDefaults"
invalidAccess = "testdata/invalid-access"
fakeRepo *fakeRepository fakeRepo *fakeRepository
) )
@ -149,6 +150,13 @@ func TestDatasourceAsConfig(t *testing.T) {
So(err, ShouldNotBeNil) So(err, ShouldNotBeNil)
}) })
Convey("invalid access should warn about invalid value and return 'proxy'", func() {
reader := &configReader{log: logger}
configs, err := reader.readConfig(invalidAccess)
So(err, ShouldBeNil)
So(configs[0].Datasources[0].Access, ShouldEqual, models.DS_ACCESS_PROXY)
})
Convey("skip invalid directory", func() { Convey("skip invalid directory", func() {
cfgProvider := &configReader{log: log.New("test logger")} cfgProvider := &configReader{log: log.New("test logger")}
cfg, err := cfgProvider.readConfig("./invalid-directory") cfg, err := cfgProvider.readConfig("./invalid-directory")

@ -0,0 +1,6 @@
apiVersion: 1
datasources:
- name: invalid-access
type: prometheus
access: INVALID_ACCESS
Loading…
Cancel
Save