package tusd_test import ( "net/http" "strings" "testing" "github.com/golang/mock/gomock" "github.com/stretchr/testify/assert" . "github.com/tus/tusd" ) func TestConcat(t *testing.T) { SubTest(t, "ExtensionDiscovery", func(t *testing.T, store *MockFullDataStore) { composer := NewStoreComposer() composer.UseCore(store) composer.UseConcater(store) handler, _ := NewHandler(Config{ StoreComposer: composer, }) (&httpTest{ Method: "OPTIONS", Code: http.StatusOK, ResHeader: map[string]string{ "Tus-Extension": "creation,creation-with-upload,creation-defer-length,concatenation", }, }).Run(handler, t) }) SubTest(t, "Partial", func(t *testing.T, store *MockFullDataStore) { SubTest(t, "Create", func(t *testing.T, store *MockFullDataStore) { store.EXPECT().NewUpload(FileInfo{ Size: 300, IsPartial: true, IsFinal: false, PartialUploads: nil, MetaData: make(map[string]string), }).Return("foo", nil) handler, _ := NewHandler(Config{ BasePath: "files", DataStore: store, }) (&httpTest{ Method: "POST", ReqHeader: map[string]string{ "Tus-Resumable": "1.0.0", "Upload-Length": "300", "Upload-Concat": "partial", }, Code: http.StatusCreated, }).Run(handler, t) }) SubTest(t, "Status", func(t *testing.T, store *MockFullDataStore) { store.EXPECT().GetInfo("foo").Return(FileInfo{ IsPartial: true, }, nil) handler, _ := NewHandler(Config{ BasePath: "files", DataStore: store, }) (&httpTest{ Method: "HEAD", URL: "foo", ReqHeader: map[string]string{ "Tus-Resumable": "1.0.0", }, Code: http.StatusOK, ResHeader: map[string]string{ "Upload-Concat": "partial", }, }).Run(handler, t) }) }) SubTest(t, "Final", func(t *testing.T, store *MockFullDataStore) { SubTest(t, "Create", func(t *testing.T, store *MockFullDataStore) { a := assert.New(t) gomock.InOrder( store.EXPECT().GetInfo("a").Return(FileInfo{ IsPartial: true, Size: 5, Offset: 5, }, nil), store.EXPECT().GetInfo("b").Return(FileInfo{ IsPartial: true, Size: 5, Offset: 5, }, nil), store.EXPECT().NewUpload(FileInfo{ Size: 10, IsPartial: false, IsFinal: true, PartialUploads: []string{"a", "b"}, MetaData: make(map[string]string), }).Return("foo", nil), store.EXPECT().ConcatUploads("foo", []string{"a", "b"}).Return(nil), ) handler, _ := NewHandler(Config{ BasePath: "files", DataStore: store, NotifyCompleteUploads: true, }) c := make(chan FileInfo, 1) handler.CompleteUploads = c (&httpTest{ Method: "POST", ReqHeader: map[string]string{ "Tus-Resumable": "1.0.0", // A space between `final;` and the first URL should be allowed due to // compatibility reasons, even if the specification does not define // it. Therefore this character is included in this test case. "Upload-Concat": "final; http://tus.io/files/a /files/b/", }, Code: http.StatusCreated, }).Run(handler, t) info := <-c a.Equal("foo", info.ID) a.EqualValues(10, info.Size) a.EqualValues(10, info.Offset) a.False(info.IsPartial) a.True(info.IsFinal) a.Equal([]string{"a", "b"}, info.PartialUploads) }) SubTest(t, "Status", func(t *testing.T, store *MockFullDataStore) { store.EXPECT().GetInfo("foo").Return(FileInfo{ IsFinal: true, PartialUploads: []string{"a", "b"}, Size: 10, Offset: 10, }, nil) handler, _ := NewHandler(Config{ BasePath: "files", DataStore: store, }) (&httpTest{ Method: "HEAD", URL: "foo", ReqHeader: map[string]string{ "Tus-Resumable": "1.0.0", }, Code: http.StatusOK, ResHeader: map[string]string{ "Upload-Concat": "final;http://tus.io/files/a http://tus.io/files/b", "Upload-Length": "10", "Upload-Offset": "10", }, }).Run(handler, t) }) SubTest(t, "CreateWithUnfinishedFail", func(t *testing.T, store *MockFullDataStore) { // This upload is still unfinished (mismatching offset and size) and // will therefore cause the POST request to fail. store.EXPECT().GetInfo("c").Return(FileInfo{ IsPartial: true, Size: 5, Offset: 3, }, nil) handler, _ := NewHandler(Config{ BasePath: "files", DataStore: store, }) (&httpTest{ Method: "POST", ReqHeader: map[string]string{ "Tus-Resumable": "1.0.0", "Upload-Concat": "final;http://tus.io/files/c", }, Code: http.StatusBadRequest, }).Run(handler, t) }) SubTest(t, "CreateExceedingMaxSizeFail", func(t *testing.T, store *MockFullDataStore) { store.EXPECT().GetInfo("huge").Return(FileInfo{ Size: 1000, Offset: 1000, }, nil) handler, _ := NewHandler(Config{ MaxSize: 100, BasePath: "files", DataStore: store, }) (&httpTest{ Method: "POST", ReqHeader: map[string]string{ "Tus-Resumable": "1.0.0", "Upload-Concat": "final;/files/huge", }, Code: http.StatusRequestEntityTooLarge, }).Run(handler, t) }) SubTest(t, "UploadToFinalFail", func(t *testing.T, store *MockFullDataStore) { store.EXPECT().GetInfo("foo").Return(FileInfo{ Size: 10, Offset: 0, IsFinal: true, }, nil) handler, _ := NewHandler(Config{ DataStore: store, }) (&httpTest{ Method: "PATCH", URL: "foo", ReqHeader: map[string]string{ "Tus-Resumable": "1.0.0", "Content-Type": "application/offset+octet-stream", "Upload-Offset": "5", }, ReqBody: strings.NewReader("hello"), Code: http.StatusForbidden, }).Run(handler, t) }) SubTest(t, "InvalidConcatHeaderFail", func(t *testing.T, store *MockFullDataStore) { handler, _ := NewHandler(Config{ DataStore: store, }) (&httpTest{ Method: "POST", URL: "", ReqHeader: map[string]string{ "Tus-Resumable": "1.0.0", "Upload-Concat": "final;", }, Code: http.StatusBadRequest, }).Run(handler, t) }) }) }