Merge pull request #66 from jezell/branch-iterator

Add branch iterator

1 files changed, 26 insertions, 0 deletions

diff --git a/branch_test.go b/branch_test.go
new file mode 100644
index 0000000..44f6338
--- /dev/null
+++ b/branch_test.go
@@ -0,0 +1,26 @@
+package git
+
+import (
+	"testing"
+)
+
+func TestBranchIterator(t *testing.T) {
+
+	repo := createTestRepo(t)
+	seedTestRepo(t, repo)
+
+	i, err := repo.NewBranchIterator(BranchLocal)
+	checkFatal(t, err)
+
+	b, bt, err := i.Next()
+	checkFatal(t, err)
+	if name, _ := b.Name(); name != "master" {
+		t.Fatalf("expected master")
+	} else if bt != BranchLocal {
+		t.Fatalf("expected BranchLocal, not %v", t)
+	}
+	b, bt, err = i.Next()
+	if !IsErrorCode(err, ErrIterOver) {
+		t.Fatal("expected iterover")
+	}
+}