Include included tests in test count

pingcap · Jul 26, 2024 · 3c938b5 · 3c938b5
1 parent 4a2376e
commit 3c938b5
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/include/hello.inc b/include/hello.inc
@@ -1 +1,2 @@
 --echo Hello from the included file
+SELECT 1
diff --git a/r/source.result b/r/source.result
@@ -1,3 +1,6 @@
 first line
 Hello from the included file
+SELECT 1
+1
+1
 last line
diff --git a/src/main.go b/src/main.go
@@ -425,7 +425,7 @@ func (t *tester) runQueries(queries []query) (int, error) {
 			if t.enableConcurrent {
 				concurrentQueue = append(concurrentQueue, q)
 			} else if err = t.execute(q); err != nil {
-				err = errors.Annotate(err, fmt.Sprintf("sql:%v", q.Query))
+				err = errors.Annotate(err, fmt.Sprintf("sql:%v line:%s", q.Query, q.location()))
 				t.addFailure(&testSuite, &err, testCnt)
 				return testCnt, err
 			}
@@ -549,10 +549,11 @@ func (t *tester) runQueries(queries []query) (int, error) {
 			if err != nil {
 				return testCnt, errors.Annotate(err, fmt.Sprintf("error loading queries from %s", fileName))
 			}
-			_, err = t.runQueries(includedQueries)
+			includeCnt, err := t.runQueries(includedQueries)
 			if err != nil {
 				return testCnt, err
 			}
+			testCnt += includeCnt
 		default:
 			log.WithFields(log.Fields{"command": q.firstWord, "arguments": q.Query, "line": q.location()}).Warn("command not implemented")
 		}