Function TestSyncIteratorPropagatesErrors

pkg/parquetquery/iters_test.go:208–236

(t *testing.T)

}

func TestSyncIteratorPropagatesErrors(t *testing.T) {
	type T struct{ A int }

	rows := []T{}
	count := 10_000
	for i := 0; i < count; i++ {
		rows = append(rows, T{i})
	}

	ctx, cancel := context.WithCancel(context.Background())

	pf := createFileWith(t, ctx, rows)

	iter := NewSyncIterator(ctx, pf.RowGroups(), 0)

	_, err := iter.Next()
	require.NoError(t, err)

	cancel()

	// iterate until we get an error and confirm it's because of the context cancellation
	for {
		_, err = iter.Next()
		if err != nil {
			break
		}
	}
	require.ErrorContains(t, err, "context canceled")
}

func BenchmarkColumnIterator(b *testing.B) {
	for _, tc := range iterTestCases {