From aa6e9b43bc89963247e9c2eb38eca47383ce91b7 Mon Sep 17 00:00:00 2001
From: Noah Hilverling <noah.hilverling@icinga.com>
Date: Wed, 29 May 2019 13:24:32 +0200
Subject: [PATCH] Utils: Add chunks_test.go

---
 utils/chunks_test.go | 51 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 51 insertions(+)
 create mode 100644 utils/chunks_test.go

diff --git a/utils/chunks_test.go b/utils/chunks_test.go
new file mode 100644
index 00000000..a61c3110
--- /dev/null
+++ b/utils/chunks_test.go
@@ -0,0 +1,51 @@
+package utils
+
+import (
+	"github.com/magiconair/properties/assert"
+	"testing"
+)
+
+func TestChunkKeys(t *testing.T) {
+	keys := []string{
+		"herp",
+		"derp",
+		"merp",
+		"berp",
+	}
+
+	ch := ChunkKeys(make(chan struct{}), keys, 2)
+	var chunks [][]string
+	for chunk := range ch {
+		chunks = append(chunks, chunk)
+	}
+
+	want := [][]string{
+		{
+			"herp",
+			"derp",
+		},
+		{
+			"merp",
+			"berp",
+		},
+	}
+
+	assert.Equal(t, chunks, want)
+
+	ch = ChunkKeys(make(chan struct{}), keys, 5)
+	chunks = nil
+	for chunk := range ch {
+		chunks = append(chunks, chunk)
+	}
+
+	want = [][]string{
+		{
+			"herp",
+			"derp",
+			"merp",
+			"berp",
+		},
+	}
+
+	assert.Equal(t, chunks, want)
+}