tusd/pkg/handler/head_test.go

150 lines
3.6 KiB
Go
Raw Normal View History

package handler_test
2015-02-09 18:37:06 +00:00
import (
"context"
2015-02-09 18:37:06 +00:00
"net/http"
"os"
"testing"
2015-12-25 21:33:27 +00:00
"github.com/golang/mock/gomock"
. "github.com/tus/tusd/pkg/handler"
2015-02-09 18:37:06 +00:00
)
func TestHead(t *testing.T) {
SubTest(t, "Status", func(t *testing.T, store *MockFullDataStore, composer *StoreComposer) {
ctrl := gomock.NewController(t)
defer ctrl.Finish()
locker := NewMockFullLocker(ctrl)
lock := NewMockFullLock(ctrl)
2019-08-25 20:10:55 +00:00
upload := NewMockFullUpload(ctrl)
gomock.InOrder(
locker.EXPECT().NewLock("yes").Return(lock, nil),
lock.EXPECT().Lock().Return(nil),
store.EXPECT().GetUpload(context.Background(), "yes").Return(upload, nil),
upload.EXPECT().GetInfo(context.Background()).Return(FileInfo{
Offset: 11,
Size: 44,
MetaData: map[string]string{
"name": "lunrjs.png",
"empty": "",
},
}, nil),
lock.EXPECT().Unlock().Return(nil),
)
composer = NewStoreComposer()
composer.UseCore(store)
composer.UseLocker(locker)
2015-02-09 18:37:06 +00:00
handler, _ := NewHandler(Config{
StoreComposer: composer,
})
2015-02-09 18:37:06 +00:00
res := (&httpTest{
Method: "HEAD",
URL: "yes",
ReqHeader: map[string]string{
"Tus-Resumable": "1.0.0",
},
Code: http.StatusOK,
ResHeader: map[string]string{
"Upload-Offset": "11",
"Upload-Length": "44",
"Cache-Control": "no-store",
},
}).Run(handler, t)
2015-02-09 18:37:06 +00:00
// Since the order of a map is not guaranteed in Go, we need to be prepared
// for the case, that the order of the metadata may have been changed
if v := res.Header().Get("Upload-Metadata"); v != "name bHVucmpzLnBuZw==,empty " &&
v != "empty ,name bHVucmpzLnBuZw==" {
t.Errorf("Expected valid metadata (got '%s')", v)
}
2015-02-09 18:37:06 +00:00
})
SubTest(t, "UploadNotFoundFail", func(t *testing.T, store *MockFullDataStore, composer *StoreComposer) {
store.EXPECT().GetUpload(context.Background(), "no").Return(nil, os.ErrNotExist)
2015-02-17 14:44:12 +00:00
handler, _ := NewHandler(Config{
StoreComposer: composer,
})
2015-11-26 15:25:34 +00:00
res := (&httpTest{
Method: "HEAD",
URL: "no",
ReqHeader: map[string]string{
"Tus-Resumable": "1.0.0",
},
Code: http.StatusNotFound,
ResHeader: map[string]string{
"Content-Length": "0",
},
}).Run(handler, t)
if res.Body.String() != "" {
t.Errorf("Expected empty body for failed HEAD request")
}
})
2018-04-28 19:39:46 +00:00
SubTest(t, "DeferLengthHeader", func(t *testing.T, store *MockFullDataStore, composer *StoreComposer) {
2019-08-25 20:10:55 +00:00
ctrl := gomock.NewController(t)
defer ctrl.Finish()
upload := NewMockFullUpload(ctrl)
gomock.InOrder(
store.EXPECT().GetUpload(context.Background(), "yes").Return(upload, nil),
upload.EXPECT().GetInfo(context.Background()).Return(FileInfo{
2019-08-25 20:10:55 +00:00
SizeIsDeferred: true,
Size: 0,
}, nil),
)
2018-04-28 19:39:46 +00:00
handler, _ := NewHandler(Config{
StoreComposer: composer,
2018-04-28 19:39:46 +00:00
})
(&httpTest{
Method: "HEAD",
URL: "yes",
ReqHeader: map[string]string{
"Tus-Resumable": "1.0.0",
},
Code: http.StatusOK,
ResHeader: map[string]string{
"Upload-Defer-Length": "1",
},
}).Run(handler, t)
})
SubTest(t, "NoDeferLengthHeader", func(t *testing.T, store *MockFullDataStore, composer *StoreComposer) {
2019-08-25 20:10:55 +00:00
ctrl := gomock.NewController(t)
defer ctrl.Finish()
upload := NewMockFullUpload(ctrl)
2018-04-28 19:39:46 +00:00
gomock.InOrder(
store.EXPECT().GetUpload(context.Background(), "yes").Return(upload, nil),
upload.EXPECT().GetInfo(context.Background()).Return(FileInfo{
2018-04-28 19:39:46 +00:00
SizeIsDeferred: false,
Size: 10,
2018-04-28 19:39:46 +00:00
}, nil),
)
handler, _ := NewHandler(Config{
StoreComposer: composer,
2018-04-28 19:39:46 +00:00
})
(&httpTest{
Method: "HEAD",
URL: "yes",
ReqHeader: map[string]string{
"Tus-Resumable": "1.0.0",
},
Code: http.StatusOK,
ResHeader: map[string]string{
"Upload-Defer-Length": "",
},
}).Run(handler, t)
})
2015-02-09 18:37:06 +00:00
}