summaryrefslogtreecommitdiff
path: root/spec/support/prometheus/metric_builders.rb
blob: c8d056d3fc882cfd16d054d8184b8895d4accbab (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
module Prometheus
  module MetricBuilders
    def simple_query(suffix = 'a', **opts)
      { query_range: "query_range_#{suffix}" }.merge(opts)
    end

    def simple_queries
      [simple_query, simple_query('b', label: 'label', unit: 'unit')]
    end

    def simple_metric(title: 'title', required_metrics: [], queries: [simple_query])
      Gitlab::Prometheus::Metric.new(title: title, required_metrics: required_metrics, weight: 1, queries: queries)
    end

    def simple_metrics(added_metric_name: 'metric_a')
      [
        simple_metric(required_metrics: %W(#{added_metric_name} metric_b), queries: simple_queries),
        simple_metric(required_metrics: [added_metric_name], queries: [simple_query('empty')]),
        simple_metric(required_metrics: %w{metric_c})
      ]
    end

    def simple_metric_group(name: 'name', metrics: simple_metrics)
      Gitlab::Prometheus::MetricGroup.new(name: name, priority: 1, metrics: metrics)
    end
  end
end