github.com/insionng/yougam@v0.0.0-20170714101924-2bc18d833463/libraries/ngaut/go-zookeeper/zk/lock_test.go (about)

     1  package zk
     2  
     3  import (
     4  	"testing"
     5  	"time"
     6  )
     7  
     8  func TestLock(t *testing.T) {
     9  	ts, err := StartTestCluster(1)
    10  	if err != nil {
    11  		t.Fatal(err)
    12  	}
    13  	defer ts.Stop()
    14  	zk, err := ts.ConnectAll()
    15  	if err != nil {
    16  		t.Fatalf("Connect returned error: %+v", err)
    17  	}
    18  	defer zk.Close()
    19  
    20  	acls := WorldACL(PermAll)
    21  
    22  	l := NewLock(zk, "/test", acls)
    23  	if err := l.Lock(); err != nil {
    24  		t.Fatal(err)
    25  	}
    26  	if err := l.Unlock(); err != nil {
    27  		t.Fatal(err)
    28  	}
    29  
    30  	val := make(chan int, 3)
    31  
    32  	if err := l.Lock(); err != nil {
    33  		t.Fatal(err)
    34  	}
    35  
    36  	l2 := NewLock(zk, "/test", acls)
    37  	go func() {
    38  		if err := l2.Lock(); err != nil {
    39  			t.Fatal(err)
    40  		}
    41  		val <- 2
    42  		if err := l2.Unlock(); err != nil {
    43  			t.Fatal(err)
    44  		}
    45  		val <- 3
    46  	}()
    47  	time.Sleep(time.Millisecond * 100)
    48  
    49  	val <- 1
    50  	if err := l.Unlock(); err != nil {
    51  		t.Fatal(err)
    52  	}
    53  	if x := <-val; x != 1 {
    54  		t.Fatalf("Expected 1 instead of %d", x)
    55  	}
    56  	if x := <-val; x != 2 {
    57  		t.Fatalf("Expected 2 instead of %d", x)
    58  	}
    59  	if x := <-val; x != 3 {
    60  		t.Fatalf("Expected 3 instead of %d", x)
    61  	}
    62  }
    63  
    64  // This tests creating a lock with a path that's more than 1 node deep (e.g. "/test-multi-level/lock"),
    65  // when a part of that path already exists (i.e. "/test-multi-level" node already exists).
    66  func TestMultiLevelLock(t *testing.T) {
    67  	ts, err := StartTestCluster(1)
    68  	if err != nil {
    69  		t.Fatal(err)
    70  	}
    71  	defer ts.Stop()
    72  	zk, err := ts.ConnectAll()
    73  	if err != nil {
    74  		t.Fatalf("Connect returned error: %+v", err)
    75  	}
    76  	defer zk.Close()
    77  
    78  	acls := WorldACL(PermAll)
    79  	path := "/test-multi-level"
    80  	if p, err := zk.Create(path, []byte{1, 2, 3, 4}, 0, WorldACL(PermAll)); err != nil {
    81  		t.Fatalf("Create returned error: %+v", err)
    82  	} else if p != path {
    83  		t.Fatalf("Create returned different path '%s' != '%s'", p, path)
    84  	}
    85  	l := NewLock(zk, "/test-multi-level/lock", acls)
    86  	defer zk.Delete("/test-multi-level", -1) // Clean up what we've created for this test
    87  	defer zk.Delete("/test-multi-level/lock", -1)
    88  	if err := l.Lock(); err != nil {
    89  		t.Fatal(err)
    90  	}
    91  	if err := l.Unlock(); err != nil {
    92  		t.Fatal(err)
    93  	}
    94  }