Add latest changes from gitlab-org/gitlab@16-7-stable-eev16.7.0-rc42

1 files changed, 53 insertions, 0 deletions

diff --git a/spec/lib/gitlab/usage/metrics/instrumentations/unique_users_all_imports_metric_spec.rb b/spec/lib/gitlab/usage/metrics/instrumentations/unique_users_all_imports_metric_spec.rb
new file mode 100644
index 00000000000..4fdabb86e23
--- /dev/null
+++ b/spec/lib/gitlab/usage/metrics/instrumentations/unique_users_all_imports_metric_spec.rb
@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::Usage::Metrics::Instrumentations::UniqueUsersAllImportsMetric, feature_category: :importers do
+  let(:expected_value) { 6 }
+  let(:expected_query) do
+    <<~SQL.squish
+    SELECT
+    (SELECT COUNT(DISTINCT "projects"."creator_id") FROM "projects" WHERE "projects"."import_type" IS NOT NULL) +
+    (SELECT COUNT(DISTINCT "bulk_imports"."user_id") FROM "bulk_imports") +
+    (SELECT COUNT(DISTINCT "jira_imports"."user_id") FROM "jira_imports") +
+    (SELECT COUNT(DISTINCT "csv_issue_imports"."user_id") FROM "csv_issue_imports") +
+    (SELECT COUNT(DISTINCT "group_import_states"."user_id") FROM "group_import_states")
+    SQL
+  end
+
+  before_all do
+    import = create :jira_import_state, created_at: 3.days.ago
+    create :jira_import_state, created_at: 35.days.ago
+    create :jira_import_state, created_at: 3.days.ago, user: import.user
+
+    create :group_import_state, created_at: 3.days.ago
+    create :issue_csv_import, created_at: 3.days.ago
+    create :bulk_import, created_at: 3.days.ago
+    create :project, import_type: :jira, created_at: 3.days.ago
+  end
+
+  before do
+    described_class::IMPORTS_METRICS.each do |submetric_class|
+      metric = submetric_class.new(time_frame: time_frame, options: options)
+      allow(metric.send(:relation).connection).to receive(:transaction_open?).and_return(false)
+    end
+  end
+
+  it_behaves_like 'a correct instrumented metric value and query', { time_frame: 'all' }
+
+  it_behaves_like 'a correct instrumented metric value and query', { time_frame: '28d' } do
+    let(:expected_value) { 5 }
+    let(:start) { 30.days.ago.to_fs(:db) }
+    let(:finish) { 2.days.ago.to_fs(:db) }
+    let(:expected_query) do
+      <<~SQL.squish
+      SELECT
+      (SELECT COUNT(DISTINCT "projects"."creator_id") FROM "projects" WHERE "projects"."import_type" IS NOT NULL AND "projects"."created_at" BETWEEN '#{start}' AND '#{finish}') +
+      (SELECT COUNT(DISTINCT "bulk_imports"."user_id") FROM "bulk_imports" WHERE "bulk_imports"."created_at" BETWEEN '#{start}' AND '#{finish}') +
+      (SELECT COUNT(DISTINCT "jira_imports"."user_id") FROM "jira_imports" WHERE "jira_imports"."created_at" BETWEEN '#{start}' AND '#{finish}') +
+      (SELECT COUNT(DISTINCT "csv_issue_imports"."user_id") FROM "csv_issue_imports" WHERE "csv_issue_imports"."created_at" BETWEEN '#{start}' AND '#{finish}') +
+      (SELECT COUNT(DISTINCT "group_import_states"."user_id") FROM "group_import_states" WHERE "group_import_states"."created_at" BETWEEN '#{start}' AND '#{finish}')
+      SQL
+    end
+  end
+end