Support positive unary operator (do nothing)

goby-lang · Oct 27, 2019 · fd56bb3 · fd56bb3
1 parent 9074403
commit fd56bb3
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 0 deletions.
diff --git a/compiler/bytecode/expression_generation.go b/compiler/bytecode/expression_generation.go
@@ -244,6 +244,8 @@ func (g *Generator) compilePrefixExpression(is *InstructionSet, exp *ast.PrefixE
 		is.define(PutObject, exp.Line(), 0)
 		g.compileExpression(is, exp.Right, scope, table)
 		is.define(Send, exp.Line(), exp.Operator, 1, "", &ArgSet{})
+	case "+":
+		g.compileExpression(is, exp.Right, scope, table)
 	}
 }
 

diff --git a/compiler/parser/parser.go b/compiler/parser/parser.go
@@ -70,6 +70,7 @@ func New(l *lexer.Lexer) *Parser {
 	p.registerPrefix(token.False, p.parseBooleanLiteral)
 	p.registerPrefix(token.Null, p.parseNilExpression)
 	p.registerPrefix(token.Minus, p.parsePrefixExpression)
+	p.registerPrefix(token.Plus, p.parsePrefixExpression)
 	p.registerPrefix(token.Asterisk, p.parsePrefixExpression)
 	p.registerPrefix(token.Bang, p.parsePrefixExpression)
 	p.registerPrefix(token.LParen, p.parseGroupedExpression)

diff --git a/vm/evaluation_test.go b/vm/evaluation_test.go
@@ -2242,6 +2242,26 @@ func TestMinusPrefixMethodCall(t *testing.T) {
 	}
 }
 
+func TestPlusPrefixMethodCall(t *testing.T) {
+	tests := []struct {
+		input    string
+		expected int
+	}{
+		{"+5", 5},
+		{"+10", 10},
+		{"+(-10)", -10},
+		{"+(-5)", -5},
+	}
+
+	for i, tt := range tests {
+		v := initTestVM()
+		evaluated := v.testEval(t, tt.input, getFilename())
+		VerifyExpected(t, i, evaluated, tt.expected)
+		v.checkCFP(t, i, 0)
+		v.checkSP(t, i, 1)
+	}
+}
+
 func TestMultiVarAssignment(t *testing.T) {
 	tests := []struct {
 		input    string