656e270bd9
We also need to upgrade the linter together with the Go version, all the changes should relate to either fixing linting problems or upgrading the Go version used to build Grafana.
100 lines
2.6 KiB
Go
100 lines
2.6 KiB
Go
package setting
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil" //nolint:staticcheck // No need to change in v8.
|
|
"math/rand"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestExpandVar_EnvSuccessful(t *testing.T) {
|
|
const key = "GF_TEST_SETTING_EXPANDER_ENV"
|
|
const expected = "aurora borealis"
|
|
err := os.Setenv(key, expected)
|
|
require.NoError(t, err)
|
|
|
|
// expanded format
|
|
{
|
|
got, err := ExpandVar(fmt.Sprintf("$__env{%s}", key))
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, expected, got)
|
|
}
|
|
|
|
// short format
|
|
{
|
|
got, err := ExpandVar(fmt.Sprintf("${%s}", key))
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, expected, got)
|
|
}
|
|
}
|
|
|
|
func TestExpandVar_FileSuccessful(t *testing.T) {
|
|
f, err := ioutil.TempFile(os.TempDir(), "file expansion *")
|
|
require.NoError(t, err)
|
|
file := f.Name()
|
|
|
|
defer func() {
|
|
err := os.Remove(file)
|
|
require.NoError(t, err)
|
|
}()
|
|
|
|
_, err = f.WriteString("hello, world")
|
|
require.NoError(t, err)
|
|
err = f.Close()
|
|
require.NoError(t, err)
|
|
|
|
got, err := ExpandVar(fmt.Sprintf("$__file{%s}", file))
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "hello, world", got)
|
|
}
|
|
|
|
func TestExpandVar_FileDoesNotExist(t *testing.T) {
|
|
got, err := ExpandVar(
|
|
fmt.Sprintf("$__file{%s%sthisisnotarealfile_%d}",
|
|
os.TempDir(),
|
|
string(os.PathSeparator),
|
|
rand.Int63()),
|
|
)
|
|
assert.Error(t, err)
|
|
assert.True(t, errors.Is(err, os.ErrNotExist))
|
|
assert.Empty(t, got)
|
|
}
|
|
|
|
func TestExpanderRegex(t *testing.T) {
|
|
tests := map[string][][]string{
|
|
// we should not expand variables where there are none
|
|
"smoketest": {},
|
|
"Pa$$word{0}": {},
|
|
"$_almost{but not quite a variable}": {},
|
|
// args are required
|
|
"$__file{}": {},
|
|
|
|
// base cases
|
|
"${ENV}": {{"", "ENV"}},
|
|
"$__env{ENV}": {{"__env", "ENV"}},
|
|
"$__file{/dev/null}": {{"__file", "/dev/null"}},
|
|
"$__vault{item}": {{"__vault", "item"}},
|
|
// contains a space in the argument
|
|
"$__file{C:\\Program Files\\grafana\\something}": {{"__file", "C:\\Program Files\\grafana\\something"}},
|
|
|
|
// complex cases
|
|
"get variable from $__env{ENV}ironment": {{"__env", "ENV"}},
|
|
"$__env{VAR1} $__file{/var/lib/grafana/secrets/var1}": {{"__env", "VAR1"}, {"__file", "/var/lib/grafana/secrets/var1"}},
|
|
"$__env{$__file{this is invalid}}": {{"__env", "$__file{this is invalid"}},
|
|
}
|
|
|
|
for input, expected := range tests {
|
|
output := regex.FindAllStringSubmatch(input, -1)
|
|
require.Len(t, output, len(expected))
|
|
for i, variable := range output {
|
|
assert.Equal(t, expected[i], variable[1:])
|
|
}
|
|
}
|
|
}
|