X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/a7a482db3954fa6470be74f0e00f6e1e105e0b6c..6d8326a0f1521b0d7c49cd90dc775d409906bc49:/services/keep-balance/balance_test.go

diff --git a/services/keep-balance/balance_test.go b/services/keep-balance/balance_test.go
index 5bc66dbf3f..c529ac150e 100644
--- a/services/keep-balance/balance_test.go
+++ b/services/keep-balance/balance_test.go
@@ -85,7 +85,8 @@ func (bal *balancerSuite) SetUpTest(c *check.C) {
 		}
 		srv.mounts = []*KeepMount{{
 			KeepMount: arvados.KeepMount{
-				UUID: fmt.Sprintf("zzzzz-mount-%015x", i),
+				UUID:           fmt.Sprintf("zzzzz-mount-%015x", i),
+				StorageClasses: map[string]bool{"default": true},
 			},
 			KeepService: srv,
 		}}
@@ -166,10 +167,11 @@ func (bal *balancerSuite) testMultipleViews(c *check.C, readonly bool) {
 		srv.mounts[0].KeepMount.DeviceID = fmt.Sprintf("writable-by-srv-%x", i)
 		srv.mounts = append(srv.mounts, &KeepMount{
 			KeepMount: arvados.KeepMount{
-				DeviceID:    fmt.Sprintf("writable-by-srv-%x", (i+1)%len(bal.srvs)),
-				UUID:        fmt.Sprintf("zzzzz-mount-%015x", i<<16),
-				ReadOnly:    readonly,
-				Replication: 1,
+				DeviceID:       fmt.Sprintf("writable-by-srv-%x", (i+1)%len(bal.srvs)),
+				UUID:           fmt.Sprintf("zzzzz-mount-%015x", i<<16),
+				ReadOnly:       readonly,
+				Replication:    1,
+				StorageClasses: map[string]bool{"default": true},
 			},
 			KeepService: srv,
 		})