X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/f5162efc3578d0420e869e3fcbda46454a855909..73d113eab7fef74d9519be5236e89b48aeb2eab2:/services/keep-balance/balance_run_test.go

diff --git a/services/keep-balance/balance_run_test.go b/services/keep-balance/balance_run_test.go
index 18a8bdcf47..4e2c6803ca 100644
--- a/services/keep-balance/balance_run_test.go
+++ b/services/keep-balance/balance_run_test.go
@@ -87,20 +87,24 @@ var stubServices = []arvados.KeepService{
 
 var stubMounts = map[string][]arvados.KeepMount{
 	"keep0.zzzzz.arvadosapi.com:25107": {{
-		UUID:     "zzzzz-ivpuk-000000000000000",
-		DeviceID: "keep0-vol0",
+		UUID:           "zzzzz-ivpuk-000000000000000",
+		DeviceID:       "keep0-vol0",
+		StorageClasses: map[string]bool{"default": true},
 	}},
 	"keep1.zzzzz.arvadosapi.com:25107": {{
-		UUID:     "zzzzz-ivpuk-100000000000000",
-		DeviceID: "keep1-vol0",
+		UUID:           "zzzzz-ivpuk-100000000000000",
+		DeviceID:       "keep1-vol0",
+		StorageClasses: map[string]bool{"default": true},
 	}},
 	"keep2.zzzzz.arvadosapi.com:25107": {{
-		UUID:     "zzzzz-ivpuk-200000000000000",
-		DeviceID: "keep2-vol0",
+		UUID:           "zzzzz-ivpuk-200000000000000",
+		DeviceID:       "keep2-vol0",
+		StorageClasses: map[string]bool{"default": true},
 	}},
 	"keep3.zzzzz.arvadosapi.com:25107": {{
-		UUID:     "zzzzz-ivpuk-300000000000000",
-		DeviceID: "keep3-vol0",
+		UUID:           "zzzzz-ivpuk-300000000000000",
+		DeviceID:       "keep3-vol0",
+		StorageClasses: map[string]bool{"default": true},
 	}},
 }