Add more tests for expressions

Author: maxmousee

expression_test.go

@@ -91,6 +91,78 @@ func TestFindExpressionsInTokens(t *testing.T) {
 	assertions.False(FindExpressionsInTokens([]Token{token}, []Expression{}))
 }
 
+func TestFindComplexExpressionsInTokens(t *testing.T) {
+	assertions := assert.New(t)
+	expression := Expression{
+		Type: "bracket",
+		Text: "(",
+	}
+
+	expression2 := Expression{
+		Type: "bracket",
+		Text: ")",
+	}
+
+	token := Token{
+		Type:     "bracket",
+		Position: 1,
+		Text:     "{",
+	}
+	token2 := Token{
+		Type:     "bracket",
+		Position: 2,
+		Text:     "(",
+	}
+	token3 := Token{
+		Type:     "bracket",
+		Position: 3,
+		Text:     ")",
+	}
+	token4 := Token{
+		Type:     "bracket",
+		Position: 4,
+		Text:     "{",
+	}
+
+	assertions.True(FindExpressionsInTokens([]Token{token, token2, token3, token4}, []Expression{expression, expression2}))
+}
+
+func TestDoNotFindComplexExpressionsInTokens(t *testing.T) {
+	assertions := assert.New(t)
+	expression := Expression{
+		Type: "bracket",
+		Text: "{",
+	}
+
+	expression2 := Expression{
+		Type: "keyword",
+		Text: "var",
+	}
+
+	token := Token{
+		Type:     "bracket",
+		Position: 1,
+		Text:     "{",
+	}
+	token2 := Token{
+		Type:     "bracket",
+		Position: 2,
+		Text:     "(",
+	}
+	token3 := Token{
+		Type:     "bracket",
+		Position: 3,
+		Text:     ")",
+	}
+	token4 := Token{
+		Type:     "bracket",
+		Position: 4,
+		Text:     "{",
+	}
+
+	assertions.False(FindExpressionsInTokens([]Token{token, token2, token3, token4}, []Expression{expression, expression2}))
+}
+
 func TestMatchToken(t *testing.T) {
 	assertions := assert.New(t)