Skip to content

Commit fa3044a

Browse files
committed
Fix plangen: correctly generate test files when explicitly skipping the row estimate.
1 parent 6d7c4c0 commit fa3044a

File tree

1 file changed

+7
-6
lines changed
  • enginetest/plangen/cmd/plangen

1 file changed

+7
-6
lines changed

enginetest/plangen/cmd/plangen/main.go

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -208,30 +208,31 @@ func generatePlansForSuite(spec PlanSpec, w *bytes.Buffer) error {
208208
writePlanString(w, planString)
209209

210210
if node.IsReadOnly() {
211-
_, _ = w.WriteString(`ExpectedEstimates: `)
212211
var planString string
213212
if tt.ExpectedEstimates != "skip" {
213+
_, _ = w.WriteString(`ExpectedEstimates: `)
214214
planString = sql.Describe(enginetest.ExtractQueryNode(node), sql.DescribeOptions{
215215
Estimates: true,
216216
})
217+
writePlanString(w, planString)
217218
} else {
218-
planString = "skip"
219+
_, _ = w.WriteString("ExpectedEstimates: \"skip\",\n")
219220
}
220-
writePlanString(w, planString)
221+
221222
if tt.ExpectedAnalysis != "skip" {
223+
_, _ = w.WriteString(`ExpectedAnalysis: `)
222224
err = enginetest.ExecuteNode(ctx, engine, node)
223225
if err != nil {
224226
exit(fmt.Errorf("%w\nfailed to execute query: %s", err, tt.Query))
225227
}
226-
_, _ = w.WriteString(`ExpectedAnalysis: `)
227228
planString = sql.Describe(enginetest.ExtractQueryNode(node), sql.DescribeOptions{
228229
Analyze: true,
229230
Estimates: true,
230231
})
232+
writePlanString(w, planString)
231233
} else {
232-
planString = "skip"
234+
_, _ = w.WriteString("ExpectedAnalysis: \"skip\",\n")
233235
}
234-
writePlanString(w, planString)
235236
}
236237
} else {
237238
_, _ = w.WriteString(`Skip: true,\n`)

0 commit comments

Comments
 (0)