spec/support/redis.rb


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

# frozen_string_literal: true
require 'gitlab/redis'

RSpec.configure do |config|
  config.after(:each, :redis) do
    Sidekiq.redis(&:flushdb)
    redis_queues_metadata_cleanup!
  end

  Gitlab::Redis::ALL_CLASSES.each do |instance_class|
    underscored_name = instance_class.store_name.underscore

    config.around(:each, :"clean_gitlab_redis_#{underscored_name}") do |example|
      public_send("redis_#{underscored_name}_cleanup!")
      redis_queues_metadata_cleanup! if underscored_name == 'queues'

      example.run

      public_send("redis_#{underscored_name}_cleanup!")
      redis_queues_metadata_cleanup! if underscored_name == 'queues'
    end
  end

  config.before(:suite) do
    Gitlab::Redis::ALL_CLASSES.each do |instance_class|
      instance_class.with(&:flushdb)
    end
  end

  config.before(:each, :use_null_store_as_repository_cache) do |example|
    null_store = ActiveSupport::Cache::NullStore.new

    allow(Gitlab::Redis::RepositoryCache).to receive(:cache_store).and_return(null_store)
  end
end