Modcc: parse the unit of `FUNCTION` (#1361)

* add function unit parsing
* add unit tests

modcc/parser.cpp

@@ -973,6 +973,14 @@ symbol_ptr Parser::parse_function() {
     auto p = parse_prototype();
     if (p == nullptr) return nullptr;
 
+    // Functions may have a unit attached
+    if (token_.type == tok::lparen) {
+        unit_description();
+        if (status_ == lexerStatus::error) {
+            return {};
+        }
+    }
+
     // check for opening left brace {
     if (!expect(tok::lbrace)) return nullptr;
 

test/unit-modcc/test_parser.cpp

@@ -205,6 +205,7 @@ TEST(Parser, net_receive) {
 }
 
 TEST(Parser, function) {
+    {
         char str[] =
             "FUNCTION foo(x, y) {"
             "  LOCAL a\n"
@@ -217,6 +218,34 @@ TEST(Parser, function) {
         std::unique_ptr<Symbol> sym;
         EXPECT_TRUE(check_parse(sym, &Parser::parse_function, str));
     }
+    {
+        char str[] =
+            "FUNCTION foo(x (mv), y (/mA)) {"
+            "  foo = x * y\n"
+            "}";
+
+        std::unique_ptr<Symbol> sym;
+        EXPECT_TRUE(check_parse(sym, &Parser::parse_function, str));
+    }
+    {
+        char str[] =
+            "FUNCTION foo(x (mv), y (/mA)) (mv/mA) {"
+            "  foo = x * y\n"
+            "}";
+
+        std::unique_ptr<Symbol> sym;
+        EXPECT_TRUE(check_parse(sym, &Parser::parse_function, str));
+    }
+    {
+        char str[] =
+            "FUNCTION foo(x (mv), y (/mA)) (mv-mA) {"
+            "  foo = x * y\n"
+            "}";
+
+        std::unique_ptr<Symbol> sym;
+        EXPECT_FALSE(check_parse(sym, &Parser::parse_function, str));
+    }
+}
 
 TEST(Parser, parse_solve) {
     std::unique_ptr<SolveExpression> s;