Function TestGetMaxRelationalNodeID

query_plan_test.go:584–608

(t *testing.T)

}

func TestGetMaxRelationalNodeID(t *testing.T) {
	for _, tt := range []struct {
		desc  string
		input *pb.QueryPlan
		want  int32
	}{
		{
			desc: "pre-sorted order",
			input: &pb.QueryPlan{
				PlanNodes: []*pb.PlanNode{
					{Index: 0, DisplayName: "Scalar Subquery", Kind: pb.PlanNode_SCALAR},
					{Index: 1, DisplayName: "Index Scan", Kind: pb.PlanNode_RELATIONAL},
					{Index: 2, DisplayName: "Index Scan", Kind: pb.PlanNode_RELATIONAL},
					{Index: 3, DisplayName: "Index Scan", Kind: pb.PlanNode_RELATIONAL},
					{Index: 4, DisplayName: "Constant", Kind: pb.PlanNode_SCALAR}, // This is not visible
				},
			},
			want: 3,
		},
	} {
		if got := getMaxRelationalNodeID(tt.input); got != tt.want {
			t.Errorf("getMaxRelationalNodeID(%s) = %d, but want = %d", tt.input, got, tt.want)
		}
	}
}