From b10dd532008ee5ea7b994b07bb6f6e458786c0eb Mon Sep 17 00:00:00 2001
From: radhika <radhika@curoverse.com>
Date: Fri, 20 Nov 2015 13:28:04 -0500
Subject: [PATCH] 7255: update
 TestPutAndGetCollectionsWithMultipleStreamsAndBlocks to verify that expected
 number of unique block locators are created.

---
 services/datamanager/datamanager_test.go | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/services/datamanager/datamanager_test.go b/services/datamanager/datamanager_test.go
index 094cd44b0f..983e25cf01 100644
--- a/services/datamanager/datamanager_test.go
+++ b/services/datamanager/datamanager_test.go
@@ -536,7 +536,7 @@ func createMultiStreamBlockCollection(t *testing.T, data string, numStreams, num
 	defer switchToken(arvadostest.AdminToken)()
 
 	manifest := ""
-	var locators []string
+	locators := make(map[string]bool)
 	for s := 0; s < numStreams; s++ {
 		manifest += fmt.Sprintf("./stream%d ", s)
 		for b := 0; b < numBlocks; b++ {
@@ -544,7 +544,7 @@ func createMultiStreamBlockCollection(t *testing.T, data string, numStreams, num
 			if err != nil {
 				t.Fatalf("Error creating block %d in stream %d: %v", b, s, err)
 			}
-			locators = append(locators, strings.Split(locator, "+A")[0])
+			locators[strings.Split(locator, "+A")[0]] = true
 			manifest += locator + " "
 		}
 		manifest += "0:1:dummyfile.txt\n"
@@ -559,7 +559,12 @@ func createMultiStreamBlockCollection(t *testing.T, data string, numStreams, num
 		t.Fatalf("Error creating collection %v", err)
 	}
 
-	return collection["uuid"].(string), locators
+	var locs []string
+	for k, _ := range locators {
+		locs = append(locs, k)
+	}
+
+	return collection["uuid"].(string), locs
 }
 
 /*
@@ -585,6 +590,9 @@ func TestPutAndGetCollectionsWithMultipleStreamsAndBlocks(t *testing.T) {
 	if toBeDeletedCollection == "" {
 		t.Fatalf("Failed to create collection with 10 blocks")
 	}
+	if len(toBeDeletedCollectionBlocks) != 10 {
+		t.Fatalf("Not all blocks are created: expected %v, found %v", 10, len(toBeDeletedCollectionBlocks))
+	}
 
 	// create a stray block that will be backdated
 	strayOldBlock := putBlock(t, "this stray block is old")
-- 
2.30.2