jobrunaggregatorlib

type AggregationJobClient interface {
	// GetJobRunForJobNameBeforeTime returns the jobRun closest to, but BEFORE, the time provided.
	// This is useful for bounding a query of GCS buckets in a window.
	// nil means that no jobRun was found before the specified time.
	GetJobRunForJobNameBeforeTime(ctx context.Context, jobName string, targetTime time.Time) (string, error)
	// GetJobRunForJobNameAfterTime returns the jobRun closest to, but AFTER, the time provided.
	// This is useful for bounding a query of GCS buckets in a window.
	// nil means that no jobRun as found after the specified time.
	GetJobRunForJobNameAfterTime(ctx context.Context, jobName string, targetTime time.Time) (string, error)

	// GetBackendDisruptionRowCountByJob gets the row count for disruption data for one job
	GetBackendDisruptionRowCountByJob(ctx context.Context, jobName, masterNodesUpdated string) (uint64, error)

	// GetBackendDisruptionStatisticsByJob gets the mean and p95 disruption per backend from the week from 10 days ago.
	GetBackendDisruptionStatisticsByJob(ctx context.Context, jobName, masterNodesUpdated string) ([]jobrunaggregatorapi.BackendDisruptionStatisticsRow, error)

	ListAggregatedTestRunsForJob(ctx context.Context, frequency, jobName string, startDay time.Time) ([]jobrunaggregatorapi.AggregatedTestRunRow, error)
}

type AvailabilityResult struct {
	ServerName         string
	SecondsUnavailable int
}

type BackendDisruption struct {
	// Name ensure self-identification
	Name string
	// ConnectionType is New or Reused
	ConnectionType string
	// LoadBalancerType is external/internal/service-network/localhost
	// we're only interested in
	LoadBalancerType   string
	DisruptedDuration  v1.Duration
	DisruptionMessages []string
}

type BackendDisruptionList struct {
	// BackendDisruptions is keyed by name to make the consumption easier
	BackendDisruptions map[string]*BackendDisruption
}

type BigQueryDataCoordinates struct {
	ProjectID string
	DataSetID string
}

type BigQueryInserter interface {
	Put(ctx context.Context, src interface{}) (err error)
}

type BigQueryJobRunWaiter struct {
	JobRunGetter      JobRunGetter
	TimeToStopWaiting time.Time
}

type CIDataClient interface {
	JobLister
	AggregationJobClient
	HistoricalDataClient

	// these deal with release tags
	ListReleaseTags(ctx context.Context) (map[string]bool, error)

	// GetLastJobRunEndTimeFromTable returns the last uploaded job runs EndTime in the given table.
	GetLastJobRunEndTimeFromTable(ctx context.Context, table string) (*time.Time, error)

	ListUploadedJobRunIDsSinceFromTable(ctx context.Context, table string, since *time.Time) (map[string]bool, error)

	ListAllKnownAlerts(ctx context.Context) ([]*jobrunaggregatorapi.KnownAlertRow, error)

	// ListReleases lists all releases from the new release table
	ListReleases(ctx context.Context) ([]jobrunaggregatorapi.ReleaseRow, error)
}

type CIGCSClient interface {
	ReadJobRunFromGCS(ctx context.Context, jobGCSRootLocation, jobName, jobRunID string, logger logrus.FieldLogger) (jobrunaggregatorapi.JobRunInfo, error)
	ReadRelatedJobRuns(ctx context.Context, jobName, gcsPrefix, startingJobRunID, endingJobRunID string,
		matcherFunc ProwJobMatcherFunc) ([]jobrunaggregatorapi.JobRunInfo, error)
}

type ClusterData struct {
	MasterNodesUpdated string
}

type ClusterJobRunWaiter struct {
	ProwJobClient      *prowjobclientset.Clientset
	TimeToStopWaiting  time.Time
	ProwJobMatcherFunc ProwJobMatcherFunc
}

type GoogleAuthenticationFlags struct {
	TokenFileLocation string
	// location of a credential file described by https://cloud.google.com/docs/authentication/production
	GoogleServiceAccountCredentialFile string
	GoogleOAuthClientCredentialFile    string
}

type HistoricalDataClient interface {
	ListDisruptionHistoricalData(ctx context.Context) ([]jobrunaggregatorapi.HistoricalData, error)
	ListAlertHistoricalData(ctx context.Context) ([]*jobrunaggregatorapi.AlertHistoricalDataRow, error)
}

type JobLister interface {
	ListAllJobsWithVariants(ctx context.Context) ([]jobrunaggregatorapi.JobRowWithVariants, error)
	GetJobVariants(ctx context.Context, jobName string) (*jobrunaggregatorapi.JobRowWithVariants, error)
	ListAllJobs(ctx context.Context) ([]jobrunaggregatorapi.JobRow, error)

	// ListProwJobRunsSince lists from the testplatform BigQuery dataset in a separate project from
	// where we normally operate. Job runs are inserted here just after their GCS artifacts are uploaded.
	// This function is used for importing runs we do not yet have into our tables.
	ListProwJobRunsSince(ctx context.Context, since *time.Time) ([]*jobrunaggregatorapi.TestPlatformProwJobRow, error)
}

type JobRunGetter interface {
	// GetRelatedJobRuns gets all related job runs for analysis
	GetRelatedJobRuns(ctx context.Context) ([]jobrunaggregatorapi.JobRunInfo, error)

	// GetRelatedJobRunsFromIdentifiers passes along minimal information known about the jobs already so that we can skip
	// querying and go directly to fetching the full job details when GetRelatedJobRuns is called
	GetRelatedJobRunsFromIdentifiers(ctx context.Context, jobRunIdentifiers []JobRunIdentifier) ([]jobrunaggregatorapi.JobRunInfo, error)
}

type JobRunIdentifier struct {
	JobName  string
	JobRunID string
}

type JobRunLocator interface {
	FindRelatedJobs(ctx context.Context) ([]jobrunaggregatorapi.JobRunInfo, error)
	FindJob(ctx context.Context, jobRunID string) (jobrunaggregatorapi.JobRunInfo, error)
}

type JobRunWaiter interface {
	// Wait waits until all job runs finish, or time out
	Wait(ctx context.Context) ([]JobRunIdentifier, error)
}

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

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

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

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

type ProwJobMatcherFunc func(prowJob *prowjobv1.ProwJob) bool

type RowCount struct {
	TotalRows int64
}

type TestCaseDetails struct {
	Name          string
	TestSuiteName string
	// Summary is filled in during the pass/fail calculation
	Summary string

	Passes   []TestCasePass
	Failures []TestCaseFailure
	Skips    []TestCaseSkip
}

type TestCaseFailure struct {
	JobRunID       string
	HumanURL       string
	GCSArtifactURL string
}

type TestCaseNeverExecuted struct {
	JobRunID       string
	HumanURL       string
	GCSArtifactURL string
}

type TestCasePass struct {
	JobRunID       string
	HumanURL       string
	GCSArtifactURL string
}

type TestCaseSkip struct {
	JobRunID       string
	HumanURL       string
	GCSArtifactURL string
}

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