Function TestLeftJoinDown

pkg/parquetquery/iters_test.go:496–600

(t *testing.T)

}

func TestLeftJoinDown(t *testing.T) {
	iter1DefLevel := 2
	iter1 := &testIterator{rows: []IteratorResult{
		testResult("x", "A", 0, 0, 1),
		testResult("x", "B", 0, 0, 2),
		testResult("x", "C", 0, 0, 3),
	}}

	// Defined at level iter2DefLevel
	iter2DefLevel := 3
	iter2 := &testIterator{rows: []IteratorResult{
		testResult("y", "A1", 0, 0, 1, 0),
		testResult("y", "A2", 0, 0, 1, 1),
		testResult("y", "B1", 0, 0, 2, 0),
		testResult("y", "B2", 0, 0, 2, 1),
		testResult("y", "C", 0, 0, 3, 0),
	}}

	var buf IteratorResult

	testCollector := &testCollector{
		reset: func(rowNumber RowNumber) {
			buf.Reset()
			buf.RowNumber = rowNumber
		},
		collect: func(r *IteratorResult, _ any) {
			buf.Append(r)
		},
		result: func() *IteratorResult {
			return &buf
		},
	}

	expected := []IteratorResult{
		{
			RowNumber: rn(0, 0, 1, 0),
			OtherEntries: []struct {
				Key   string
				Value interface{}
			}{
				{Key: "y", Value: "A1"},
				{Key: "x", Value: "A"},
			},
		},
		{
			RowNumber: rn(0, 0, 1, 1),
			OtherEntries: []struct {
				Key   string
				Value interface{}
			}{
				{Key: "y", Value: "A2"},
			},
		},
		{
			RowNumber: rn(0, 0, 2, 0),
			OtherEntries: []struct {
				Key   string
				Value interface{}