mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-17 19:55:26 +00:00
45 lines
2.4 KiB
Go
45 lines
2.4 KiB
Go
package expressions
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestEvaluate(t *testing.T) {
|
|
items := []struct {
|
|
input string
|
|
expected string
|
|
extra map[string]interface{}
|
|
}{
|
|
{input: "{{url_encode('test}aaa')}}", expected: "test%7Daaa", extra: map[string]interface{}{}},
|
|
{input: "{{hex_encode('PING')}}", expected: "50494e47", extra: map[string]interface{}{}},
|
|
// TODO #1501
|
|
//{input: "{{hex_encode('{{')}}", expected: "7b7b", extra: map[string]interface{}{}},
|
|
//{input: `{{concat("{{", 123, "*", 123, "}}")}}`, expected: "{{123*123}}", extra: map[string]interface{}{}},
|
|
//{input: `{{concat("{{", "123*123", "}}")}}`, expected: "{{123*123}}", extra: map[string]interface{}{}},
|
|
//{input: `{{"{{" + '123*123' + "}}"}}`, expected: "{{123*123}}", extra: map[string]interface{}{}},
|
|
{input: `{{concat(123,'*',123)}}`, expected: "123*123", extra: map[string]interface{}{}},
|
|
{input: `{{1+1}}`, expected: "2", extra: map[string]interface{}{}},
|
|
{input: `{{"1"+"1"}}`, expected: "11", extra: map[string]interface{}{}},
|
|
{input: `{{"1" + '*' + "1"}}`, expected: "1*1", extra: map[string]interface{}{}},
|
|
{input: `{{"a" + 'b' + "c"}}`, expected: "abc", extra: map[string]interface{}{}},
|
|
{input: `{{10*2}}`, expected: "20", extra: map[string]interface{}{}},
|
|
{input: `{{10/2}}`, expected: "5", extra: map[string]interface{}{}},
|
|
{input: `{{10-2}}`, expected: "8", extra: map[string]interface{}{}},
|
|
{input: "test", expected: "test", extra: map[string]interface{}{}},
|
|
{input: "{{hex_encode(Item)}}", expected: "50494e47", extra: map[string]interface{}{"Item": "PING"}},
|
|
{input: "{{hex_encode(Item)}}\r\n", expected: "50494e47\r\n", extra: map[string]interface{}{"Item": "PING"}},
|
|
{input: "{{someTestData}}{{hex_encode('PING')}}", expected: "{{someTestData}}50494e47", extra: map[string]interface{}{}},
|
|
{input: `_IWP_JSON_PREFIX_{{base64("{\"iwp_action\":\"add_site\",\"params\":{\"username\":\"\"}}")}}`, expected: "_IWP_JSON_PREFIX_eyJpd3BfYWN0aW9uIjoiYWRkX3NpdGUiLCJwYXJhbXMiOnsidXNlcm5hbWUiOiIifX0=", extra: map[string]interface{}{}},
|
|
{input: "{{}}", expected: "{{}}", extra: map[string]interface{}{}},
|
|
{input: `"{{hex_encode('PING')}}"`, expected: `"50494e47"`, extra: map[string]interface{}{}},
|
|
}
|
|
for _, item := range items {
|
|
value, err := Evaluate(item.input, item.extra)
|
|
require.Nil(t, err, "could not evaluate helper")
|
|
|
|
require.Equal(t, item.expected, value, "could not get correct expression")
|
|
}
|
|
}
|