loadtest

type ChannelImportData struct {
	Team        string `json:"team"`
	Name        string `json:"name"`
	DisplayName string `json:"display_name"`
	Type        string `json:"type"`
	Header      string `json:"header,omitempty"`
	Purpose     string `json:"purpose,omitempty"`
	Scheme      string `json:"scheme,omitempty"`
}

type ClientTimingStats struct {
	Routes map[string]*RouteStats
}

type ConnectionConfiguration struct {
	ServerURL                   string
	WebsocketURL                string
	PProfURL                    string
	DriverName                  string
	DataSource                  string
	DBEndpoint                  string // deprecated
	LocalCommands               bool
	SSHHostnamePort             string
	SSHUsername                 string
	SSHPassword                 string
	SSHKey                      string
	MattermostInstallDir        string
	ConfigFileLoc               string
	AdminEmail                  string
	AdminPassword               string
	SkipBulkload                bool
	MaxIdleConns                int
	MaxIdleConnsPerHost         int
	IdleConnTimeoutMilliseconds int
}

type DirectChannelImportData struct {
	Members     *[]string `json:"members"`
	FavoritedBy *[]string `json:"favorited_by"`

	Header *string `json:"header"`
}

type EmojiImportData struct {
	Name  string `json:"name"`
	Image string `json:"image"`
}

type EntityConfig struct {
	EntityNumber        int
	EntityName          string
	EntityActions       []randutil.Choice
	UserData            UserImportData
	Users               []UserImportData
	ChannelMap          map[string]map[string]string
	TeamMap             map[string]string
	TownSquareMap       map[string]string
	Client              *model.Client4
	AdminClient         *model.Client4
	WebSocketClient     *model.WebSocketClient
	ActionRate          time.Duration
	LoadTestConfig      *LoadTestConfig
	StatusReportChannel chan<- UserEntityStatusReport
	StopChannel         <-chan bool
	StopWaitGroup       *sync.WaitGroup
	Info                map[string]interface{}
	// contains filtered or unexported fields
}

type GenerateBulkloadFileResult struct {
	File     bytes.Buffer
	Users    []UserImportData
	Teams    []TeamImportData
	Channels []ChannelImportData
	Schemes  []SchemeImportData
	Emojis   []EmojiImportData
}

type Instance struct {
	Id             string
	Index          int
	EntityStartNum int
	Seed           int64
	// contains filtered or unexported fields
}

type LayoutChannel struct {
	Num int
}

type LayoutTeam struct {
	Num      int
	Volume   int
	Channels []LayoutChannel
}

type LineImportData struct {
	Type          string                   `json:"type"`
	Scheme        *SchemeImportData        `json:"scheme,omitempty"`
	Team          *TeamImportData          `json:"team,omitempty"`
	Channel       *ChannelImportData       `json:"channel,omitempty"`
	User          *UserImportData          `json:"user,omitempty"`
	Post          *PostImportData          `json:"post,omitempty"`
	DirectChannel *DirectChannelImportData `json:"direct_channel,omitempty"`
	Emoji         *EmojiImportData         `json:"emoji,omitempty"`
	Version       int                      `json:"version"`
}

type LoadTestConfig struct {
	LoadtestEnviromentConfig  LoadtestEnviromentConfig
	ConnectionConfiguration   ConnectionConfiguration
	UserEntitiesConfiguration UserEntitiesConfiguration
	ResultsConfiguration      ResultsConfiguration
	LogSettings               LoggerSettings
}

type LoadtestEnviromentConfig struct {
	NumTeams                  int
	NumChannelsPerTeam        int
	NumPrivateChannelsPerTeam int
	NumDirectMessageChannels  int
	NumGroupMessageChannels   int
	NumUsers                  int
	NumTeamSchemes            int
	NumChannelSchemes         int
	NumEmoji                  int
	NumPlugins                int

	PercentHighVolumeChannels float64
	PercentMidVolumeChannels  float64
	PercentLowVolumeChannels  float64

	PercentUsersHighVolumeChannel float64
	PercentUsersMidVolumeChannel  float64
	PercentUsersLowVolumeChannel  float64

	PercentHighVolumeTeams float64
	PercentMidVolumeTeams  float64
	PercentLowVolumeTeams  float64

	PercentUsersHighVolumeTeams float64
	PercentUsersMidVolumeTeams  float64
	PercentUsersLowVolumeTeams  float64

	HighVolumeTeamSelectionWeight int
	MidVolumeTeamSelectionWeight  int
	LowVolumeTeamSelectionWeight  int

	HighVolumeChannelSelectionWeight int
	MidVolumeChannelSelectionWeight  int
	LowVolumeChannelSelectionWeight  int

	NumPosts      int
	PostTimeRange int64
	ReplyChance   float64

	PercentCustomSchemeTeams    float64
	PercentCustomSchemeChannels float64
}

