diff options
Diffstat (limited to 'spec/support/helpers/usage_data_helpers.rb')
-rw-r--r-- | spec/support/helpers/usage_data_helpers.rb | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/spec/support/helpers/usage_data_helpers.rb b/spec/support/helpers/usage_data_helpers.rb index df79049123d..d05676a649e 100644 --- a/spec/support/helpers/usage_data_helpers.rb +++ b/spec/support/helpers/usage_data_helpers.rb @@ -174,6 +174,22 @@ module UsageDataHelpers allow(Gitlab::Prometheus::Internal).to receive(:prometheus_enabled?).and_return(false) end + def stub_prometheus_queries + stub_request(:get, %r{^https?://::1:9090/-/ready}) + .to_return( + status: 200, + body: [{}].to_json, + headers: { 'Content-Type' => 'application/json' } + ) + + stub_request(:get, %r{^https?://::1:9090/api/v1/query\?query=.*}) + .to_return( + status: 200, + body: [{}].to_json, + headers: { 'Content-Type' => 'application/json' } + ) + end + def clear_memoized_values(values) values.each { |v| described_class.clear_memoization(v) } end @@ -242,4 +258,12 @@ module UsageDataHelpers end end end + + def load_sample_metric_definition(filename: 'sample_metric.yml') + load_metric_yaml(fixture_file("lib/generators/gitlab/usage_metric_definition_generator/#{filename}")) + end + + def load_metric_yaml(data) + ::Gitlab::Config::Loader::Yaml.new(data).load_raw! + end end |