fixes coinbase/mesh-cli#248

coinbase · Jan 19, 2022 · 2df21f0 · 2df21f0
1 parent 2868f10
commit 2df21f0
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 0 deletions.
diff --git a/constructor/dsl/dsl_test.go b/constructor/dsl/dsl_test.go
@@ -330,6 +330,37 @@ func TestParse(t *testing.T) {
 			file:        "blah_blah.txt",
 			expectedErr: ErrIncorrectExtension,
 		},
+		"action: valid type": {
+			file: "action_valid.ros",
+			expectedWorkflows: []*job.Workflow{
+				{
+					Name:        string(job.CreateAccount),
+					Concurrency: job.ReservedWorkflowConcurrency,
+					Scenarios: []*job.Scenario{
+						{
+							Name: "create",
+							Actions: []*job.Action{
+								{
+									Type:       job.SetVariable,
+									Input:      `{"network":"chrysalis-devnet", "blockchain":"iota"}`,
+									OutputPath: "network",
+								},
+								{
+									Type:       job.GenerateKey,
+									Input:      `{"curve_type": "edwards25519"}`,
+									OutputPath: "key",
+								},
+								{
+									Type:       job.Derive,
+									Input:      `{"network_identifier": {{network}},"public_key": {{key.public_key}}}`,
+									OutputPath: "account",
+								},
+							},
+						},
+					},
+				},
+			},
+		},
 	}
 
 	for name, test := range tests {

diff --git a/constructor/dsl/parser.go b/constructor/dsl/parser.go
@@ -17,6 +17,7 @@ package dsl
 import (
 	"bufio"
 	"context"
+	"encoding/json"
 	"errors"
 	"fmt"
 	"os"
@@ -147,6 +148,10 @@ func parseActionType(line string) (job.ActionType, string, string, error) {
 		divide:   job.Division,
 	} {
 		tokens = strings.SplitN(remaining, symbol, split2)
+
+		if json.Valid([]byte(strings.TrimSuffix(remaining, ";"))) {
+			break
+		}
 		if len(tokens) == split2 {
 			if len(mathOperation) == 0 {
 				return "", "", "", ErrInvalidMathSymbol

diff --git a/constructor/dsl/testdata/action_valid.ros b/constructor/dsl/testdata/action_valid.ros
@@ -0,0 +1,10 @@
+create_account(1){
+  create{
+    network = {"network":"chrysalis-devnet", "blockchain":"iota"};
+    key = generate_key({"curve_type": "edwards25519"});
+    account = derive({
+      "network_identifier": {{network}},
+      "public_key": {{key.public_key}}
+    });
+  }
+}