diff options
Diffstat (limited to 'spec/support/helpers/usage_data_helpers.rb')
-rw-r--r-- | spec/support/helpers/usage_data_helpers.rb | 17 |
1 files changed, 6 insertions, 11 deletions
diff --git a/spec/support/helpers/usage_data_helpers.rb b/spec/support/helpers/usage_data_helpers.rb index fab775dd404..d92fcdc2d4a 100644 --- a/spec/support/helpers/usage_data_helpers.rb +++ b/spec/support/helpers/usage_data_helpers.rb @@ -88,6 +88,8 @@ module UsageDataHelpers projects_jira_active projects_jira_server_active projects_jira_cloud_active + projects_jira_dvcs_cloud_active + projects_jira_dvcs_server_active projects_slack_active projects_slack_slash_commands_active projects_custom_issue_tracker_active @@ -136,6 +138,7 @@ module UsageDataHelpers USAGE_DATA_KEYS = %i( active_user_count counts + counts_monthly recorded_at edition version @@ -160,6 +163,7 @@ module UsageDataHelpers web_ide_clientside_preview_enabled ingress_modsecurity_enabled object_store + topology ).freeze def stub_usage_data_connections @@ -220,17 +224,8 @@ module UsageDataHelpers ) end - def expect_prometheus_api_to(*receive_matchers) - expect_next_instance_of(Gitlab::PrometheusClient) do |client| - receive_matchers.each { |m| expect(client).to m } - end - end - - def allow_prometheus_queries - allow_next_instance_of(Gitlab::PrometheusClient) do |client| - allow(client).to receive(:aggregate).and_return({}) - allow(client).to receive(:query).and_return({}) - end + def expect_prometheus_client_to(*receive_matchers) + receive_matchers.each { |m| expect(prometheus_client).to m } end def for_defined_days_back(days: [29, 2]) |