125 lines
4.3 KiB
Go
125 lines
4.3 KiB
Go
// Copyright 2019 Dolthub, Inc.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package env
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/dolthub/dolt/go/libraries/utils/config"
|
|
)
|
|
|
|
const (
|
|
email = "bigbillieb@fake.horse"
|
|
name = "Billy Bob"
|
|
)
|
|
|
|
func TestConfig(t *testing.T) {
|
|
dEnv, _ := createTestEnv(true, true)
|
|
|
|
lCfg, _ := dEnv.Config.GetConfig(LocalConfig)
|
|
gCfg, _ := dEnv.Config.GetConfig(GlobalConfig)
|
|
|
|
lCfg.SetStrings(map[string]string{config.UserEmailKey: email, config.UserNameKey: "local_override"})
|
|
gCfg.SetStrings(map[string]string{config.UserNameKey: name})
|
|
|
|
assert.Equal(t, email, dEnv.Config.GetStringOrDefault(config.UserEmailKey, "no"))
|
|
assert.Equal(t, "local_override", dEnv.Config.GetStringOrDefault(config.UserNameKey, "no"))
|
|
assert.Equal(t, "yes", dEnv.Config.GetStringOrDefault("bad_key", "yes"))
|
|
|
|
assert.Equal(t, email, dEnv.Config.IfEmptyUseConfig("", config.UserEmailKey))
|
|
assert.Equal(t, "not empty", dEnv.Config.IfEmptyUseConfig("not empty", config.UserEmailKey))
|
|
|
|
assert.Equal(t, "", dEnv.Config.IfEmptyUseConfig("", "missing"))
|
|
|
|
_, err := dEnv.Config.GetString("missing")
|
|
assert.Equal(t, config.ErrConfigParamNotFound, err)
|
|
}
|
|
|
|
func TestFailsafes(t *testing.T) {
|
|
dEnv, _ := createTestEnv(true, true)
|
|
|
|
lCfg, _ := dEnv.Config.GetConfig(LocalConfig)
|
|
require.NoError(t, lCfg.Unset([]string{config.UserNameKey}))
|
|
|
|
dEnv.Config.SetFailsafes(DefaultFailsafeConfig)
|
|
|
|
assert.Equal(t, DefaultEmail, dEnv.Config.GetStringOrDefault(config.UserEmailKey, "none"))
|
|
assert.Equal(t, DefaultName, dEnv.Config.GetStringOrDefault(config.UserNameKey, "none"))
|
|
|
|
dEnv.Config.SetFailsafes(map[string]string{config.UserEmailKey: "new", "abc": "def"})
|
|
|
|
assert.Equal(t, "new", dEnv.Config.GetStringOrDefault(config.UserEmailKey, "none"))
|
|
assert.Equal(t, DefaultName, dEnv.Config.GetStringOrDefault(config.UserNameKey, "none"))
|
|
assert.Equal(t, "def", dEnv.Config.GetStringOrDefault("abc", "none"))
|
|
}
|
|
|
|
func TestWritableDoltConfig(t *testing.T) {
|
|
dEnv, _ := createTestEnv(true, true)
|
|
|
|
localName := "Willy"
|
|
|
|
gCfg, _ := dEnv.Config.GetConfig(GlobalConfig)
|
|
lCfg, _ := dEnv.Config.GetConfig(LocalConfig)
|
|
require.NoError(t, gCfg.SetStrings(map[string]string{config.UserNameKey: name}))
|
|
require.NoError(t, lCfg.SetStrings(map[string]string{config.UserNameKey: localName}))
|
|
|
|
cfg := dEnv.Config.WriteableConfig()
|
|
|
|
assert.Equal(t, localName, cfg.GetStringOrDefault(config.UserNameKey, "none"))
|
|
|
|
require.NoError(t, cfg.SetStrings(map[string]string{"test": "abc"}))
|
|
require.NoError(t, cfg.Unset([]string{config.UserNameKey}))
|
|
|
|
assert.Equal(t, name, cfg.GetStringOrDefault(config.UserNameKey, "none"))
|
|
assert.Equal(t, "abc", cfg.GetStringOrDefault("test", "none"))
|
|
|
|
_, err := lCfg.GetString(config.UserNameKey)
|
|
assert.Equal(t, config.ErrConfigParamNotFound, err)
|
|
|
|
assert.Equal(t, name, gCfg.GetStringOrDefault(config.UserNameKey, "none"))
|
|
_, err = gCfg.GetString("test")
|
|
assert.Equal(t, config.ErrConfigParamNotFound, err)
|
|
}
|
|
|
|
func TestWritableDoltConfigNoLocal(t *testing.T) {
|
|
dEnv, _ := createTestEnv(true, false)
|
|
|
|
newName := "Willy"
|
|
|
|
gCfg, _ := dEnv.Config.GetConfig(GlobalConfig)
|
|
require.NoError(t, gCfg.SetStrings(map[string]string{config.UserNameKey: name, "test": "abc"}))
|
|
|
|
cfg := dEnv.Config.WriteableConfig()
|
|
|
|
assert.Equal(t, name, cfg.GetStringOrDefault(config.UserNameKey, "none"))
|
|
assert.Equal(t, "abc", cfg.GetStringOrDefault("test", "none"))
|
|
|
|
require.NoError(t, cfg.SetStrings(map[string]string{config.UserNameKey: newName}))
|
|
require.NoError(t, cfg.Unset([]string{"test"}))
|
|
|
|
assert.Equal(t, newName, cfg.GetStringOrDefault(config.UserNameKey, "none"))
|
|
|
|
_, err := cfg.GetString("test")
|
|
assert.Equal(t, config.ErrConfigParamNotFound, err)
|
|
|
|
assert.Equal(t, newName, gCfg.GetStringOrDefault(config.UserNameKey, "none"))
|
|
|
|
_, err = gCfg.GetString("test")
|
|
assert.Equal(t, config.ErrConfigParamNotFound, err)
|
|
}
|