X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/f5764a05d616e8d3c55b94503d5a1f789ac66ea7..1267366d66cadcc09972721bf7381f96f8fe73d6:/services/keep-balance/balance_test.go

diff --git a/services/keep-balance/balance_test.go b/services/keep-balance/balance_test.go
index 2259b3d8cf..e372d37841 100644
--- a/services/keep-balance/balance_test.go
+++ b/services/keep-balance/balance_test.go
@@ -524,7 +524,7 @@ func (bal *balancerSuite) TestChangeStorageClasses(c *check.C) {
 	bal.srvs[9].mounts = []*KeepMount{{
 		KeepMount: arvados.KeepMount{
 			Replication:    1,
-			StorageClasses: []string{"special"},
+			StorageClasses: map[string]bool{"special": true},
 			UUID:           "zzzzz-mount-special00000009",
 			DeviceID:       "9-special",
 		},
@@ -532,7 +532,7 @@ func (bal *balancerSuite) TestChangeStorageClasses(c *check.C) {
 	}, {
 		KeepMount: arvados.KeepMount{
 			Replication:    1,
-			StorageClasses: []string{"special", "special2"},
+			StorageClasses: map[string]bool{"special": true, "special2": true},
 			UUID:           "zzzzz-mount-special20000009",
 			DeviceID:       "9-special-and-special2",
 		},
@@ -544,7 +544,7 @@ func (bal *balancerSuite) TestChangeStorageClasses(c *check.C) {
 	bal.srvs[13].mounts = []*KeepMount{{
 		KeepMount: arvados.KeepMount{
 			Replication:    1,
-			StorageClasses: []string{"special2"},
+			StorageClasses: map[string]bool{"special2": true},
 			UUID:           "zzzzz-mount-special2000000d",
 			DeviceID:       "13-special2",
 		},
@@ -552,7 +552,7 @@ func (bal *balancerSuite) TestChangeStorageClasses(c *check.C) {
 	}, {
 		KeepMount: arvados.KeepMount{
 			Replication:    1,
-			StorageClasses: []string{"default"},
+			StorageClasses: map[string]bool{"default": true},
 			UUID:           "zzzzz-mount-00000000000000d",
 			DeviceID:       "13-default",
 		},