testutils

type BenchmarkingT interface {
	Error(args ...interface{})
	ResetTimer()
	ReportAllocs()
	N() int
}

type CollectorTestSuite struct {
	// NewCollector function to create collector instances for testing
	NewCollector func() collector.Collector

	// ExpectedMetrics lists the metric names that should be collected
	ExpectedMetrics []string

	// RequiresWindows indicates if the collector requires Windows to run
	RequiresWindows bool

	// MinimumMetricCount is the minimum number of metrics expected
	MinimumMetricCount int
}

type MockCollector struct {
	// Name of the collector for identification
	Name string

	// Metrics to return when Collect() is called
	Metrics []prometheus.Metric

	// Error to return when Collect() is called (if not nil)
	CollectError error

	// CollectCallCount tracks how many times Collect was called
	CollectCallCount int

	// CollectDelay simulates collection time
	CollectDelay time.Duration

	// IsEnabled controls whether the collector should be considered enabled
	IsEnabled bool
}

type MockRegistry struct {
	// Metrics to return when Gather() is called
	MetricFamilies []*dto.MetricFamily

	// Error to return when Gather() is called
	GatherError error

	// GatherCallCount tracks how many times Gather was called
	GatherCallCount int
}

type TestingT interface {
	Helper()
	Error(args ...interface{})
	Errorf(format string, args ...interface{})
	Run(name string, f func(TestingT)) bool
}