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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
|
# frozen_string_literal: true
RSpec.configure do |config|
def gitlab_sidekiq_inline
# We need to cleanup the queues before running jobs in specs because the
# middleware might have written to redis
redis_queues_cleanup!
redis_queues_metadata_cleanup!
# Scoped inline! is thread-safe which breaks capybara specs
# see https://github.com/sidekiq/sidekiq/issues/6069
Sidekiq::Testing.inline!
# Set a thread-local sidekiq capsule as it may be accessed in the
# Gitlab::SidekiqMiddleware::ConcurrencyLimit::WorkerExecutionTracker
Thread.current[:sidekiq_capsule] = Sidekiq::Capsule.new('test', Sidekiq.default_configuration)
yield
ensure
Thread.current[:sidekiq_capsule] = nil
Sidekiq::Testing.fake! # fake is the default so we reset it to that
redis_queues_cleanup!
redis_queues_metadata_cleanup!
end
def with_sidekiq_context
allow(Sidekiq).to receive(:server?).and_return(true)
end
# As we'll review the examples with this tag, we should either:
# - fix the example to not require Sidekiq inline mode (and remove this tag)
# - explicitly keep the inline mode and change the tag for `:sidekiq_inline` instead
config.around(:example, :sidekiq_might_not_need_inline) do |example|
gitlab_sidekiq_inline { example.run }
end
config.around(:example, :sidekiq_inline) do |example|
gitlab_sidekiq_inline { example.run }
end
config.before(:example, :sidekiq_inline) do
with_sidekiq_context
end
config.before(:example, :sidekiq_might_not_need_inline) do
with_sidekiq_context
end
config.before(:example, :with_sidekiq_context) do
with_sidekiq_context
end
# Some specs need to run mailers through Sidekiq explicitly, rather
# than the ActiveJob test adapter. There is a Rails bug that means we
# have to do some extra steps to make this happen:
# https://github.com/rails/rails/issues/37270
#
# In particular, we can't use an `around` hook because then the 'before' part
# of that will run before the `before_setup` hook in ActiveJob::TestHelper,
# which doesn't do what we want.
#
config.before(:example, :sidekiq_mailers) do
queue_adapter_changed_jobs.each { |k| k.queue_adapter = :sidekiq }
queue_adapter_changed_jobs.each(&:disable_test_adapter)
end
config.after(:example, :sidekiq_mailers) do
queue_adapter_changed_jobs.each do |klass|
klass.queue_adapter = :test
klass.enable_test_adapter(ActiveJob::QueueAdapters::TestAdapter.new)
end
end
end
|
