From 931f187301d0c262a4ecdded891e4fed9387b4e4 Mon Sep 17 00:00:00 2001
From: Carlos Martín Nieto <cmn@dwim.me>
Date: Tue, 21 May 2013 15:14:26 +0200
Subject: Implement a reference iterator

Wrap the reference iterators, and provide a Iter() function to get
them through a channel.
---
 reference_test.go | 92 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)

(limited to 'reference_test.go')

diff --git a/reference_test.go b/reference_test.go
index 8043833..a03f638 100644
--- a/reference_test.go
+++ b/reference_test.go
@@ -3,6 +3,7 @@ package git
 import (
 	"os"
 	"runtime"
+	"sort"
 	"testing"
 	"time"
 )
@@ -71,6 +72,97 @@ func TestRefModification(t *testing.T) {
 
 }
 
+func TestIterator(t *testing.T) {
+	repo := createTestRepo(t)
+	defer os.RemoveAll(repo.Workdir())
+
+	loc, err := time.LoadLocation("Europe/Berlin")
+	checkFatal(t, err)
+	sig := &Signature{
+		Name:  "Rand Om Hacker",
+		Email: "random@hacker.com",
+		When:  time.Date(2013, 03, 06, 14, 30, 0, 0, loc),
+	}
+
+	idx, err := repo.Index()
+	checkFatal(t, err)
+	err = idx.AddByPath("README")
+	checkFatal(t, err)
+	treeId, err := idx.WriteTree()
+	checkFatal(t, err)
+
+	message := "This is a commit\n"
+	tree, err := repo.LookupTree(treeId)
+	checkFatal(t, err)
+	commitId, err := repo.CreateCommit("HEAD", sig, sig, message, tree)
+	checkFatal(t, err)
+
+	_, err = repo.CreateReference("refs/heads/one", commitId, true)
+	checkFatal(t, err)
+
+	_, err = repo.CreateReference("refs/heads/two", commitId, true)
+	checkFatal(t, err)
+
+	_, err = repo.CreateReference("refs/heads/three", commitId, true)
+	checkFatal(t, err)
+
+	iter, err := repo.NewReferenceIterator()
+	checkFatal(t, err)
+
+	var list []string
+	expected := []string{
+		"refs/heads/master",
+		"refs/heads/one",
+		"refs/heads/three",
+		"refs/heads/two",
+	}
+
+	// test some manual iteration
+	name, err := iter.Next()
+	for err == nil {
+		list = append(list, name)
+		name, err = iter.Next()
+	}
+	if err != ErrIterOver {
+		t.Fatal("Iteration not over")
+	}
+
+
+	sort.Strings(list)
+	compareStringList(t, expected, list)
+
+	// test the channel iteration
+	list = []string{}
+	iter, err = repo.NewReferenceIterator()
+	for name := range iter.Iter() {
+		list = append(list, name)
+	}
+
+	sort.Strings(list)
+	compareStringList(t, expected, list)
+
+	iter, err = repo.NewReferenceIteratorGlob("refs/heads/t*")
+	expected = []string{
+		"refs/heads/three",
+		"refs/heads/two",
+	}
+
+	list = []string{}
+	for name := range iter.Iter() {
+		list = append(list, name)
+	}
+
+	compareStringList(t, expected, list)
+}
+
+func compareStringList(t *testing.T, expected, actual []string) {
+	for i, v := range expected {
+		if actual[i] != v {
+			t.Fatalf("Bad list")
+		}
+	}
+}
+
 func checkRefType(t *testing.T, ref *Reference, kind int) {
 	if ref.Type() == kind {
 		return
-- 
cgit v1.2.3