make ExtractQueryNode public

Author: andy-wm-arthur

enginetest/enginetests.go

@@ -351,7 +351,7 @@ func TestQueryPlan(t *testing.T, harness Harness, e *sqle.Engine, query string,
 			}
 		}
 
-		assert.Equal(t, expectedPlan, extractQueryNode(node).String(), "Unexpected result for query: "+query)
+		assert.Equal(t, expectedPlan, ExtractQueryNode(node).String(), "Unexpected result for query: "+query)
 	})
 
 }
@@ -371,7 +371,7 @@ func TestQueryPlanWithEngine(t *testing.T, harness Harness, e *sqle.Engine, tt q
 			}
 		}
 
-		assert.Equal(t, tt.ExpectedPlan, extractQueryNode(node).String(), "Unexpected result for query: "+tt.Query)
+		assert.Equal(t, tt.ExpectedPlan, ExtractQueryNode(node).String(), "Unexpected result for query: "+tt.Query)
 	})
 
 }

enginetest/evaluation.go

@@ -691,12 +691,12 @@ func assertSchemasEqualWithDefaults(t *testing.T, expected, actual sql.Schema) b
 	return assert.Equal(t, ec, ac)
 }
 
-func extractQueryNode(node sql.Node) sql.Node {
+func ExtractQueryNode(node sql.Node) sql.Node {
 	switch node := node.(type) {
 	case *plan.QueryProcess:
-		return extractQueryNode(node.Child())
+		return ExtractQueryNode(node.Child())
 	case *analyzer.Releaser:
-		return extractQueryNode(node.Child)
+		return ExtractQueryNode(node.Child)
 	default:
 		return node
 	}

enginetest/testgen_test.go

@@ -45,7 +45,7 @@ func TestWriteQueryPlans(t *testing.T) {
 
 		node, err := engine.Analyzer.Analyze(ctx, parsed, nil)
 		require.NoError(t, err)
-		planString := extractQueryNode(node).String()
+		planString := ExtractQueryNode(node).String()
 
 		if strings.Contains(tt.Query, "`") {
 			_, _ = w.WriteString(fmt.Sprintf(`Query: "%s",`, tt.Query))
@@ -102,7 +102,7 @@ func TestWriteIndexQueryPlans(t *testing.T) {
 
 		node, err := engine.Analyzer.Analyze(ctx, parsed, nil)
 		require.NoError(t, err)
-		planString := extractQueryNode(node).String()
+		planString := ExtractQueryNode(node).String()
 
 		if strings.Contains(tt.Query, "`") {
 			_, _ = w.WriteString(fmt.Sprintf(`Query: "%s",`, tt.Query))