123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194 |
- // Copyright 2019 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- 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
- fileName string
- allowedTypes string
- err error
- }{
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: "",
- err: nil,
- },
- {
- data: testContent,
- fileName: "dir/test.txt",
- allowedTypes: "",
- err: nil,
- },
- {
- data: testContent,
- fileName: "../../../test.txt",
- allowedTypes: "",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: "",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: ",",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: "|",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: "*/*",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: "*/*,",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: "*/*|",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: "text/plain",
- err: nil,
- },
- {
- data: testContent,
- fileName: "dir/test.txt",
- allowedTypes: "text/plain",
- err: nil,
- },
- {
- data: testContent,
- fileName: "/dir.txt/test.js",
- allowedTypes: ".js",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: " text/plain ",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: ".txt",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: " .txt,.js",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: " .txt|.js",
- err: nil,
- },
- {
- data: testContent,
- fileName: "../../test.txt",
- allowedTypes: " .txt|.js",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: " .txt ,.js ",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: "text/plain, .txt",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: "text/*",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: "text/*,.js",
- err: nil,
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: "text/**",
- err: ErrFileTypeForbidden{"text/plain; charset=utf-8"},
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: "application/x-gzip",
- err: ErrFileTypeForbidden{"text/plain; charset=utf-8"},
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: ".zip",
- err: ErrFileTypeForbidden{"text/plain; charset=utf-8"},
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: ".zip,.txtx",
- err: ErrFileTypeForbidden{"text/plain; charset=utf-8"},
- },
- {
- data: testContent,
- fileName: "test.txt",
- allowedTypes: ".zip|.txtx",
- err: ErrFileTypeForbidden{"text/plain; charset=utf-8"},
- },
- {
- data: b.Bytes(),
- fileName: "test.txt",
- allowedTypes: "application/x-gzip",
- err: nil,
- },
- }
-
- for _, kase := range kases {
- assert.Equal(t, kase.err, Verify(kase.data, kase.fileName, kase.allowedTypes))
- }
- }
|