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
|
# frozen_string_literal: true
FactoryBot.define do
after(:build) do |object, _|
next unless object.respond_to?(:factory_bot_built=)
# This will help the PreventCrossDatabaseModification to temporarily
# allow the object table when it's saved later.
object.factory_bot_built = true
end
before(:create) do |_object, _|
Thread.current[:factory_bot_objects] ||= 0
Thread.current[:factory_bot_objects] += 1
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)
|
