X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/0210746ac1fef5a588279b208313ed06e51b19de..4d7567b7b3577b561dd064d397c10a9331c3ee16:/services/keep-balance/balance_test.go

diff --git a/services/keep-balance/balance_test.go b/services/keep-balance/balance_test.go
index 423546c46a..e372d37841 100644
--- a/services/keep-balance/balance_test.go
+++ b/services/keep-balance/balance_test.go
@@ -13,7 +13,6 @@ import (
 	"time"
 
 	"git.curoverse.com/arvados.git/sdk/go/arvados"
-
 	check "gopkg.in/check.v1"
 )
 
@@ -525,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",
 		},
@@ -533,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",
 		},
@@ -545,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",
 		},
@@ -553,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",
 		},
@@ -694,7 +693,7 @@ func (bal *balancerSuite) try(c *check.C, t tester) {
 
 // srvList returns the KeepServices, sorted in rendezvous order and
 // then selected by idx. For example, srvList(3, slots{0, 1, 4})
-// returns the the first-, second-, and fifth-best servers for storing
+// returns the first-, second-, and fifth-best servers for storing
 // bal.knownBlkid(3).
 func (bal *balancerSuite) srvList(knownBlockID int, order slots) (srvs []*KeepService) {
 	for _, i := range order {