From 538b773e6d3fef40fc9fc6a6b1cac10537b18b23 Mon Sep 17 00:00:00 2001
From: Jakob Stoklund Olesen <jolesen@mozilla.com>
Date: Wed, 14 Sep 2016 09:10:32 -0700
Subject: [PATCH] Parse test commands in a .cton file.

The top-level parse_test() function now parses test commands in the
preamble of a .cton file.
---
 cranelift/src/libreader/parser.rs | 39 +++++++++++++++++++++++++++++--
 1 file changed, 37 insertions(+), 2 deletions(-)

diff --git a/cranelift/src/libreader/parser.rs b/cranelift/src/libreader/parser.rs
index 5e18af59be..7f8cc543ad 100644
--- a/cranelift/src/libreader/parser.rs
+++ b/cranelift/src/libreader/parser.rs
@@ -20,6 +20,7 @@ use cretonne::ir::entities::{AnyEntity, NO_EBB, NO_INST, NO_VALUE};
 use cretonne::ir::instructions::{InstructionFormat, InstructionData, VariableArgs, JumpData,
                                  BranchData, ReturnData};
 use testfile::{TestFile, DetailedFunction, Comment};
+use testcommand::TestCommand;
 
 pub use lexer::Location;
 
@@ -36,8 +37,11 @@ pub fn parse_functions(text: &str) -> Result<Vec<Function>> {
 ///
 /// The returned `TestFile` contains direct references to substrings of `text`.
 pub fn parse_test<'a>(text: &'a str) -> Result<TestFile<'a>> {
-    Parser::new(text);
-    unimplemented!()
+    let mut parser = Parser::new(text);
+    Ok(TestFile {
+        commands: parser.parse_test_commands(),
+        functions: try!(parser.parse_function_list()),
+    })
 }
 
 /// A parse error is returned when the parse failed.
@@ -289,6 +293,14 @@ impl<'a> Parser<'a> {
         self.lookahead.take().expect("No token to consume")
     }
 
+    // Consume the whole line following the current lookahead token.
+    // Return the text of the line tail.
+    fn consume_line(&mut self) -> &'a str {
+        let rest = self.lex.rest_of_line();
+        self.consume();
+        rest
+    }
+
     // Get the current lookahead token, after making sure there is one.
     fn token(&mut self) -> Option<Token<'a>> {
         while self.lookahead == None {
@@ -481,6 +493,15 @@ impl<'a> Parser<'a> {
         }
     }
 
+    /// Parse a list of test commands.
+    pub fn parse_test_commands(&mut self) -> Vec<TestCommand<'a>> {
+        let mut list = Vec::new();
+        while self.token() == Some(Token::Identifier("test")) {
+            list.push(TestCommand::new(self.consume_line()));
+        }
+        list
+    }
+
     /// Parse a list of function definitions.
     ///
     /// This is the top-level parse function matching the whole contents of a file.
@@ -1326,4 +1347,18 @@ mod tests {
         assert_eq!(dfunc.comments[6].entity, AnyEntity::Function);
         assert_eq!(dfunc.comments[7].entity, AnyEntity::Function);
     }
+
+    #[test]
+    fn test_file() {
+        let tf = parse_test("; before
+                             test cfg option=5
+                             test verify
+                             function comment() {}")
+            .unwrap();
+        assert_eq!(tf.commands.len(), 2);
+        assert_eq!(tf.commands[0].command, "cfg");
+        assert_eq!(tf.commands[1].command, "verify");
+        assert_eq!(tf.functions.len(), 1);
+        assert_eq!(tf.functions[0].function.name, "comment");
+    }
 }