github.com/dolthub/dolt/go@v0.40.5-0.20240520175717-68db7794bea6/libraries/doltcore/env/config_test.go (about)

     1  // Copyright 2019 Dolthub, Inc.
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  //     http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    12  // See the License for the specific language governing permissions and
    13  // limitations under the License.
    14  
    15  package env
    16  
    17  import (
    18  	"testing"
    19  
    20  	"github.com/stretchr/testify/assert"
    21  	"github.com/stretchr/testify/require"
    22  
    23  	"github.com/dolthub/dolt/go/libraries/utils/config"
    24  )
    25  
    26  const (
    27  	email = "bigbillieb@fake.horse"
    28  	name  = "Billy Bob"
    29  )
    30  
    31  func TestConfig(t *testing.T) {
    32  	dEnv, _ := createTestEnv(true, true)
    33  
    34  	lCfg, _ := dEnv.Config.GetConfig(LocalConfig)
    35  	gCfg, _ := dEnv.Config.GetConfig(GlobalConfig)
    36  
    37  	lCfg.SetStrings(map[string]string{config.UserEmailKey: email, config.UserNameKey: "local_override"})
    38  	gCfg.SetStrings(map[string]string{config.UserNameKey: name})
    39  
    40  	assert.Equal(t, email, dEnv.Config.GetStringOrDefault(config.UserEmailKey, "no"))
    41  	assert.Equal(t, "local_override", dEnv.Config.GetStringOrDefault(config.UserNameKey, "no"))
    42  	assert.Equal(t, "yes", dEnv.Config.GetStringOrDefault("bad_key", "yes"))
    43  
    44  	assert.Equal(t, email, dEnv.Config.IfEmptyUseConfig("", config.UserEmailKey))
    45  	assert.Equal(t, "not empty", dEnv.Config.IfEmptyUseConfig("not empty", config.UserEmailKey))
    46  
    47  	assert.Equal(t, "", dEnv.Config.IfEmptyUseConfig("", "missing"))
    48  
    49  	_, err := dEnv.Config.GetString("missing")
    50  	assert.Equal(t, config.ErrConfigParamNotFound, err)
    51  }
    52  
    53  func TestFailsafes(t *testing.T) {
    54  	dEnv, _ := createTestEnv(true, true)
    55  
    56  	lCfg, _ := dEnv.Config.GetConfig(LocalConfig)
    57  	require.NoError(t, lCfg.Unset([]string{config.UserNameKey}))
    58  
    59  	dEnv.Config.SetFailsafes(DefaultFailsafeConfig)
    60  
    61  	assert.Equal(t, DefaultEmail, dEnv.Config.GetStringOrDefault(config.UserEmailKey, "none"))
    62  	assert.Equal(t, DefaultName, dEnv.Config.GetStringOrDefault(config.UserNameKey, "none"))
    63  
    64  	dEnv.Config.SetFailsafes(map[string]string{config.UserEmailKey: "new", "abc": "def"})
    65  
    66  	assert.Equal(t, "new", dEnv.Config.GetStringOrDefault(config.UserEmailKey, "none"))
    67  	assert.Equal(t, DefaultName, dEnv.Config.GetStringOrDefault(config.UserNameKey, "none"))
    68  	assert.Equal(t, "def", dEnv.Config.GetStringOrDefault("abc", "none"))
    69  }
    70  
    71  func TestWritableDoltConfig(t *testing.T) {
    72  	dEnv, _ := createTestEnv(true, true)
    73  
    74  	localName := "Willy"
    75  
    76  	gCfg, _ := dEnv.Config.GetConfig(GlobalConfig)
    77  	lCfg, _ := dEnv.Config.GetConfig(LocalConfig)
    78  	require.NoError(t, gCfg.SetStrings(map[string]string{config.UserNameKey: name}))
    79  	require.NoError(t, lCfg.SetStrings(map[string]string{config.UserNameKey: localName}))
    80  
    81  	cfg := dEnv.Config.WriteableConfig()
    82  
    83  	assert.Equal(t, localName, cfg.GetStringOrDefault(config.UserNameKey, "none"))
    84  
    85  	require.NoError(t, cfg.SetStrings(map[string]string{"test": "abc"}))
    86  	require.NoError(t, cfg.Unset([]string{config.UserNameKey}))
    87  
    88  	assert.Equal(t, name, cfg.GetStringOrDefault(config.UserNameKey, "none"))
    89  	assert.Equal(t, "abc", cfg.GetStringOrDefault("test", "none"))
    90  
    91  	_, err := lCfg.GetString(config.UserNameKey)
    92  	assert.Equal(t, config.ErrConfigParamNotFound, err)
    93  
    94  	assert.Equal(t, name, gCfg.GetStringOrDefault(config.UserNameKey, "none"))
    95  	_, err = gCfg.GetString("test")
    96  	assert.Equal(t, config.ErrConfigParamNotFound, err)
    97  }
    98  
    99  func TestWritableDoltConfigNoLocal(t *testing.T) {
   100  	dEnv, _ := createTestEnv(true, false)
   101  
   102  	newName := "Willy"
   103  
   104  	gCfg, _ := dEnv.Config.GetConfig(GlobalConfig)
   105  	require.NoError(t, gCfg.SetStrings(map[string]string{config.UserNameKey: name, "test": "abc"}))
   106  
   107  	cfg := dEnv.Config.WriteableConfig()
   108  
   109  	assert.Equal(t, name, cfg.GetStringOrDefault(config.UserNameKey, "none"))
   110  	assert.Equal(t, "abc", cfg.GetStringOrDefault("test", "none"))
   111  
   112  	require.NoError(t, cfg.SetStrings(map[string]string{config.UserNameKey: newName}))
   113  	require.NoError(t, cfg.Unset([]string{"test"}))
   114  
   115  	assert.Equal(t, newName, cfg.GetStringOrDefault(config.UserNameKey, "none"))
   116  
   117  	_, err := cfg.GetString("test")
   118  	assert.Equal(t, config.ErrConfigParamNotFound, err)
   119  
   120  	assert.Equal(t, newName, gCfg.GetStringOrDefault(config.UserNameKey, "none"))
   121  
   122  	_, err = gCfg.GetString("test")
   123  	assert.Equal(t, config.ErrConfigParamNotFound, err)
   124  }