(t *testing.T, querierTestConfig *querier.TestConfig)
| 239 | } |
| 240 | |
| 241 | func testSetup(t *testing.T, querierTestConfig *querier.TestConfig) (*promql.Engine, storage.QueryableFunc, Pusher, log.Logger, RulesLimits, prometheus.Registerer) { |
| 242 | tracker := promql.NewActiveQueryTracker(t.TempDir(), 20, promslog.NewNopLogger()) |
| 243 | |
| 244 | timeout := time.Minute * 2 |
| 245 | |
| 246 | if querierTestConfig != nil && querierTestConfig.Cfg.Timeout != 0 { |
| 247 | timeout = querierTestConfig.Cfg.Timeout |
| 248 | } |
| 249 | engine := promql.NewEngine(promql.EngineOpts{ |
| 250 | MaxSamples: 1e6, |
| 251 | ActiveQueryTracker: tracker, |
| 252 | Timeout: timeout, |
| 253 | }) |
| 254 | |
| 255 | // Mock the pusher |
| 256 | pusher := newPusherMock() |
| 257 | pusher.MockPush(&cortexpb.WriteResponse{}, nil) |
| 258 | |
| 259 | l := log.NewLogfmtLogger(os.Stdout) |
| 260 | l = level.NewFilter(l, level.AllowInfo()) |
| 261 | |
| 262 | reg := prometheus.NewRegistry() |
| 263 | queryable := testQueryableFunc(querierTestConfig, reg, l) |
| 264 | |
| 265 | return engine, queryable, pusher, l, &ruleLimits{maxRuleGroups: 20, maxRulesPerRuleGroup: 15}, reg |
| 266 | } |
| 267 | |
| 268 | func newManager(t *testing.T, cfg Config) *DefaultMultiTenantManager { |
| 269 | engine, queryable, pusher, logger, overrides, reg := testSetup(t, nil) |
no test coverage detected