107 lines
2.2 KiB
Go
107 lines
2.2 KiB
Go
package tusd_test
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
. "github.com/tus/tusd"
|
|
)
|
|
|
|
type closingStringReader struct {
|
|
*strings.Reader
|
|
closed bool
|
|
}
|
|
|
|
func (reader *closingStringReader) Close() error {
|
|
reader.closed = true
|
|
return nil
|
|
}
|
|
|
|
func TestGet(t *testing.T) {
|
|
SubTest(t, "Download", func(t *testing.T, store *MockFullDataStore) {
|
|
reader := &closingStringReader{
|
|
Reader: strings.NewReader("hello"),
|
|
}
|
|
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
locker := NewMockLocker(ctrl)
|
|
|
|
gomock.InOrder(
|
|
locker.EXPECT().LockUpload("yes"),
|
|
store.EXPECT().GetInfo("yes").Return(FileInfo{
|
|
Offset: 5,
|
|
Size: 20,
|
|
MetaData: map[string]string{
|
|
"filename": "file.jpg\"evil",
|
|
},
|
|
}, nil),
|
|
store.EXPECT().GetReader("yes").Return(reader, nil),
|
|
locker.EXPECT().UnlockUpload("yes"),
|
|
)
|
|
|
|
composer := NewStoreComposer()
|
|
composer.UseCore(store)
|
|
composer.UseGetReader(store)
|
|
composer.UseLocker(locker)
|
|
|
|
handler, _ := NewHandler(Config{
|
|
StoreComposer: composer,
|
|
})
|
|
|
|
(&httpTest{
|
|
Method: "GET",
|
|
URL: "yes",
|
|
ResHeader: map[string]string{
|
|
"Content-Length": "5",
|
|
"Content-Type": "application/octet-stream",
|
|
"Content-Disposition": `attachment;filename="file.jpg\"evil"`,
|
|
},
|
|
Code: http.StatusOK,
|
|
ResBody: "hello",
|
|
}).Run(handler, t)
|
|
|
|
if !reader.closed {
|
|
t.Error("expected reader to be closed")
|
|
}
|
|
})
|
|
|
|
SubTest(t, "EmptyDownload", func(t *testing.T, store *MockFullDataStore) {
|
|
store.EXPECT().GetInfo("yes").Return(FileInfo{
|
|
Offset: 0,
|
|
}, nil)
|
|
|
|
handler, _ := NewHandler(Config{
|
|
DataStore: store,
|
|
})
|
|
|
|
(&httpTest{
|
|
Method: "GET",
|
|
URL: "yes",
|
|
ResHeader: map[string]string{
|
|
"Content-Length": "0",
|
|
"Content-Disposition": `attachment`,
|
|
},
|
|
Code: http.StatusNoContent,
|
|
ResBody: "",
|
|
}).Run(handler, t)
|
|
})
|
|
|
|
SubTest(t, "NotProvided", func(t *testing.T, store *MockFullDataStore) {
|
|
composer := NewStoreComposer()
|
|
composer.UseCore(store)
|
|
|
|
handler, _ := NewUnroutedHandler(Config{
|
|
StoreComposer: composer,
|
|
})
|
|
|
|
(&httpTest{
|
|
Method: "GET",
|
|
URL: "foo",
|
|
Code: http.StatusNotImplemented,
|
|
}).Run(http.HandlerFunc(handler.GetFile), t)
|
|
})
|
|
}
|