From 3290e2bef56c129bd3aee2c9b88ff90bfae44c84 Mon Sep 17 00:00:00 2001
From: Sebastien Coavoux <s.coavoux@free.fr>
Date: Thu, 21 Jul 2016 10:37:51 -0400
Subject: [PATCH] Enh: Test - Edit catalog_test to reproduce #1854

Signed-off-by: Sebastien Coavoux <alignak@pyseb.cx>
---
 registry/storage/catalog_test.go | 25 ++++++++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

diff --git a/registry/storage/catalog_test.go b/registry/storage/catalog_test.go
index b7288ed2..c125cb17 100644
--- a/registry/storage/catalog_test.go
+++ b/registry/storage/catalog_test.go
@@ -33,9 +33,13 @@ func setupFS(t *testing.T) *setupEnv {
 	repos := []string{
 		"foo/a",
 		"foo/b",
+		"foo-bar/a",
 		"bar/c",
 		"bar/d",
+		"bar/e",
 		"foo/d/in",
+		"foo-bar/b",
+		"test",
 	}
 
 	for _, repo := range repos {
@@ -45,9 +49,13 @@ func setupFS(t *testing.T) *setupEnv {
 	expected := []string{
 		"bar/c",
 		"bar/d",
+		"bar/e",
 		"foo/a",
 		"foo/b",
 		"foo/d/in",
+		"foo-bar/a",
+		"foo-bar/b",
+		"test",
 	}
 
 	return &setupEnv{
@@ -118,7 +126,7 @@ func TestCatalog(t *testing.T) {
 func TestCatalogInParts(t *testing.T) {
 	env := setupFS(t)
 
-	chunkLen := 2
+	chunkLen := 3
 	p := make([]string, chunkLen)
 
 	numFilled, err := env.registry.Repositories(env.ctx, p, "")
@@ -144,12 +152,23 @@ func TestCatalogInParts(t *testing.T) {
 	lastRepo = p[len(p)-1]
 	numFilled, err = env.registry.Repositories(env.ctx, p, lastRepo)
 
+	if err != io.EOF || numFilled != len(p) {
+		t.Errorf("Expected end of catalog")
+	}
+
+	if !testEq(p, env.expected[chunkLen*2:chunkLen*3], numFilled) {
+		t.Errorf("Expected catalog third chunk err")
+	}
+
+	lastRepo = p[len(p)-1]
+	numFilled, err = env.registry.Repositories(env.ctx, p, lastRepo)
+
 	if err != io.EOF {
 		t.Errorf("Catalog has more values which we aren't expecting")
 	}
 
-	if !testEq(p, env.expected[chunkLen*2:chunkLen*3-1], numFilled) {
-		t.Errorf("Expected catalog third chunk err")
+	if numFilled != 0 {
+		t.Errorf("Expected catalog fourth chunk err")
 	}
 
 }