Add regression test for when one index is strictly better than another for a lookup.

Author: nicktobey

enginetest/queries/index_queries.go

@@ -4065,6 +4065,78 @@ var IndexPrefixQueries = []ScriptTest{
 			},
 		},
 	},
+	{
+		Name: "multiple nullable index prefixes",
+		SetUpScript: []string{
+			"create table test(pk int primary key, shared1 int, shared2 int, a3 int, a4 int, b3 int, b4 int, unique key a_idx(shared1, shared2, a3, a4), unique key b_idx(shared1, shared2, b3, b4))",
+		},
+		Assertions: []ScriptTestAssertion{
+			{
+				Query:           "select * from test where shared1 = 1 and shared2 = 2 and a3 = 3;",
+				Expected:        []sql.Row{},
+				ExpectedIndexes: []string{"a_idx"},
+			},
+			{
+				Query:           "select * from test where shared1 = 1 and shared2 = 2 and b3 = 3;",
+				Expected:        []sql.Row{},
+				ExpectedIndexes: []string{"b_idx"},
+			},
+		},
+	},
+	{
+		Name: "multiple non-unique index prefixes",
+		SetUpScript: []string{
+			"create table test(pk int primary key, shared1 int not null, shared2 int not null, a3 int not null, a4 int not null, b3 int not null, b4 int not null, key a_idx(shared1, shared2, a3, a4), key b_idx(shared1, shared2, b3, b4))",
+		},
+		Assertions: []ScriptTestAssertion{
+			{
+				Query:           "select * from test where shared1 = 1 and shared2 = 2 and a3 = 3;",
+				Expected:        []sql.Row{},
+				ExpectedIndexes: []string{"a_idx"},
+			},
+			{
+				Query:           "select * from test where shared1 = 1 and shared2 = 2 and b3 = 3;",
+				Expected:        []sql.Row{},
+				ExpectedIndexes: []string{"b_idx"},
+			},
+		},
+	},
+	{
+		Name: "multiple non-unique nullable index prefixes",
+		SetUpScript: []string{
+			"create table test(pk int primary key, shared1 int, shared2 int, a3 int, a4 int, b3 int, b4 int, key a_idx(shared1, shared2, a3, a4), key b_idx(shared1, shared2, b3, b4))",
+		},
+		Assertions: []ScriptTestAssertion{
+			{
+				Query:           "select * from test where shared1 = 1 and shared2 = 2 and a3 = 3;",
+				Expected:        []sql.Row{},
+				ExpectedIndexes: []string{"a_idx"},
+			},
+			{
+				Query:           "select * from test where shared1 = 1 and shared2 = 2 and b3 = 3;",
+				Expected:        []sql.Row{},
+				ExpectedIndexes: []string{"b_idx"},
+			},
+		},
+	},
+	{
+		Name: "unique and non-unique nullable index prefixes",
+		SetUpScript: []string{
+			"create table test(pk int primary key, shared1 int, shared2 int, a3 int, a4 int, b3 int, b4 int, unique key a_idx(shared1, shared2, a3, a4), key b_idx(shared1, shared2, b3, b4))",
+		},
+		Assertions: []ScriptTestAssertion{
+			{
+				Query:           "select * from test where shared1 = 1 and shared2 = 2 and a3 = 3;",
+				Expected:        []sql.Row{},
+				ExpectedIndexes: []string{"a_idx"},
+			},
+			{
+				Query:           "select * from test where shared1 = 1 and shared2 = 2 and b3 = 3;",
+				Expected:        []sql.Row{},
+				ExpectedIndexes: []string{"b_idx"},
+			},
+		},
+	},
 }
 
 var IndexQueries = []ScriptTest{