@@ -48,32 +48,32 @@ func TestIsValidNetworkType(t *testing.T) {
4848 }
4949}
5050
51- func TestIsValidDatastore (t * testing.T ) {
51+ func TestDatastoreType_IsValid (t * testing.T ) {
5252 t .Parallel ()
5353
5454 tests := []struct {
5555 name string
56- datastore string
56+ datastore DatastoreType
5757 expected bool
5858 }{
5959 {
6060 name : "file is valid" ,
61- datastore : "file" ,
61+ datastore : DatastoreTypeFile ,
6262 expected : true ,
6363 },
6464 {
6565 name : "catalog is valid" ,
66- datastore : "catalog" ,
66+ datastore : DatastoreTypeCatalog ,
6767 expected : true ,
6868 },
6969 {
7070 name : "invalid value" ,
71- datastore : "invalid" ,
71+ datastore : DatastoreType ( "invalid" ) ,
7272 expected : false ,
7373 },
7474 {
7575 name : "empty value" ,
76- datastore : "" ,
76+ datastore : DatastoreType ( "" ) ,
7777 expected : false ,
7878 },
7979 }
@@ -82,7 +82,7 @@ func TestIsValidDatastore(t *testing.T) {
8282 t .Run (tt .name , func (t * testing.T ) {
8383 t .Parallel ()
8484
85- assert .Equal (t , tt .expected , isValidDatastore ( tt .datastore ))
85+ assert .Equal (t , tt .expected , tt .datastore . IsValid ( ))
8686 })
8787 }
8888}
@@ -121,15 +121,15 @@ func TestEnvironment_Validate(t *testing.T) {
121121 name : "valid environment with file datastore" ,
122122 environment : Environment {
123123 NetworkTypes : []string {"testnet" },
124- Datastore : "file" ,
124+ Datastore : DatastoreTypeFile ,
125125 },
126126 wantErr : false ,
127127 },
128128 {
129129 name : "valid environment with catalog datastore" ,
130130 environment : Environment {
131131 NetworkTypes : []string {"mainnet" },
132- Datastore : "catalog" ,
132+ Datastore : DatastoreTypeCatalog ,
133133 },
134134 wantErr : false ,
135135 },
@@ -177,7 +177,7 @@ func TestEnvironment_Validate(t *testing.T) {
177177 name : "invalid datastore value" ,
178178 environment : Environment {
179179 NetworkTypes : []string {"testnet" },
180- Datastore : "invalid" ,
180+ Datastore : DatastoreType ( "invalid" ) ,
181181 },
182182 wantErr : true ,
183183 errContains : "invalid datastore value: invalid" ,
@@ -231,27 +231,27 @@ func TestLoad(t *testing.T) {
231231 // Test specific environment configurations
232232 dev := config .Environments ["development" ]
233233 assert .Equal (t , []string {"testnet" }, dev .NetworkTypes )
234- assert .Equal (t , "file" , dev .Datastore )
234+ assert .Equal (t , DatastoreTypeFile , dev .Datastore )
235235
236236 staging := config .Environments ["staging" ]
237237 assert .ElementsMatch (t , []string {"testnet" , "mainnet" }, staging .NetworkTypes )
238- assert .Equal (t , "catalog" , staging .Datastore )
238+ assert .Equal (t , DatastoreTypeCatalog , staging .Datastore )
239239
240240 prod := config .Environments ["production" ]
241241 assert .Equal (t , []string {"mainnet" }, prod .NetworkTypes )
242- assert .Equal (t , "catalog" , prod .Datastore )
242+ assert .Equal (t , DatastoreTypeCatalog , prod .Datastore )
243243
244244 local := config .Environments ["local" ]
245245 assert .Equal (t , []string {"testnet" }, local .NetworkTypes )
246- assert .Equal (t , "file" , local .Datastore )
246+ assert .Equal (t , DatastoreTypeFile , local .Datastore )
247247
248248 dudeenv := config .Environments ["dudeenv" ]
249249 assert .Equal (t , []string {"testnet" }, dudeenv .NetworkTypes )
250- assert .Equal (t , "file" , dudeenv .Datastore ) // Not set in YAML, should default to "file"
250+ assert .Equal (t , DatastoreTypeFile , dudeenv .Datastore ) // Not set in YAML, should default to "file"
251251
252252 testtest := config .Environments ["testtest" ]
253253 assert .Equal (t , []string {"mainnet" }, testtest .NetworkTypes )
254- assert .Equal (t , "catalog" , testtest .Datastore )
254+ assert .Equal (t , DatastoreTypeCatalog , testtest .Datastore )
255255 },
256256 },
257257 {
@@ -283,7 +283,7 @@ func TestLoad(t *testing.T) {
283283
284284 // dudeenv doesn't have datastore specified in YAML, should default to "file"
285285 dudeenv := config .Environments ["dudeenv" ]
286- assert .Equal (t , "file" , dudeenv .Datastore )
286+ assert .Equal (t , DatastoreTypeFile , dudeenv .Datastore )
287287 },
288288 },
289289 }
0 commit comments