Add latest changes from gitlab-org/gitlab@14-5-stable-eev14.5.0-rc42

1 files changed, 106 insertions, 0 deletions

diff --git a/spec/lib/gitlab/database/migrations/observers/transaction_duration_spec.rb b/spec/lib/gitlab/database/migrations/observers/transaction_duration_spec.rb
new file mode 100644
index 00000000000..e65f89747c4
--- /dev/null
+++ b/spec/lib/gitlab/database/migrations/observers/transaction_duration_spec.rb
@@ -0,0 +1,106 @@
+# frozen_string_literal: true
+require 'spec_helper'
+
+RSpec.describe Gitlab::Database::Migrations::Observers::TransactionDuration do
+  subject(:transaction_duration_observer) { described_class.new(observation, directory_path) }
+
+  let(:observation) { Gitlab::Database::Migrations::Observation.new(migration_version, migration_name) }
+  let(:directory_path) { Dir.mktmpdir }
+  let(:log_file) { "#{directory_path}/#{migration_version}_#{migration_name}-transaction-duration.json" }
+  let(:transaction_duration) { Gitlab::Json.parse(File.read(log_file)) }
+  let(:migration_version) { 20210422152437 }
+  let(:migration_name) { 'test' }
+
+  after do
+    FileUtils.remove_entry(directory_path)
+  end
+
+  it 'records real and sub transactions duration', :delete do
+    observe
+
+    entry = transaction_duration[0]
+    start_time, end_time, transaction_type = entry.values_at('start_time', 'end_time', 'transaction_type')
+    start_time = DateTime.parse(start_time)
+    end_time = DateTime.parse(end_time)
+
+    aggregate_failures do
+      expect(transaction_duration.size).to eq(3)
+      expect(start_time).to be_before(end_time)
+      expect(transaction_type).not_to be_nil
+    end
+  end
+
+  context 'when there are sub-transactions' do
+    it 'records transaction duration' do
+      observe_sub_transaction
+
+      expect(transaction_duration.size).to eq(1)
+
+      entry = transaction_duration[0]['transaction_type']
+
+      expect(entry).to eql 'sub_transaction'
+    end
+  end
+
+  context 'when there are real-transactions' do
+    it 'records transaction duration', :delete do
+      observe_real_transaction
+
+      expect(transaction_duration.size).to eq(1)
+
+      entry = transaction_duration[0]['transaction_type']
+
+      expect(entry).to eql 'real_transaction'
+    end
+  end
+
+  private
+
+  def observe
+    transaction_duration_observer.before
+    run_transaction
+    transaction_duration_observer.after
+    transaction_duration_observer.record
+  end
+
+  def observe_sub_transaction
+    transaction_duration_observer.before
+    run_sub_transactions
+    transaction_duration_observer.after
+    transaction_duration_observer.record
+  end
+
+  def observe_real_transaction
+    transaction_duration_observer.before
+    run_real_transactions
+    transaction_duration_observer.after
+    transaction_duration_observer.record
+  end
+
+  def run_real_transactions
+    ActiveRecord::Base.transaction do
+    end
+  end
+
+  def run_sub_transactions
+    ActiveRecord::Base.transaction(requires_new: true) do
+    end
+  end
+
+  def run_transaction
+    ActiveRecord::Base.connection_pool.with_connection do |connection|
+      Gitlab::Database::SharedModel.using_connection(connection) do
+        Gitlab::Database::SharedModel.transaction do
+          Gitlab::Database::SharedModel.transaction(requires_new: true) do
+            Gitlab::Database::SharedModel.transaction do
+              Gitlab::Database::SharedModel.transaction do
+                Gitlab::Database::SharedModel.transaction(requires_new: true) do
+                end
+              end
+            end
+          end
+        end
+      end
+    end
+  end
+end