mutex_test.go

package etcdplus

import (
	"github.com/coreos/go-etcd/etcd"
	"testing"
	"time"
)

func TestMutex(t *testing.T) {
	c := etcd.NewClient()
	uuid := getUUID()
	mutex := NewMutex(c, uuid)

	err := mutex.Lock(1 * time.Second)
	if err != nil {
		t.Fatal(err)
	}

	fatal := make(chan bool)
	go func() {
		err := mutex.Lock(1 * time.Second) // should not succeed
		if err == nil {
			fatal <- true
		} else {
			fatal <- false
		}
	}()

	if <-fatal {
		t.Fatal("Should not be able to acquire a lock twice.")
	}

	success := make(chan bool)
	go func() {
		err := mutex.Lock(1 * time.Second)
		if err == nil {
			success <- true
		} else {
			success <- false
			t.Fatal(err)
		}
	}()

	wait := time.After(500 * time.Millisecond)
	<-wait
	err = mutex.Unlock()
	if err != nil {
		t.Fatal(err)
	}

	if !(<-success) {
		t.Fatal("Fail to acquire lock after it's released.")
	}
}