Skip to content

Commit 91a43a7

Browse files
validate history collected by recording client
Signed-off-by: Abdur Rehman <[email protected]>
1 parent b2f243c commit 91a43a7

File tree

1 file changed

+15
-0
lines changed
  • tests/antithesis/test-template/robustness

1 file changed

+15
-0
lines changed

tests/antithesis/test-template/robustness/main.go

+15
Original file line numberDiff line numberDiff line change
@@ -19,18 +19,25 @@ package main
1919
import (
2020
"context"
2121
"fmt"
22+
"math/rand"
2223
"os"
2324
"sync"
25+
"testing"
2426
"time"
2527

2628
"github.com/antithesishq/antithesis-sdk-go/assert"
29+
"github.com/stretchr/testify/require"
30+
"go.uber.org/zap"
2731
"golang.org/x/time/rate"
2832

33+
"go.etcd.io/etcd/tests/v3/framework/e2e"
2934
"go.etcd.io/etcd/tests/v3/robustness/client"
3035
"go.etcd.io/etcd/tests/v3/robustness/identity"
3136
robustnessrand "go.etcd.io/etcd/tests/v3/robustness/random"
3237
"go.etcd.io/etcd/tests/v3/robustness/report"
38+
"go.etcd.io/etcd/tests/v3/robustness/scenarios"
3339
"go.etcd.io/etcd/tests/v3/robustness/traffic"
40+
"go.etcd.io/etcd/tests/v3/robustness/validate"
3441
)
3542

3643
var profile = traffic.Profile{
@@ -82,6 +89,14 @@ func testRobustness(ctx context.Context, baseTime time.Time, duration time.Durat
8289
assert.Reachable("Completed robustness traffic generation", nil)
8390
}
8491

92+
func validateReport(ctx context.Context, lg *zap.Logger, c *e2e.EtcdProcessCluster, s scenarios.TestScenario, t *testing.T) {
93+
r := report.TestReport{Logger: lg, Cluster: c}
94+
persistedRequests, err := report.PersistedRequestsCluster(lg, c)
95+
require.NoError(t, err)
96+
validateConfig := validate.Config{ExpectRevisionUnique: s.Traffic.ExpectUniqueRevision()}
97+
r.Visualize = validate.ValidateAndReturnVisualize(t, lg, validateConfig, r.Client, persistedRequests, 5*time.Minute).Visualize
98+
}
99+
85100
func connect(endpoint string, ids identity.Provider, baseTime time.Time) *client.RecordingClient {
86101
cli, err := client.NewRecordingClient([]string{endpoint}, ids, baseTime)
87102
if err != nil {

0 commit comments

Comments
 (0)