types

type Cluster struct {
	GRPCEndpoint string `yaml:"grpcEndpoint"`

	AdminClient    admin.Client    `yaml:"-"`
	FrontendClient frontend.Client `yaml:"-"`
}

type Operation struct {
	Type          ReplicationSimulationOperation `yaml:"op"`
	At            time.Duration                  `yaml:"at"`
	Cluster       string                         `yaml:"cluster"`
	SourceCluster string                         `yaml:"sourceCluster"`
	TargetCluster string                         `yaml:"targetCluster"`

	WorkflowType                         string                              `yaml:"workflowType"`
	WorkflowID                           string                              `yaml:"workflowID"`
	WorkflowExecutionStartToCloseTimeout time.Duration                       `yaml:"workflowExecutionStartToCloseTimeout"`
	WorkflowDuration                     time.Duration                       `yaml:"workflowDuration"`
	ActivityCount                        int                                 `yaml:"activityCount"`
	DelayStartSeconds                    int32                               `yaml:"delayStartSeconds"`
	CronSchedule                         string                              `yaml:"cronSchedule"`
	ActiveClusterSelectionPolicy         *types.ActiveClusterSelectionPolicy `yaml:"activeClusterSelectionPolicy"`

	Query            string `yaml:"query"`
	ConsistencyLevel string `yaml:"consistencyLevel"`

	SignalName  string `yaml:"signalName"`
	SignalInput any    `yaml:"signalInput"`

	EventID int64 `yaml:"eventID"`

	Domain                    string            `yaml:"domain"`
	NewActiveCluster          string            `yaml:"newActiveCluster"`
	NewActiveClustersByRegion map[string]string `yaml:"newActiveClustersByRegion"`
	FailoverTimeout           *int32            `yaml:"failoverTimeoutSec"`

	// RunIDKey specifies a key to store/retrieve RunID for this operation
	RunIDKey string `yaml:"runIDKey"`

	Want Validation `yaml:"want"`
}

type OperationFunction func(t *testing.T, op *Operation, simCfg *ReplicationSimulationConfig) error

type ReplicationDomainConfig struct {
	ActiveClusterName string `yaml:"activeClusterName"`

	ActiveClustersByRegion map[string]string `yaml:"activeClustersByRegion"`
}

type ReplicationSimulation struct {
	RunIDRegistry map[string]string
}

type ReplicationSimulationConfig struct {
	// Clusters is the map of all clusters
	Clusters map[string]*Cluster `yaml:"clusters"`

	// PrimaryCluster is used for domain registration
	PrimaryCluster string `yaml:"primaryCluster"`

	Domains map[string]ReplicationDomainConfig `yaml:"domains"`

	Operations []*Operation `yaml:"operations"`
}

type ReplicationSimulationOperation string

type Validation struct {
	Status                      string `yaml:"status"`
	StartedByWorkersInCluster   string `yaml:"startedByWorkersInCluster"`
	CompletedByWorkersInCluster string `yaml:"completedByWorkersInCluster"`
	Error                       string `yaml:"error"`
	QueryResult                 any    `yaml:"queryResult"`
}

type WorkflowInput struct {
	Duration      time.Duration
	ActivityCount int
}

type WorkflowOutput struct {
	Count int
}