Test cache in robustness tests

Signed-off-by: Marek Siarkowicz <[email protected]>

Author: serathius

tests/robustness/client/client.go

@@ -24,6 +24,7 @@ import (
 	"go.uber.org/zap"
 
 	"go.etcd.io/etcd/api/v3/mvccpb"
+	"go.etcd.io/etcd/cache/v3"
 	clientv3 "go.etcd.io/etcd/client/v3"
 	"go.etcd.io/etcd/tests/v3/robustness/identity"
 	"go.etcd.io/etcd/tests/v3/robustness/model"
@@ -64,6 +65,14 @@ func NewRecordingClient(endpoints []string, ids identity.Provider, baseTime time
 	if err != nil {
 		return nil, err
 	}
+	c, err := cache.New(cc.Watcher, "")
+	if err != nil {
+		return nil, err
+	}
+	cc.Watcher = &cacheWatcher{
+		Cache:   c,
+		Watcher: cc.Watcher,
+	}
 	return &RecordingClient{
 		ID:           ids.NewClientID(),
 		client:       *cc,
@@ -72,6 +81,24 @@ func NewRecordingClient(endpoints []string, ids identity.Provider, baseTime time
 	}, nil
 }
 
+type cacheWatcher struct {
+	Cache   *cache.Cache
+	Watcher clientv3.Watcher
+}
+
+func (cw *cacheWatcher) Watch(ctx context.Context, key string, opts ...clientv3.OpOption) clientv3.WatchChan {
+	return cw.Cache.Watch(ctx, key, opts...)
+}
+
+func (cw *cacheWatcher) RequestProgress(ctx context.Context) error {
+	return cw.Watcher.RequestProgress(ctx)
+}
+
+func (cw *cacheWatcher) Close() error {
+	cw.Cache.Close()
+	return cw.Watcher.Close()
+}
+
 func (c *RecordingClient) Close() error {
 	return c.client.Close()
 }