Merge pull request #8612 from lorneli/clientv3_integration

clientv3/integration: test leasing txn invalidates deleted cache

clientv3/integration/leasing_test.go

@@ -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})