vitalif
/
etcd
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge pull request #8612 from lorneli/clientv3_integration 

clientv3/integration: test leasing txn invalidates deleted cache
release-3.3
Xiang Li 2017-10-02 12:29:35 -07:00 committed by GitHub
parent f361dcc639 aac652009d
commit 3a566fd3ad
1 changed files with 70 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

70
clientv3/integration/leasing_test.go
View File

@ -686,6 +686,76 @@ func TestLeasingTxnOwnerGet(t *testing.T) {
}
}
func TestLeasingTxnOwnerDeleteRange(t *testing.T) {
defer testutil.AfterTest(t)
clus := integration.NewClusterV3(t, &integration.ClusterConfig{Size: 1})
defer clus.Terminate(t)
lkv, closeLKV, err := leasing.NewKV(clus.Client(0), "pfx/")
testutil.AssertNil(t, err)
defer closeLKV()
keyCount := rand.Intn(10) + 1
for i := 0; i < keyCount; i++ {
k := fmt.Sprintf("k-%d", i)
if _, perr := clus.Client(0).Put(context.TODO(), k, k+k); perr != nil {
t.Fatal(perr)
}
}
// cache in lkv
resp, err := lkv.Get(context.TODO(), "k-", clientv3.WithPrefix())
if err != nil {
t.Fatal(err)
}
if len(resp.Kvs) != keyCount {
t.Fatalf("expected %d keys, got %d", keyCount, len(resp.Kvs))
}
if _, terr := lkv.Txn(context.TODO()).Then(clientv3.OpDelete("k-", clientv3.WithPrefix())).Commit(); terr != nil {
t.Fatal(terr)
}
resp, err = lkv.Get(context.TODO(), "k-", clientv3.WithPrefix())
if err != nil {
t.Fatal(err)
}
if len(resp.Kvs) != 0 {
t.Fatalf("expected no keys, got %d", len(resp.Kvs))
}
}
func TestLeasingTxnOwnerDelete(t *testing.T) {
defer testutil.AfterTest(t)
clus := integration.NewClusterV3(t, &integration.ClusterConfig{Size: 1})
defer clus.Terminate(t)
lkv, closeLKV, err := leasing.NewKV(clus.Client(0), "pfx/")
testutil.AssertNil(t, err)
defer closeLKV()
if _, err = clus.Client(0).Put(context.TODO(), "k", "abc"); err != nil {
t.Fatal(err)
}
// cache in lkv
if _, gerr := lkv.Get(context.TODO(), "k"); gerr != nil {
t.Fatal(gerr)
}
if _, terr := lkv.Txn(context.TODO()).Then(clientv3.OpDelete("k")).Commit(); terr != nil {
t.Fatal(terr)
}
resp, err := lkv.Get(context.TODO(), "k")
if err != nil {
t.Fatal(err)
}
if len(resp.Kvs) != 0 {
t.Fatalf("expected no keys, got %d", len(resp.Kvs))
}
}
func TestLeasingTxnOwnerIf(t *testing.T) {
defer testutil.AfterTest(t)
clus := integration.NewClusterV3(t, &integration.ClusterConfig{Size: 1})