diff options
Diffstat (limited to 'reference_test.go')
| -rw-r--r-- | reference_test.go | 18 |
1 files changed, 18 insertions, 0 deletions
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() |