Fix upload file type check (#7890)
* fix upload file type check * make the function simple and added tests * Update comment as per @silverwindtokarchuk/v1.17
parent
a678ea44b8
commit
2d0b90c967
@ -0,0 +1,47 @@ |
||||
// Copyright 2019 The Gitea Authors. All rights reserved.
|
||||
// Use of this source code is governed by a MIT-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package upload |
||||
|
||||
import ( |
||||
"bytes" |
||||
"compress/gzip" |
||||
"testing" |
||||
|
||||
"github.com/stretchr/testify/assert" |
||||
) |
||||
|
||||
func TestUpload(t *testing.T) { |
||||
testContent := []byte(`This is a plain text file.`) |
||||
var b bytes.Buffer |
||||
w := gzip.NewWriter(&b) |
||||
w.Write(testContent) |
||||
w.Close() |
||||
|
||||
kases := []struct { |
||||
data []byte |
||||
allowedTypes []string |
||||
err error |
||||
}{ |
||||
{ |
||||
data: testContent, |
||||
allowedTypes: []string{"text/plain"}, |
||||
err: nil, |
||||
}, |
||||
{ |
||||
data: testContent, |
||||
allowedTypes: []string{"application/x-gzip"}, |
||||
err: ErrFileTypeForbidden{"text/plain; charset=utf-8"}, |
||||
}, |
||||
{ |
||||
data: b.Bytes(), |
||||
allowedTypes: []string{"application/x-gzip"}, |
||||
err: nil, |
||||
}, |
||||
} |
||||
|
||||
for _, kase := range kases { |
||||
assert.Equal(t, kase.err, VerifyAllowedContentType(kase.data, kase.allowedTypes)) |
||||
} |
||||
} |
Loading…
Reference in new issue