6b0d493350
So the fake can enjoy it, too. We don't use `prepend` because that'll require we change `allow_any_instance_of` to `expect_next_instance_of`, but that's not very easy to do. We can do that later.
34 lines
1 KiB
Ruby
34 lines
1 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
# This class extends an OpenStruct object by adding predicate methods to mimic
|
|
# ActiveRecord access. We rely on the initial values being true or false to
|
|
# determine whether to define a predicate method because for a newly-added
|
|
# column that has not been migrated yet, there is no way to determine the
|
|
# column type without parsing db/schema.rb.
|
|
module Gitlab
|
|
class FakeApplicationSettings < OpenStruct
|
|
include ApplicationSettingImplementation
|
|
|
|
# Mimic ActiveRecord predicate methods for boolean values
|
|
def self.define_predicate_methods(options)
|
|
options.each do |key, value|
|
|
next if key.to_s.end_with?('?')
|
|
next unless [true, false].include?(value)
|
|
|
|
define_method "#{key}?" do
|
|
actual_key = key.to_s.chomp('?')
|
|
self[actual_key]
|
|
end
|
|
end
|
|
end
|
|
|
|
def initialize(options = {})
|
|
super
|
|
|
|
FakeApplicationSettings.define_predicate_methods(options)
|
|
end
|
|
|
|
alias_method :read_attribute, :[]
|
|
alias_method :has_attribute?, :[]
|
|
end
|
|
end
|