mirror of
https://github.com/etcd-io/etcd.git
synced 2024-09-27 06:25:44 +00:00
59 lines
1.9 KiB
Go
59 lines
1.9 KiB
Go
// Copyright 2023 The etcd Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package validate
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/anishathalye/porcupine"
|
|
"go.uber.org/zap"
|
|
|
|
"go.etcd.io/etcd/tests/v3/robustness/model"
|
|
"go.etcd.io/etcd/tests/v3/robustness/traffic"
|
|
)
|
|
|
|
// ValidateAndReturnVisualize returns visualize as porcupine.linearizationInfo used to generate visualization is private.
|
|
func ValidateAndReturnVisualize(t *testing.T, lg *zap.Logger, cfg Config, reports []traffic.ClientReport) (visualize func(basepath string) error) {
|
|
eventHistory := validateWatch(t, cfg, reports)
|
|
allOperations := operations(reports)
|
|
watchEvents := uniqueWatchEvents(reports)
|
|
patchedOperations := patchOperationsWithWatchEvents(allOperations, watchEvents)
|
|
return validateOperationsAndVisualize(t, lg, patchedOperations, eventHistory)
|
|
}
|
|
|
|
func operations(reports []traffic.ClientReport) []porcupine.Operation {
|
|
var ops []porcupine.Operation
|
|
for _, r := range reports {
|
|
ops = append(ops, r.OperationHistory.Operations()...)
|
|
}
|
|
return ops
|
|
}
|
|
|
|
func uniqueWatchEvents(reports []traffic.ClientReport) map[model.EtcdOperation]traffic.TimedWatchEvent {
|
|
persisted := map[model.EtcdOperation]traffic.TimedWatchEvent{}
|
|
for _, r := range reports {
|
|
for _, resp := range r.Watch {
|
|
for _, event := range resp.Events {
|
|
persisted[event.Op] = traffic.TimedWatchEvent{Time: resp.Time, WatchEvent: event}
|
|
}
|
|
}
|
|
}
|
|
return persisted
|
|
}
|
|
|
|
type Config struct {
|
|
ExpectRevisionUnique bool
|
|
}
|