Skip to content

Commit f672d38

Browse files
committed
[no-relnote] Rename config constants
Signed-off-by: Evan Lezar <elezar@nvidia.com>
1 parent eb39b97 commit f672d38

File tree

2 files changed

+25
-7
lines changed

2 files changed

+25
-7
lines changed

internal/config/config.go

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,9 @@ import (
3232

3333
const (
3434
FilePathOverrideEnvVar = "NVIDIA_CTK_CONFIG_FILE_PATH"
35+
RelativeFilePath = "nvidia-container-runtime/config.toml"
3536

36-
configOverride = "XDG_CONFIG_HOME"
37-
configFilePath = "nvidia-container-runtime/config.toml"
37+
configRootOverride = "XDG_CONFIG_HOME"
3838

3939
nvidiaCTKExecutable = "nvidia-ctk"
4040
nvidiaCTKDefaultFilePath = "/usr/bin/nvidia-ctk"
@@ -79,11 +79,12 @@ func GetConfigFilePath() string {
7979
if configFilePathOverride := os.Getenv(FilePathOverrideEnvVar); configFilePathOverride != "" {
8080
return configFilePathOverride
8181
}
82-
if XDGConfigDir := os.Getenv(configOverride); len(XDGConfigDir) != 0 {
83-
return filepath.Join(XDGConfigDir, configFilePath)
82+
configRoot := "/etc"
83+
if XDGConfigDir := os.Getenv(configRootOverride); len(XDGConfigDir) != 0 {
84+
configRoot = XDGConfigDir
8485
}
8586

86-
return filepath.Join("/etc", configFilePath)
87+
return filepath.Join(configRoot, RelativeFilePath)
8788
}
8889

8990
// GetConfig sets up the config struct. Values are read from a toml file

internal/config/config_test.go

Lines changed: 19 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,9 +27,26 @@ import (
2727

2828
func TestGetConfigWithCustomConfig(t *testing.T) {
2929
testDir := t.TempDir()
30-
t.Setenv(configOverride, testDir)
30+
t.Setenv(configRootOverride, testDir)
3131

32-
filename := filepath.Join(testDir, configFilePath)
32+
filename := filepath.Join(testDir, RelativeFilePath)
33+
34+
// By default debug is disabled
35+
contents := []byte("[nvidia-container-runtime]\ndebug = \"/nvidia-container-toolkit.log\"")
36+
37+
require.NoError(t, os.MkdirAll(filepath.Dir(filename), 0766))
38+
require.NoError(t, os.WriteFile(filename, contents, 0600))
39+
40+
cfg, err := GetConfig()
41+
require.NoError(t, err)
42+
require.Equal(t, "/nvidia-container-toolkit.log", cfg.NVIDIAContainerRuntimeConfig.DebugFilePath)
43+
}
44+
45+
func TestGetConfigWithConfigFilePathOverride(t *testing.T) {
46+
testDir := t.TempDir()
47+
filename := filepath.Join(testDir, RelativeFilePath)
48+
49+
t.Setenv(FilePathOverrideEnvVar, filename)
3350

3451
// By default debug is disabled
3552
contents := []byte("[nvidia-container-runtime]\ndebug = \"/nvidia-container-toolkit.log\"")

0 commit comments

Comments
 (0)