2019-12-04 20:38:33 +05:30
|
|
|
# frozen_string_literal: true
|
|
|
|
|
2019-09-04 21:01:54 +05:30
|
|
|
require 'spec_helper'
|
|
|
|
|
2020-07-28 23:09:34 +05:30
|
|
|
RSpec.describe Gitlab::PhabricatorImport::WorkerState, :clean_gitlab_redis_shared_state do
|
2019-09-04 21:01:54 +05:30
|
|
|
subject(:state) { described_class.new('weird-project-id') }
|
2019-12-21 20:55:43 +05:30
|
|
|
|
2019-09-04 21:01:54 +05:30
|
|
|
let(:key) { 'phabricator-import/jobs/project-weird-project-id/job-count' }
|
|
|
|
|
|
|
|
describe '#add_job' do
|
|
|
|
it 'increments the counter for jobs' do
|
|
|
|
set_value(3)
|
|
|
|
|
|
|
|
expect { state.add_job }.to change { get_value }.from('3').to('4')
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
describe '#remove_job' do
|
|
|
|
it 'decrements the counter for jobs' do
|
|
|
|
set_value(3)
|
|
|
|
|
|
|
|
expect { state.remove_job }.to change { get_value }.from('3').to('2')
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
describe '#running_count' do
|
|
|
|
it 'reads the value' do
|
|
|
|
set_value(9)
|
|
|
|
|
|
|
|
expect(state.running_count).to eq(9)
|
|
|
|
end
|
|
|
|
|
|
|
|
it 'returns 0 when nothing was set' do
|
|
|
|
expect(state.running_count).to eq(0)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def set_value(value)
|
|
|
|
redis.with { |r| r.set(key, value) }
|
|
|
|
end
|
|
|
|
|
|
|
|
def get_value
|
|
|
|
redis.with { |r| r.get(key) }
|
|
|
|
end
|
|
|
|
|
|
|
|
def redis
|
|
|
|
Gitlab::Redis::SharedState
|
|
|
|
end
|
|
|
|
end
|