mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2026-11-10 12:11:33 -10:00
Om
b104625865
The octet-stream branch compared the schema format against "bindary", so a requestBody with format: binary went out with Content-Type: text/plain and only format: byte kept application/octet-stream.
47 lines
1.3 KiB
Go
47 lines
1.3 KiB
Go
package openapi
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/projectdiscovery/nuclei/v3/pkg/input/types"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestOpenAPIOctetStreamContentType(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
schema string
|
|
contentType string
|
|
}{
|
|
{name: "binary format", schema: "{type: string, format: binary}", contentType: "application/octet-stream"},
|
|
{name: "byte format", schema: "{type: string, format: byte}", contentType: "application/octet-stream"},
|
|
{name: "plain string", schema: "{type: string}", contentType: "text/plain"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
spec := `openapi: 3.0.0
|
|
info: {title: t, version: "1"}
|
|
servers: [{url: "http://example.com"}]
|
|
paths:
|
|
/upload:
|
|
post:
|
|
requestBody:
|
|
content:
|
|
application/octet-stream: {schema: ` + tt.schema + `}
|
|
responses:
|
|
"200": {description: ok}
|
|
`
|
|
var contentTypes []string
|
|
err := New().Parse(strings.NewReader(spec), func(rr *types.RequestResponse) bool {
|
|
value, _ := rr.Request.Headers.Get("Content-Type")
|
|
contentTypes = append(contentTypes, value)
|
|
return false
|
|
}, "spec.yaml")
|
|
require.NoError(t, err)
|
|
require.Equal(t, []string{tt.contentType}, contentTypes)
|
|
})
|
|
}
|
|
}
|