mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-19 17:55:26 +00:00
03612a28b8
In some case you may need to use custom type in `metadata`, like nested array or nested k-v:
```yaml
info:
metadata:
components:
- bootstrap
- jquery
```
So this commit allowed use any custom type in `metadata`.
47 lines
1.7 KiB
Go
47 lines
1.7 KiB
Go
package format
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/model"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/model/types/severity"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/model/types/stringslice"
|
|
)
|
|
|
|
func TestToMarkdownTableString(t *testing.T) {
|
|
info := model.Info{
|
|
Name: "Test Template Name",
|
|
Authors: stringslice.StringSlice{Value: []string{"forgedhallpass", "ice3man"}},
|
|
Description: "Test description",
|
|
SeverityHolder: severity.Holder{Severity: severity.High},
|
|
Tags: stringslice.StringSlice{Value: []string{"cve", "misc"}},
|
|
Reference: stringslice.StringSlice{Value: "reference1"},
|
|
Metadata: map[string]interface{}{
|
|
"customDynamicKey1": "customDynamicValue1",
|
|
"customDynamicKey2": "customDynamicValue2",
|
|
},
|
|
}
|
|
|
|
result := ToMarkdownTableString(&info)
|
|
|
|
expectedOrderedAttributes := `| Name | Test Template Name |
|
|
| Authors | forgedhallpass, ice3man |
|
|
| Tags | cve, misc |
|
|
| Severity | high |
|
|
| Description | Test description |`
|
|
|
|
expectedDynamicAttributes := []string{
|
|
"| customDynamicKey1 | customDynamicValue1 |",
|
|
"| customDynamicKey2 | customDynamicValue2 |",
|
|
"", // the expected result ends in a new line (\n)
|
|
}
|
|
|
|
actualAttributeSlice := strings.Split(result, "\n")
|
|
dynamicAttributeIndex := len(actualAttributeSlice) - len(expectedDynamicAttributes)
|
|
assert.Equal(t, strings.Split(expectedOrderedAttributes, "\n"), actualAttributeSlice[:dynamicAttributeIndex]) // the first part of the result is ordered
|
|
assert.ElementsMatch(t, expectedDynamicAttributes, actualAttributeSlice[dynamicAttributeIndex:]) // dynamic parameters are not ordered
|
|
}
|