mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-17 22:05:27 +00:00
aeb5dbd293
* feat: add RawStringSlice and refactor * restored old logic + refactor * restored files --------- Co-authored-by: mzack <marco.rivoli.nvh@gmail.com>
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.NewRaw("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
|
|
}
|