@@ -81,7 +81,7 @@ func TestStore_GetAlertConfigs(t *testing.T) {
81
81
// The storage contains some configs.
82
82
{
83
83
user1Cfg := prepareAlertmanagerConfig ("user-1" )
84
- user1Dir , user1TemplateDir := prepareUserDir (t , storeDir , "user-1" )
84
+ user1Dir , user1TemplateDir := prepareUserDir (t , storeDir , true , "user-1" )
85
85
require .NoError (t , os .WriteFile (filepath .Join (user1Dir , "user-1.yaml" ), []byte (user1Cfg ), os .ModePerm ))
86
86
87
87
require .NoError (t , os .WriteFile (filepath .Join (user1TemplateDir , "template.tpl" ), []byte ("testTemplate" ), os .ModePerm ))
@@ -95,7 +95,7 @@ func TestStore_GetAlertConfigs(t *testing.T) {
95
95
96
96
// Add another user config.
97
97
user2Cfg := prepareAlertmanagerConfig ("user-2" )
98
- user2Dir , _ := prepareUserDir (t , storeDir , "user-2" )
98
+ user2Dir , _ := prepareUserDir (t , storeDir , false , "user-2" )
99
99
require .NoError (t , os .WriteFile (filepath .Join (user2Dir , "user-2.yaml" ), []byte (user2Cfg ), os .ModePerm ))
100
100
101
101
configs , err = store .GetAlertConfigs (ctx , []string {"user-1" , "user-2" })
@@ -107,11 +107,13 @@ func TestStore_GetAlertConfigs(t *testing.T) {
107
107
}
108
108
}
109
109
110
- func prepareUserDir (t * testing.T , storeDir string , user string ) (userDir string , templateDir string ) {
110
+ func prepareUserDir (t * testing.T , storeDir string , createTemplateDir bool , user string ) (userDir string , templateDir string ) {
111
111
userDir = filepath .Join (storeDir , user )
112
112
templateDir = filepath .Join (userDir , templatesDir )
113
113
require .NoError (t , os .MkdirAll (userDir , os .ModePerm ))
114
- require .NoError (t , os .MkdirAll (templateDir , os .ModePerm ))
114
+ if createTemplateDir {
115
+ require .NoError (t , os .MkdirAll (templateDir , os .ModePerm ))
116
+ }
115
117
return
116
118
}
117
119
0 commit comments