X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/214ad40ef2c70a9a13817059073521f1ae4ef338..9f4f850c2818cbf5eea45e30d27b13c20bd2be0c:/services/keep-balance/balance_test.go

diff --git a/services/keep-balance/balance_test.go b/services/keep-balance/balance_test.go
index df04145b9d..f9fca1431b 100644
--- a/services/keep-balance/balance_test.go
+++ b/services/keep-balance/balance_test.go
@@ -2,7 +2,7 @@
 //
 // SPDX-License-Identifier: AGPL-3.0
 
-package main
+package keepbalance
 
 import (
 	"crypto/md5"
@@ -321,6 +321,35 @@ func (bal *balancerSuite) TestDecreaseReplTimestampCollision(c *check.C) {
 		desired:    map[string]int{"default": 2},
 		current:    slots{0, 1, 2},
 		timestamps: []int64{12345678, 10000000, 10000000}})
+	bal.try(c, tester{
+		desired:     map[string]int{"default": 0},
+		current:     slots{0, 1, 2},
+		timestamps:  []int64{12345678, 12345678, 12345678},
+		shouldTrash: slots{0},
+		shouldTrashMounts: []string{
+			bal.srvs[bal.knownRendezvous[0][0]].mounts[0].UUID}})
+	bal.try(c, tester{
+		desired:     map[string]int{"default": 2},
+		current:     slots{0, 1, 2, 5, 6},
+		timestamps:  []int64{12345678, 12345679, 10000000, 10000000, 10000000},
+		shouldTrash: slots{2},
+		shouldTrashMounts: []string{
+			bal.srvs[bal.knownRendezvous[0][2]].mounts[0].UUID}})
+	bal.try(c, tester{
+		desired:     map[string]int{"default": 2},
+		current:     slots{0, 1, 2, 5, 6},
+		timestamps:  []int64{12345678, 12345679, 12345671, 10000000, 10000000},
+		shouldTrash: slots{2, 5},
+		shouldTrashMounts: []string{
+			bal.srvs[bal.knownRendezvous[0][2]].mounts[0].UUID,
+			bal.srvs[bal.knownRendezvous[0][5]].mounts[0].UUID}})
+	bal.try(c, tester{
+		desired:     map[string]int{"default": 2},
+		current:     slots{0, 1, 2, 5, 6},
+		timestamps:  []int64{12345678, 12345679, 12345679, 10000000, 10000000},
+		shouldTrash: slots{5},
+		shouldTrashMounts: []string{
+			bal.srvs[bal.knownRendezvous[0][5]].mounts[0].UUID}})
 }
 
 func (bal *balancerSuite) TestDecreaseReplBlockTooNew(c *check.C) {