Function TestLeftJoinUp

pkg/parquetquery/iters_test.go:441–494

(t *testing.T)

}

func TestLeftJoinUp(t *testing.T) {
	// Defined at level 0
	iter1 := &testIterator{rows: []IteratorResult{
		testResult("x", "A", 1),
		testResult("x", "B", 2),
	}}

	// Defined at level 1
	iter2 := &testIterator{rows: []IteratorResult{
		testResult("y", "A", 1, 0),
		testResult("y", "B", 1, 1),
		testResult("y", "C", 2, 0),
	}}

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

	got := []IteratorResult{}

	j, err := NewLeftJoinIterator(0, []Iterator{iter1, iter2}, nil, nil)
	require.NoError(t, err)

	for {
		res, err := j.Next()
		require.NoError(t, err)
		if res == nil {
			break
		}
		got = append(got, clone(res))
	}

	require.Equal(t, expected, got)
}

func TestLeftJoinDown(t *testing.T) {
	iter1DefLevel := 2