type LoggerSettings struct {
	EnableConsole bool
	ConsoleJson   bool
	ConsoleLevel  string
	EnableFile    bool
	FileJson      bool
	FileLevel     string
	FileLocation  string
}

type MattermostLocalConnection struct {
	// contains filtered or unexported fields
}

type MattermostSSHConnection struct {
	Client *ssh.Client
	// contains filtered or unexported fields
}

type OrganizationLayout struct {
	Config             *LoadtestEnviromentConfig
	TeamNameToIdMap    map[string]string
	ChannelNameToIdMap map[string]string // Assumes unique channel names
}

type PostImportData struct {
	Team    *string `json:"team"`
	Channel *string `json:"channel"`
	User    *string `json:"user"`

	Message  *string `json:"message"`
	CreateAt *int64  `json:"create_at"`

	FlaggedBy *[]string `json:"flagged_by"`
}

type ResultsConfiguration struct {
	PProfDelayMinutes int
	PProfLength       int
}

type RoleImportData struct {
	Name        string   `json:"name"`
	DisplayName string   `json:"display_name"`
	Description string   `json:"description"`
	Permissions []string `json:"permissions"`
}

type RouteStatResults struct {
	Max                float64
	Min                float64
	Mean               float64
	Median             float64
	Percentile90       float64
	Percentile95       float64
	InterQuartileRange float64
}

type RouteStats struct {
	Name               string
	NumHits            int64
	NumErrors          int64
	ErrorRate          float64
	Duration           []float64
	DurationLastMinute *ratecounter.AvgRateCounter `json:"-"`
	Max                float64
	Min                float64
	Mean               float64
	Median             float64
	Percentile90       float64
	Percentile95       float64
	InterQuartileRange float64
}

type SchemeImportData struct {
	Name                    string          `json:"name"`
	DisplayName             string          `json:"display_name"`
	Description             string          `json:"description"`
	Scope                   string          `json:"scope"`
	DefaultTeamAdminRole    *RoleImportData `json:"default_team_admin_role,omitempty"`
	DefaultTeamUserRole     *RoleImportData `json:"default_team_user_role,omitempty"`
	DefaultChannelAdminRole *RoleImportData `json:"default_channel_admin_role,omitempty"`
	DefaultChannelUserRole  *RoleImportData `json:"default_channel_user_role,omitempty"`
}

type ServerCLICommandRunner interface {
	RunCommand(string) (bool, string)
	RunPlatformCommand(string) (bool, string)
	SendLoadtestFile(buf *bytes.Buffer) error
	Close() error
}

type ServerSetupData struct {
	// TeamIdMap maps team name to team id
	TeamIdMap map[string]string
	// ChannelIdMap maps team name and channel name to channel id
	ChannelIdMap map[string]map[string]string
	// TownSquareIdMap maps team name to the channel id of the corresponding default channel.
	TownSquareIdMap map[string]string

	BulkloadResult GenerateBulkloadFileResult
}

type TeamImportData struct {
	Name            string `json:"name"`
	DisplayName     string `json:"display_name"`
	Type            string `json:"type"`
	Description     string `json:"description,omitempty"`
	AllowOpenInvite bool   `json:"allow_open_invite,omitempty"`
	Scheme          string `json:"scheme,omitempty"`
}

type TestRun struct {
	UserEntities []randutil.Choice
}

type TimedRoundTripper struct {
	// contains filtered or unexported fields
}

type TimedRoundTripperReport struct {
	Method          string
	Path            string
	RequestDuration time.Duration
	StatusCode      int
}

