Skip to content

Commit abf6b8a

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

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
@@ -31,9 +31,9 @@ import (
3131

3232
const (
3333
FilePathOverrideEnvVar = "NVCTK_CONFIG_FILE_PATH"
34+
RelativeFilePath = "nvidia-container-runtime/config.toml"
3435

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

3838
nvidiaCTKExecutable = "nvidia-ctk"
3939
nvidiaCTKDefaultFilePath = "/usr/bin/nvidia-ctk"
@@ -76,11 +76,12 @@ func GetConfigFilePath() string {
7676
if configFilePathOverride := os.Getenv(FilePathOverrideEnvVar); configFilePathOverride != "" {
7777
return configFilePathOverride
7878
}
79-
if XDGConfigDir := os.Getenv(configOverride); len(XDGConfigDir) != 0 {
80-
return filepath.Join(XDGConfigDir, configFilePath)
79+
configRoot := "/etc"
80+
if XDGConfigDir := os.Getenv(configRootOverride); len(XDGConfigDir) != 0 {
81+
configRoot = XDGConfigDir
8182
}
8283

83-
return filepath.Join("/etc", configFilePath)
84+
return filepath.Join(configRoot, RelativeFilePath)
8485
}
8586

8687
// 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)