26 lines
558 B
Ruby
26 lines
558 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
require 'spec_helper'
|
||
|
|
|
||
|
|
RSpec.describe Gitlab::Database::Consistency do
|
||
|
|
let(:session) do
|
||
|
|
Gitlab::Database::LoadBalancing::Session.current
|
||
|
|
end
|
||
|
|
|
||
|
|
describe '.with_read_consistency' do
|
||
|
|
it 'sticks to primary database' do
|
||
|
|
expect(session).not_to be_using_primary
|
||
|
|
|
||
|
|
block = -> (&control) do
|
||
|
|
described_class.with_read_consistency do
|
||
|
|
expect(session).to be_using_primary
|
||
|
|
|
||
|
|
control.call
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
expect { |probe| block.call(&probe) }.to yield_control
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|