1 files changed, 35 insertions, 0 deletions
diff --git a/spec/lib/gitlab/usage/metrics/instrumentations/count_user_auth_metric_spec.rb b/spec/lib/gitlab/usage/metrics/instrumentations/count_user_auth_metric_spec.rb
new file mode 100644
index 00000000000..2f49c427bd0
--- /dev/null
+++ b/spec/lib/gitlab/usage/metrics/instrumentations/count_user_auth_metric_spec.rb
@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe Gitlab::Usage::Metrics::Instrumentations::CountUserAuthMetric do
+  context 'with all time frame' do
+    let(:expected_value) { 2 }
+
+    before do
+      user = create(:user)
+      user2 = create(:user)
+      create(:authentication_event, user: user, provider: :ldapmain, result: :success)
+      create(:authentication_event, user: user2, provider: :ldapsecondary, result: :success)
+      create(:authentication_event, user: user2, provider: :group_saml, result: :success)
+      create(:authentication_event, user: user2, provider: :group_saml, result: :success)
+      create(:authentication_event, user: user, provider: :group_saml, result: :failed)
+    end
+
+    it_behaves_like 'a correct instrumented metric value', { time_frame: 'all', data_source: 'database' }
+  end
+
+  context 'with 28d time frame' do
+    let(:expected_value) { 1 }
+
+    before do
+      user = create(:user)
+      user2 = create(:user)
+
+      create(:authentication_event, created_at: 1.year.ago, user: user, provider: :ldapmain, result: :success)
+      create(:authentication_event, created_at: 1.week.ago, user: user2, provider: :ldapsecondary, result: :success)
+    end
+
+    it_behaves_like 'a correct instrumented metric value', { time_frame: '28d', data_source: 'database' }
+  end
+end