Add enable_source/disable_source

Author: dveeden

include/hello.inc

@@ -1,6 +1,10 @@
-first line
+before source
 Hello from the included file
 SELECT 1
 1
 1
-last line
+after source
+Goodbye from the included file
+SELECT 3
+3
+3

r/source.result

@@ -47,6 +47,7 @@ var (
 	retryConnCount   int
 	collationDisable bool
 	checkErr         bool
+	disableSource    bool
 )
 
 func init() {
@@ -516,7 +517,15 @@ func (t *tester) runQueries(queries []query) (int, error) {
 						q.location(), q.Query))
 			}
 			t.replaceRegex = regex
+		case Q_ENABLE_SOURCE:
+			disableSource = false
+		case Q_DISABLE_SOURCE:
+			disableSource = true
 		case Q_SOURCE:
+			if disableSource {
+				log.WithFields(log.Fields{"line": q.location()}).Warn("source command disabled")
+				break
+			}
 			fileName := strings.TrimSpace(q.Query)
 			cwd, err := os.Getwd()
 			if err != nil {

src/main.go

@@ -124,6 +124,8 @@ const (
 	Q_COMMENT /* Comments, ignored. */
 	Q_COMMENT_WITH_COMMAND
 	Q_EMPTY_LINE
+	Q_DISABLE_SOURCE
+	Q_ENABLE_SOURCE
 )
 
 // ParseQueries parses an array of string into an array of query object.

src/query.go

@@ -114,6 +114,8 @@ var commandMap = map[string]int{
 	"single_query":               Q_SINGLE_QUERY,
 	"begin_concurrent":           Q_BEGIN_CONCURRENT,
 	"end_concurrent":             Q_END_CONCURRENT,
+	"disable_source":             Q_DISABLE_SOURCE,
+	"enable_source":              Q_ENABLE_SOURCE,
 }
 
 func findType(cmdName string) int {

src/type.go

@@ -1,3 +1,9 @@
---echo first line
---source include/hello.inc
---echo last line
+--echo before source
+--source include/hello1.inc
+--echo after source
+
+--disable_source
+--source include/hello2.inc
+--enable_source
+
+--source include/hello3.inc