spec/support/factory_bot.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
36
37
38
39
40
41

# frozen_string_literal: true

FactoryBot.define do
  after(:build) do |object, _|
    next unless object.respond_to?(:factory_bot_built=)

    object.factory_bot_built = true
  end

  before(:create) do |object, _|
    next unless object.respond_to?(:factory_bot_built=)

    object.factory_bot_built = false
  end
end

FactoryBot::SyntaxRunner.class_eval do
  include RSpec::Mocks::ExampleMethods
  include StubMethodCalls

  # FactoryBot doesn't allow yet to add a helper that can be used in factories
  # While the fixture_file_upload helper is reasonable to be used there:
  #
  # https://github.com/thoughtbot/factory_bot/issues/564#issuecomment-389491577
  def fixture_file_upload(*args, **kwargs)
    Rack::Test::UploadedFile.new(*args, **kwargs)
  end
end

# Patching FactoryBot to allow stubbing non AR models
# See https://github.com/thoughtbot/factory_bot/pull/1466
module Gitlab
  module FactoryBotStubPatch
    def has_settable_id?(result_instance)
      result_instance.class.respond_to?(:primary_key) &&
        result_instance.class.primary_key
    end
  end
end

FactoryBot::Strategy::Stub.prepend(Gitlab::FactoryBotStubPatch)