tusd/pkg/etcd3locker/locker_test.go

73 lines
1.7 KiB
Go
Raw Normal View History

package etcd3locker
import (
"os"
"testing"
"time"
etcd_harness "github.com/chen-anders/go-etcd-harness"
"github.com/coreos/etcd/clientv3"
"github.com/stretchr/testify/assert"
"github.com/tus/tusd/pkg/handler"
)
var _ handler.Locker = &Etcd3Locker{}
func TestEtcd3Locker(t *testing.T) {
a := assert.New(t)
harness, err := etcd_harness.New(os.Stderr)
if err != nil {
t.Fatalf("failed starting etcd harness: %v", err)
}
t.Logf("will use etcd harness endpoint: %v", harness.Endpoint)
defer func() {
harness.Stop()
t.Logf("cleaned up etcd harness")
}()
client, err := clientv3.New(clientv3.Config{
Endpoints: []string{harness.Endpoint},
DialTimeout: 5 * time.Second,
})
if err != nil {
t.Fatalf("Unable to connect to etcd3: %v", err)
}
defer client.Close()
shortTTL := 3
testPrefix := "/test-tusd"
lockerOptions := NewLockerOptions(shortTTL, testPrefix)
locker, err := NewWithLockerOptions(client, lockerOptions)
a.NoError(err)
lock1, err := locker.NewLock("one")
a.NoError(err)
a.NoError(lock1.Lock())
//a.Equal(handler.ErrFileLocked, lock1.Lock())
time.Sleep(5 * time.Second)
// test that we can't take over the upload via a different etcd3 session
// while an upload is already taking place; testing etcd3 session KeepAlive
lock2, err := locker.NewLock("one")
a.NoError(err)
a.Equal(handler.ErrFileLocked, lock2.Lock())
a.NoError(lock1.Unlock())
a.Equal(ErrLockNotHeld, lock1.Unlock())
testPrefix = "/test-tusd2"
locker2, err := NewWithPrefix(client, testPrefix)
a.NoError(err)
lock3, err := locker2.NewLock("one")
a.NoError(err)
a.NoError(lock3.Lock())
a.Equal(handler.ErrFileLocked, lock3.Lock())
a.Equal(handler.ErrFileLocked, lock3.Lock())
a.NoError(lock3.Unlock())
a.Equal(ErrLockNotHeld, lock3.Unlock())
}