type UserChannelImportData struct {
	Name  string `json:"name"`
	Roles string `json:"roles"`
}

type UserEntitiesConfiguration struct {
	TestLengthMinutes                 int
	NumActiveEntities                 int
	ActionRateMilliseconds            int
	ActionRateMaxVarianceMilliseconds int
	EnableRequestTiming               bool
	ChannelLinkChance                 float64
	UploadImageChance                 float64
	LinkPreviewChance                 float64
	CustomEmojiChance                 float64
	CustomEmojiReactionChance         float64
	SystemEmojiReactionChance         float64
	NeedsProfilesByIdChance           float64
	NeedsProfilesByUsernameChance     float64
	NeedsProfileStatusChance          float64
	DoStatusPolling                   bool
	RandomizeEntitySelection          bool
	UserProfileUpdateFullnameChance   float64
	UserProfileUpdateUsernameChance   float64
	UserProfileUpdateNicknameChance   float64
	UserProfileUpdatePositionChance   float64
	UserProfileUpdateEmailChance      float64
	UserProfileUpdateImageChance      float64
	PublicChannelCreationChance       float64
	PrivateChannelCreationChance      float64
	DirectChannelCreationChance       float64
	GroupChannelCreationChance        float64
	NumPostReactionsPerUser           int
	PostReactionsRateMilliseconds     int
	NumPostsGetBeforeAfter            int
	GetPostsAroundLastUnreadChance    float64
	NumGetPostsAroundLastUnread       int
}

type UserEntity struct {
	Name    string
	Actions []randutil.Choice
}

type UserEntityRates struct {
	RateDescription   string
	ErrorRate         *ratecounter.RateCounter
	ActionSendRate    *ratecounter.RateCounter
	ActionRecieveRate *ratecounter.RateCounter
}

type UserEntityStat map[string]int64

type UserEntityStatistics struct {
	StartTime time.Time

	TotalErrors               UserEntityStat
	TotalEntitiesActive       UserEntityStat
	TotalEntitiesLaunching    UserEntityStat
	TotalEntitiesFailedLaunch UserEntityStat
	TotalEntitiesFailedActive UserEntityStat
	TotalEntitiesStopped      UserEntityStat

	UserEntityRatesPerSecond UserEntityRates
	UserEntityRatesPerMinute UserEntityRates
	UserEntityRatesPerHour   UserEntityRates

	RouteTimings map[string]ewma.MovingAverage
	Routes       []string
}

type UserEntityStatusReport struct {
	Status  int
	Err     error
	Details string
	Config  *EntityConfig
}

type UserEntityWithRateMultiplier struct {
	Entity         UserEntity
	RateMultiplier float64
}

type UserImportData struct {
	Username    string `json:"username"`
	Email       string `json:"email"`
	AuthService string `json:"auth_service,omitempty"`
	AuthData    string `json:"auth_data,omitempty"`
	Password    string `json:"password,omitempty"`
	Nickname    string `json:"nickname,omitempty"`
	FirstName   string `json:"first_name,omitempty"`
	LastName    string `json:"last_name,omitempty"`
	Position    string `json:"position,omitempty"`
	Roles       string `json:"roles"`
	Locale      string `json:"locale,omitempty"`

	Teams      []UserTeamImportData `json:"teams"`
	TeamChoice []randutil.Choice    `json:"-"`

	Theme              string `json:"theme,omitempty"`
	SelectedFont       string `json:"display_font,omitempty"`
	UseMilitaryTime    string `json:"military_time,omitempty"`
	NameFormat         string `json:"teammate_name_display,omitempty"`
	CollapsePreviews   string `json:"link_previews,omitempty"`
	MessageDisplay     string `json:"message_display,omitempty"`
	ChannelDisplayMode string `json:"channel_display_mode,omitempty"`
}

type UserTeamImportData struct {
	Name          string                  `json:"name"`
	Roles         string                  `json:"roles"`
	Channels      []UserChannelImportData `json:"channels"`
	ChannelChoice []randutil.Choice       `json:"-"`
}

type VersionImportData struct {
	Type    string `json:"type"`
	Version string `json:"version"`
}