wasp

type Alert struct {
	Annotations struct {
		DashboardUID string `json:"__dashboardUid__"`
		OrgID        string `json:"__orgId__"`
		PanelID      string `json:"__panelId__"`
		Description  string `json:"description"`
		RunbookURL   string `json:"runbook_url"`
		Summary      string `json:"summary"`
	} `json:"annotations"`
	EndsAt      time.Time `json:"endsAt"`
	Fingerprint string    `json:"fingerprint"`
	Receivers   []struct {
		Active       interface{} `json:"active"`
		Integrations interface{} `json:"integrations"`
		Name         string      `json:"name"`
	} `json:"receivers"`
	StartsAt time.Time `json:"startsAt"`
	Status   struct {
		InhibitedBy []interface{} `json:"inhibitedBy"`
		SilencedBy  []interface{} `json:"silencedBy"`
		State       string        `json:"state"`
	} `json:"status"`
	UpdatedAt    time.Time         `json:"updatedAt"`
	GeneratorURL string            `json:"generatorURL"`
	Labels       map[string]string `json:"labels"`
}

type AlertChecker struct {
	URL                 string
	APIKey              string
	RequirementLabelKey string
	T                   *testing.T
	// contains filtered or unexported fields
}

type AlertGroupsResponse struct {
	Alerts []Alert `json:"alerts"`
	Labels struct {
		Alertname     string `json:"alertname"`
		GrafanaFolder string `json:"grafana_folder"`
	} `json:"labels"`
	Receiver struct {
		Active       interface{} `json:"active"`
		Integrations interface{} `json:"integrations"`
		Name         string      `json:"name"`
	} `json:"receiver"`
}

type CallResult struct {
	Failed     bool          `json:"failed,omitempty"`
	Timeout    bool          `json:"timeout,omitempty"`
	Duration   time.Duration `json:"duration"`
	StartedAt  *time.Time    `json:"started_at,omitempty"`
	FinishedAt *time.Time    `json:"finished_at,omitempty"`
	Data       interface{}   `json:"data,omitempty"`
	Error      string        `json:"error,omitempty"`
}

type Config struct {
	T                 *testing.T
	GenName           string
	LoadType          string
	Labels            map[string]string
	LokiConfig        *LokiConfig
	Schedule          []*Segment
	CallResultBufLen  int
	StatsPollInterval time.Duration
	CallTimeout       time.Duration
	Gun               Gun
	VU                VirtualUser
	Logger            zerolog.Logger
	SharedData        interface{}
	// contains filtered or unexported fields
}

type Dashboard struct{}

type Generator struct {
	Log zerolog.Logger

	ResponsesWaitGroup *sync.WaitGroup

	ResponsesCtx context.Context

	ResponsesChan chan CallResult
	// contains filtered or unexported fields
}

type Gun interface {
	Call(l *Generator) CallResult
}

type HTTPMockServer struct {
	Sleep time.Duration
	// contains filtered or unexported fields
}

type HTTPMockServerConfig struct {
	FirstAPILatency   time.Duration
	FirstAPIHTTPCode  int
	SecondAPILatency  time.Duration
	SecondAPIHTTPCode int
}

type LocalLogger struct{}

type LokiClient struct {
	lokiClient.Client
}

type LokiConfig struct {
	// URL url to Loki endpoint
	URL string `yaml:"url"`
	// Token is Loki authorization token
	Token string `yaml:"token"`
	// BatchWait max time to wait until sending a new batch
	BatchWait time.Duration
	// BatchSize size of a messages batch
	BatchSize int
	// Timeout is batch send timeout
	Timeout time.Duration
	// BackoffConfig backoff configuration
	BackoffConfig backoff.Config
	// Headers are additional request headers
	Headers map[string]string
	// The tenant ID to use when pushing logs to Loki (empty string means
	// single tenant mode)
	TenantID string
	// When enabled, Promtail will not retry batches that get a
	// 429 'Too Many Requests' response from the distributor. Helps
	// prevent HOL blocking in multitenant deployments.
	DropRateLimitedBatches bool
	// ExposePrometheusMetrics if enabled exposes Promtail Prometheus metrics
	ExposePrometheusMetrics bool
	MaxStreams              int
	MaxLineSize             int
	MaxLineSizeTruncate     bool
}

type MockGun struct {
	Data []string
	// contains filtered or unexported fields
}

type MockGunConfig struct {
	// FailRatio in percentage, 0-100
	FailRatio int
	// TimeoutRatio in percentage, 0-100
	TimeoutRatio int
	// CallSleep time spent waiting inside a call
	CallSleep time.Duration
}

type MockHTTPGun struct {
	Data []string
	// contains filtered or unexported fields
}

type MockHTTPGunConfig struct {
	TargetURL string
}

type MockVirtualUser struct {
	Data []string
	// contains filtered or unexported fields
}

type MockVirtualUserConfig struct {
	// FailRatio in percentage, 0-100
	FailRatio int
	// TimeoutRatio in percentage, 0-100
	TimeoutRatio int
	// CallSleep time spent waiting inside a call
	CallSleep time.Duration
}

type MockWSServer struct {
	// Logf controls where logs are sent.
	Logf  func(f string, v ...interface{})
	Sleep time.Duration
}

type Profile struct {
	Generators []*Generator
}

type ProfileGunPart struct {
	Name     string
	Schedule []*Segment
	Gun      Gun
}

type ProfileVUPart struct {
	Name     string
	Schedule []*Segment
	VU       VirtualUser
}

type ResponseData struct {
	OKData *SliceBuffer[any]

	OKResponses *SliceBuffer[CallResult]

	FailResponses *SliceBuffer[CallResult]
	// contains filtered or unexported fields
}

type Segment struct {
	From         int64
	Increase     int64
	Steps        int64
	StepDuration time.Duration
	// contains filtered or unexported fields
}

type SliceBuffer[T any] struct {
	Idx      int
	Capacity int
	Data     []T
}

type Stats struct {
	CurrentRPS     atomic.Int64 `json:"currentRPS"`
	CurrentVUs     atomic.Int64 `json:"currentVUs"`
	LastSegment    atomic.Int64 `json:"last_segment"`
	CurrentSegment atomic.Int64 `json:"current_schedule_segment"`
	CurrentStep    atomic.Int64 `json:"current_schedule_step"`
	RunStopped     atomic.Bool  `json:"runStopped"`
	RunFailed      atomic.Bool  `json:"runFailed"`
	Success        atomic.Int64 `json:"success"`
	Failed         atomic.Int64 `json:"failed"`
	CallTimeout    atomic.Int64 `json:"callTimeout"`
}

type VirtualUser interface {
	Call(l *Generator)
	Stop(l *Generator)
	Clone(l *Generator) VirtualUser
	Setup(l *Generator) error
	Teardown(l *Generator) error
	StopChan() chan struct{}
}

type WSMockVU struct {
	Data []string
	// contains filtered or unexported fields
}

type WSMockVUConfig struct {
	TargetURl string
}

type WaspAlert struct {
	Name                 string
	AlertType            string
	TestName             string
	GenName              string
	RequirementGroupName string
	AlertIf              alert.ConditionEvaluator
	CustomAlert          timeseries.Option
}