From 98ff2625a4341607d4e1f8140ea0dc21547dcb60 Mon Sep 17 00:00:00 2001
From: Gregory Eremin <g@erem.in>
Date: Mon, 26 Oct 2015 13:22:01 +0300
Subject: [PATCH] Stats group tests

---
 stats/group_test.go | 70 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 70 insertions(+)
 create mode 100644 stats/group_test.go

diff --git a/stats/group_test.go b/stats/group_test.go
new file mode 100644
index 0000000..ba81331
--- /dev/null
+++ b/stats/group_test.go
@@ -0,0 +1,70 @@
+package stats
+
+import (
+	"testing"
+	"time"
+)
+
+func TestNewGroup(t *testing.T) {
+	g := NewGroup(newGroupItemMock(), newGroupItemMock())
+
+	if bc := len(g.backends); bc != 2 {
+		t.Fatalf("Expected backends to contain 2 items, got %d", bc)
+	}
+}
+
+func TestGroupAdd(t *testing.T) {
+	g := NewGroup(newGroupItemMock(), newGroupItemMock())
+
+	dur := 5 * time.Second
+	g.Add("", dur)
+
+	for _, m := range g.backends {
+		select {
+		case dur1 := <-m.(*groupItemMock).addCalls:
+			if dur1 != dur {
+				t.Errorf("Expected a 5s duration, got %d", dur1)
+			}
+		default:
+			t.Error("Mock item didn't receive an Add call")
+		}
+	}
+}
+
+func TestGroupError(t *testing.T) {
+	g := NewGroup(newGroupItemMock(), newGroupItemMock())
+
+	g.Error("")
+
+	for _, m := range g.backends {
+		select {
+		case <-m.(*groupItemMock).errorCalls:
+		default:
+			t.Error("Mock item didn't receive an Error call")
+		}
+	}
+}
+
+//
+// Mock
+//
+
+type groupItemMock struct {
+	addCalls   chan time.Duration
+	errorCalls chan struct{}
+}
+
+func (g *groupItemMock) Add(_ string, dur time.Duration) {
+	g.addCalls <- dur
+}
+
+func (g *groupItemMock) Error(_ string) {
+	g.errorCalls <- struct{}{}
+}
+
+func newGroupItemMock() *groupItemMock {
+	return &groupItemMock{
+		addCalls:   make(chan time.Duration, 1),
+		errorCalls: make(chan struct{}, 1),
+	}
+}