feat: display comment on result file

sqlness-cli/tests/local/input.result

@@ -1,3 +1,4 @@
+-- https://dev.mysql.com/doc/refman/8.0/en/example-auto-increment.html
 DROP TABLE IF EXISTS animals;
 
 affected_rows: 0

sqlness/examples/interceptor-replace/simple/replace.result

@@ -13,6 +13,7 @@ SELECT 0;
 
 SELECT 1;
 
+-- example of capture group replacement
 -- SQLNESS REPLACE (?P<y>\d{4})-(?P<m>\d{2})-(?P<d>\d{2}) $m/$d/$y
 2012-03-14, 2013-01-01 and 2014-07-05;
 

sqlness/src/case.rs

@@ -35,14 +35,20 @@ impl TestCase {
         let reader = BufReader::new(file);
         for line in reader.lines() {
             let line = line?;
-            // intercept command start with INTERCEPTOR_PREFIX
-            if line.starts_with(&cfg.interceptor_prefix) {
-                query.push_interceptor(&cfg.interceptor_prefix, line);
+
+            // record comment
+            if line.starts_with(COMMENT_PREFIX) {
+                query.push_comment(line.clone());
+
+                // intercept command start with INTERCEPTOR_PREFIX
+                if line.starts_with(&cfg.interceptor_prefix) {
+                    query.push_interceptor(&cfg.interceptor_prefix, line);
+                }
                 continue;
             }
 
-            // ignore comment and empty line
-            if line.starts_with(COMMENT_PREFIX) || line.is_empty() {
+            // ignore empty line
+            if line.is_empty() {
                 continue;
             }
 
@@ -89,8 +95,8 @@ pub struct QueryContext {
 
 #[derive(Default)]
 struct Query {
+    comment_lines: Vec<String>,
     query_lines: Vec<String>,
-    interceptor_lines: Vec<String>,
     interceptor_factories: Vec<InterceptorFactoryRef>,
     interceptors: Vec<InterceptorRef>,
 }
@@ -112,7 +118,10 @@ impl Query {
                 self.interceptors.push(interceptor);
             }
         }
-        self.interceptor_lines.push(interceptor_line);
+    }
+
+    fn push_comment(&mut self, comment_line: String) {
+        self.comment_lines.push(comment_line);
     }
 
     fn append_query_line(&mut self, line: &str) {
@@ -165,8 +174,8 @@ impl Query {
     where
         W: Write,
     {
-        for interceptor in &self.interceptor_lines {
-            writer.write_all(interceptor.as_bytes())?;
+        for comment in &self.comment_lines {
+            writer.write_all(comment.as_bytes())?;
             writer.write("\n".as_bytes())?;
         }
         for line in &self.query_lines {