multi_threaded_migration_spec.rb 1.13 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41
require 'spec_helper'

describe Gitlab::Database::MultiThreadedMigration do
  let(:migration) do
    Class.new { include Gitlab::Database::MultiThreadedMigration }.new
  end

  describe '#connection' do
    after do
      Thread.current[described_class::MULTI_THREAD_AR_CONNECTION] = nil
    end

    it 'returns the thread-local connection if present' do
      Thread.current[described_class::MULTI_THREAD_AR_CONNECTION] = 10

      expect(migration.connection).to eq(10)
    end

    it 'returns the global connection if no thread-local connection was set' do
      expect(migration.connection).to eq(ActiveRecord::Base.connection)
    end
  end

  describe '#with_multiple_threads' do
    it 'starts multiple threads and yields the supplied block in every thread' do
      output = Queue.new

      migration.with_multiple_threads(2) do
        output << migration.connection.execute('SELECT 1')
      end

      expect(output.size).to eq(2)
    end

    it 'joins the threads when the join parameter is set' do
      expect_any_instance_of(Thread).to receive(:join).and_call_original

      migration.with_multiple_threads(1) { }
    end
  end
end