From 4abda3a60b31101d95d9c82a82691016b77a7ade Mon Sep 17 00:00:00 2001
From: Carlos Martín Nieto <cmn@dwim.me>
Date: Fri, 9 Aug 2013 18:22:26 +0200
Subject: reference: get references out of the iterator

Allow getting references out of the iterator instead of just names.
---
 reference_test.go | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

(limited to 'reference_test.go')

diff --git a/reference_test.go b/reference_test.go
index 77afaeb..eb9bb3f 100644
--- a/reference_test.go
+++ b/reference_test.go
@@ -112,6 +112,24 @@ func TestIterator(t *testing.T) {
 	sort.Strings(list)
 	compareStringList(t, expected, list)
 
+	// test the iterator for full refs, rather than just names
+	iter, err = repo.NewReferenceIterator()
+	checkFatal(t, err)
+	count := 0
+	_, err = iter.Next()
+	for err == nil {
+		count++
+		_, err = iter.Next()
+	}
+	if err != ErrIterOver {
+		t.Fatal("Iteration not over")
+	}
+
+	if count != 4 {
+		t.Fatalf("Wrong number of references returned %v", count)
+	}
+
+
 	// test the channel iteration
 	list = []string{}
 	iter, err = repo.NewReferenceIterator()
-- 
cgit v1.2.3