Add latest changes from gitlab-org/gitlab@master

1 files changed, 49 insertions, 0 deletions

diff --git a/spec/initializers/active_record_transaction_observer_spec.rb b/spec/initializers/active_record_transaction_observer_spec.rb
new file mode 100644
index 00000000000..a834037dce5
--- /dev/null
+++ b/spec/initializers/active_record_transaction_observer_spec.rb
@@ -0,0 +1,49 @@
+# frozen_string_literal: true
+
+require 'spec_helper'
+
+RSpec.describe 'ActiveRecord Transaction Observer', feature_category: :application_performance do
+  def load_initializer
+    load Rails.root.join('config/initializers/active_record_transaction_observer.rb')
+  end
+
+  context 'when DBMS is available' do
+    before do
+      allow_next_instance_of(ActiveRecord::Base.connection) do |connection| # rubocop:disable Database/MultipleDatabases
+        allow(connection).to receive(:active?).and_return(true)
+      end
+    end
+
+    it 'calls Gitlab::Database::Transaction::Observer' do
+      allow(Feature::FlipperFeature).to receive(:table_exists?).and_return(true)
+
+      expect(Gitlab::Database::Transaction::Observer).to receive(:register!)
+
+      load_initializer
+    end
+
+    context 'when flipper table does not exist' do
+      before do
+        allow(Feature::FlipperFeature).to receive(:table_exists?).and_raise(ActiveRecord::NoDatabaseError)
+      end
+
+      it 'does not calls Gitlab::Database::Transaction::Observer' do
+        expect(Gitlab::Database::Transaction::Observer).not_to receive(:register!)
+
+        load_initializer
+      end
+    end
+  end
+
+  context 'when DBMS is not available' do
+    before do
+      allow(ActiveRecord::Base).to receive(:connection).and_raise(PG::ConnectionBad)
+    end
+
+    it 'does not calls Gitlab::Database::Transaction::Observer' do
+      expect(Gitlab::Database::Transaction::Observer).not_to receive(:register!)
+
+      load_initializer
+    end
+  end
+end