From 5f01bd7abdd39b4d8c701c8a73f3c3e49fcd70b9 Mon Sep 17 00:00:00 2001
From: Jesse Ezell <jesse.ezell@neudesic.com>
Date: Wed, 12 Mar 2014 15:49:11 -0700
Subject: add branch iterator / remove useless repo from reference iterator

---
 branch_test.go | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 branch_test.go

(limited to 'branch_test.go')

diff --git a/branch_test.go b/branch_test.go
new file mode 100644
index 0000000..e95c877
--- /dev/null
+++ b/branch_test.go
@@ -0,0 +1,28 @@
+package git
+
+import (
+	"testing"
+)
+
+func Test_List_Branches(t *testing.T) {
+
+	repo := createTestRepo(t)
+	seedTestRepo(t, repo)
+
+	i, err := repo.NewBranchIterator(BranchLocal)
+	checkFatal(t, err)
+
+	ref, err := i.Next()
+	checkFatal(t, err)
+	if ref.Name() != "refs/heads/master" {
+		t.Fatalf("expected refs/heads/master, not %v", ref.Name())
+	}
+	ref, err = i.Next()
+	if ref != nil {
+		t.Fatal("expected nil")
+	}
+
+	if err != ErrIterOver {
+		t.Fatal("expected iterover")
+	}
+}
-- 
cgit v1.2.3