Test multiple multiple row iters at the same time

Signed-off-by: Javi Fontan <[email protected]>

Author: jfontan

repositories_test.go

@@ -2,6 +2,7 @@ package gitquery
 
 import (
 	"io"
+	"sync"
 	"testing"
 
 	"github.com/stretchr/testify/require"
@@ -159,6 +160,29 @@ func (d *testCommitIter) Close() error {
 	return nil
 }
 
+func testRepoIter(num int, require *require.Assertions, pool *RepositoryPool) {
+	cIter := &testCommitIter{}
+
+	rowRepoIter, err := NewRowRepoIter(pool, cIter)
+	require.Nil(err)
+
+	count := 0
+	for {
+		row, err := rowRepoIter.Next()
+		if err != nil {
+			require.Equal(io.EOF, err)
+			break
+		}
+
+		require.Nil(row)
+
+		count++
+	}
+
+	// 9 is the number of commits from the test repo
+	require.Equal(9*num, count)
+}
+
 func TestRepositoryRowIterator(t *testing.T) {
 	require := require.New(t)
 
@@ -173,24 +197,19 @@ func TestRepositoryRowIterator(t *testing.T) {
 		require.Nil(err)
 	}
 
-	cIter := &testCommitIter{}
-
-	rowRepoIter, err := NewRowRepoIter(&pool, cIter)
-	require.Nil(err)
+	testRepoIter(max, require, &pool)
 
-	count := 0
-	for {
-		row, err := rowRepoIter.Next()
-		if err != nil {
-			require.Equal(io.EOF, err)
-			break
-		}
+	// Test multiple iterators at the same time
 
-		require.Nil(row)
+	var wg sync.WaitGroup
 
-		count++
+	for i := 0; i < 4; i++ {
+		wg.Add(1)
+		go func() {
+			testRepoIter(max, require, &pool)
+			wg.Done()
+		}()
 	}
 
-	// 9 is the number of commits from the test repo
-	require.Equal(9*max, count)
+	wg.Wait()
 }