tusd/limitedstore/limitedstore_test.go

84 lines
1.6 KiB
Go

package limitedstore
import (
"github.com/tus/tusd"
"io"
"testing"
)
type dataStore struct {
t *testing.T
firstUploadCreated bool
uploadTerminated bool
}
func (store *dataStore) NewUpload(info tusd.FileInfo) (string, error) {
if !store.firstUploadCreated {
if info.Size != 80 {
store.t.Errorf("expect size to be 80, got %v", info.Size)
}
store.firstUploadCreated = true
return "1", nil
}
if info.Size != 50 {
store.t.Errorf("expect size to be 50, got %v", info.Size)
}
return "2", nil
}
func (store *dataStore) WriteChunk(id string, offset int64, src io.Reader) (int64, error) {
return 0, nil
}
func (store *dataStore) GetInfo(id string) (tusd.FileInfo, error) {
return tusd.FileInfo{}, nil
}
func (store *dataStore) GetReader(id string) (io.Reader, error) {
return nil, tusd.ErrNotImplemented
}
func (store *dataStore) Terminate(id string) error {
if id != "1" {
store.t.Errorf("expect first upload to be terminated, got %v", id)
}
store.uploadTerminated = true
return nil
}
func TestLimitedStore(t *testing.T) {
dataStore := &dataStore{
t: t,
}
store := New(100, dataStore)
// Create new upload (80 bytes)
id, err := store.NewUpload(tusd.FileInfo{
Size: 80,
})
if err != nil {
t.Fatal(err)
}
if id != "1" {
t.Errorf("expected first upload to be created, got %v", id)
}
// Create new upload (50 bytes)
id, err = store.NewUpload(tusd.FileInfo{
Size: 50,
})
if err != nil {
t.Fatal(err)
}
if id != "2" {
t.Errorf("expected second upload to be created, got %v", id)
}
if !dataStore.uploadTerminated {
t.Error("expected first upload to be terminated")
}
}