tusd/concat_test.go

package tusd_test

import (
	"net/http"
	"testing"

	"github.com/stretchr/testify/assert"

	. "github.com/tus/tusd"
)

type concatPartialStore struct {
	t *assert.Assertions
	zeroStore
}

func (s concatPartialStore) NewUpload(info FileInfo) (string, error) {
	s.t.True(info.IsPartial)
	s.t.False(info.IsFinal)
	s.t.Nil(info.PartialUploads)

	return "foo", nil
}

func (s concatPartialStore) GetInfo(id string) (FileInfo, error) {
	return FileInfo{
		IsPartial: true,
	}, nil
}

func (s concatPartialStore) ConcatUploads(id string, uploads []string) error {
	return nil
}

func TestConcatPartial(t *testing.T) {
	handler, _ := NewHandler(Config{
		MaxSize:  400,
		BasePath: "files",
		DataStore: concatPartialStore{
			t: assert.New(t),
		},
	})

	(&httpTest{
		Name:   "Successful OPTIONS request",
		Method: "OPTIONS",
		URL:    "",
		ResHeader: map[string]string{
			"Tus-Extension": "creation,concatenation",
		},
		Code: http.StatusNoContent,
	}).Run(handler, t)

	(&httpTest{
		Name:   "Successful POST request",
		Method: "POST",
		ReqHeader: map[string]string{
			"Tus-Resumable": "1.0.0",
			"Upload-Length": "300",
			"Upload-Concat": "partial",
		},
		Code: http.StatusCreated,
	}).Run(handler, t)

	(&httpTest{
		Name:   "Successful HEAD request",
		Method: "HEAD",
		URL:    "foo",
		ReqHeader: map[string]string{
			"Tus-Resumable": "1.0.0",
		},
		Code: http.StatusNoContent,
		ResHeader: map[string]string{
			"Upload-Concat": "partial",
		},
	}).Run(handler, t)
}

type concatFinalStore struct {
	t *assert.Assertions
	zeroStore
}

func (s concatFinalStore) NewUpload(info FileInfo) (string, error) {
	s.t.False(info.IsPartial)
	s.t.True(info.IsFinal)
	s.t.Equal([]string{"a", "b"}, info.PartialUploads)

	return "foo", nil
}

func (s concatFinalStore) GetInfo(id string) (FileInfo, error) {
	if id == "a" || id == "b" {
		return FileInfo{
			IsPartial: true,
			Size:      5,
			Offset:    5,
		}, nil
	}

	if id == "c" {
		return FileInfo{
			IsPartial: true,
			Size:      5,
			Offset:    3,
		}, nil
	}

	if id == "foo" {
		return FileInfo{
			IsFinal:        true,
			PartialUploads: []string{"a", "b"},
			Size:           10,
			Offset:         10,
		}, nil
	}

	return FileInfo{}, ErrNotFound
}

func (s concatFinalStore) ConcatUploads(id string, uploads []string) error {
	s.t.Equal("foo", id)
	s.t.Equal([]string{"a", "b"}, uploads)

	return nil
}

func TestConcatFinal(t *testing.T) {
	a := assert.New(t)

	handler, _ := NewHandler(Config{
		MaxSize:  400,
		BasePath: "files",
		DataStore: concatFinalStore{
			t: a,
		},
		NotifyCompleteUploads: true,
	})

	c := make(chan FileInfo, 1)
	handler.CompleteUploads = c

	(&httpTest{
		Name:   "Successful POST request",
		Method: "POST",
		ReqHeader: map[string]string{
			"Tus-Resumable": "1.0.0",
			"Upload-Concat": "final; http://tus.io/files/a /files/b/",
		},
		Code: http.StatusCreated,
	}).Run(handler, t)

	info := <-c
	a.Equal("foo", info.ID)
	a.Equal(int64(10), info.Size)
	a.Equal(int64(10), info.Offset)
	a.False(info.IsPartial)
	a.True(info.IsFinal)
	a.Equal([]string{"a", "b"}, info.PartialUploads)

	(&httpTest{
		Name:   "Successful HEAD request",
		Method: "HEAD",
		URL:    "foo",
		ReqHeader: map[string]string{
			"Tus-Resumable": "1.0.0",
		},
		Code: http.StatusNoContent,
		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)

	(&httpTest{
		Name:   "Concatenating non finished upload (id: c)",
		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)

	handler, _ = NewHandler(Config{
		MaxSize:  9,
		BasePath: "files",
		DataStore: concatFinalStore{
			t: assert.New(t),
		},
	})

	(&httpTest{
		Name:   "Exceeding MaxSize",
		Method: "POST",
		ReqHeader: map[string]string{
			"Tus-Resumable": "1.0.0",
			"Upload-Concat": "final; http://tus.io/files/a /files/b/",
		},
		Code: http.StatusRequestEntityTooLarge,
	}).Run(handler, t)
}