tests: Fix return time for failed requests globally

Signed-off-by: Marek Siarkowicz <siarkowicz@google.com>
etcd-io · Dec 5, 2022 · 9d03a46 · 9d03a46
1 parent ec72022
commit 9d03a46
Show file tree

Hide file tree

Showing 3 changed files with 34 additions and 14 deletions.
diff --git a/tests/linearizability/client.go b/tests/linearizability/client.go
@@ -24,7 +24,7 @@ import (
 
 type recordingClient struct {
 	client  clientv3.Client
-	history *history
+	history *appendableHistory
 }
 
 func NewClient(endpoints []string, ids idProvider) (*recordingClient, error) {
@@ -39,7 +39,7 @@ func NewClient(endpoints []string, ids idProvider) (*recordingClient, error) {
 	}
 	return &recordingClient{
 		client:  *cc,
-		history: NewHistory(ids),
+		history: newAppendableHistory(ids),
 	}, nil
 }
 

diff --git a/tests/linearizability/history.go b/tests/linearizability/history.go
@@ -21,25 +21,26 @@ import (
 	clientv3 "go.etcd.io/etcd/client/v3"
 )
 
-type history struct {
+type appendableHistory struct {
 	// id of the next write operation. If needed a new id might be requested from idProvider.
 	id         int
 	idProvider idProvider
 
-	operations []porcupine.Operation
-	failed     []porcupine.Operation
+	history
 }
 
-func NewHistory(ids idProvider) *history {
-	return &history{
+func newAppendableHistory(ids idProvider) *appendableHistory {
+	return &appendableHistory{
 		id:         ids.ClientId(),
 		idProvider: ids,
-		operations: []porcupine.Operation{},
-		failed:     []porcupine.Operation{},
+		history: history{
+			operations: []porcupine.Operation{},
+			failed:     []porcupine.Operation{},
+		},
 	}
 }
 
-func (h *history) AppendGet(key string, start, end time.Time, resp *clientv3.GetResponse) {
+func (h *appendableHistory) AppendGet(key string, start, end time.Time, resp *clientv3.GetResponse) {
 	var readData string
 	if len(resp.Kvs) == 1 {
 		readData = string(resp.Kvs[0].Value)
@@ -53,7 +54,7 @@ func (h *history) AppendGet(key string, start, end time.Time, resp *clientv3.Get
 	})
 }
 
-func (h *history) AppendPut(key, value string, start, end time.Time, resp *clientv3.PutResponse, err error) {
+func (h *appendableHistory) AppendPut(key, value string, start, end time.Time, resp *clientv3.PutResponse, err error) {
 	if err != nil {
 		h.failed = append(h.failed, porcupine.Operation{
 			ClientId: h.id,
@@ -79,7 +80,24 @@ func (h *history) AppendPut(key, value string, start, end time.Time, resp *clien
 	})
 }
 
-func (h *history) Operations() []porcupine.Operation {
+type history struct {
+	operations []porcupine.Operation
+	failed     []porcupine.Operation
+}
+
+func (h history) Merge(h2 history) history {
+	result := history{
+		operations: make([]porcupine.Operation, 0, len(h.operations)+len(h2.operations)),
+		failed:     make([]porcupine.Operation, 0, len(h.failed)+len(h2.failed)),
+	}
+	result.operations = append(result.operations, h.operations...)
+	result.operations = append(result.operations, h2.operations...)
+	result.failed = append(result.failed, h.failed...)
+	result.failed = append(result.failed, h2.failed...)
+	return result
+}
+
+func (h history) Operations() []porcupine.Operation {
 	operations := make([]porcupine.Operation, 0, len(h.operations)+len(h.failed))
 	var maxTime int64
 	for _, op := range h.operations {

diff --git a/tests/linearizability/linearizability_test.go b/tests/linearizability/linearizability_test.go
@@ -140,11 +140,12 @@ type FailpointConfig struct {
 	waitBetweenTriggers time.Duration
 }
 
-func simulateTraffic(ctx context.Context, t *testing.T, clus *e2e.EtcdProcessCluster, config trafficConfig) (operations []porcupine.Operation) {
+func simulateTraffic(ctx context.Context, t *testing.T, clus *e2e.EtcdProcessCluster, config trafficConfig) []porcupine.Operation {
 	mux := sync.Mutex{}
 	endpoints := clus.EndpointsV3()
 
 	ids := newIdProvider()
+	h := history{}
 	limiter := rate.NewLimiter(rate.Limit(config.maximalQPS), 200)
 
 	startTime := time.Now()
@@ -162,12 +163,13 @@ func simulateTraffic(ctx context.Context, t *testing.T, clus *e2e.EtcdProcessClu
 
 			config.traffic.Run(ctx, c, limiter, ids)
 			mux.Lock()
-			operations = append(operations, c.history.Operations()...)
+			h = h.Merge(c.history.history)
 			mux.Unlock()
 		}(c)
 	}
 	wg.Wait()
 	endTime := time.Now()
+	operations := h.Operations()
 	t.Logf("Recorded %d operations", len(operations))
 
 	qps := float64(len(operations)) / float64(endTime.Sub(startTime)) * float64(time.Second)