blob: 54512068ee75eb5893ef909808115bd5ae37a35e (
plain) (
blame)
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
|
# frozen_string_literal: true
require "cases/helper"
module ActiveRecord
module AttributeMethods
class ReadTest < ActiveRecord::TestCase
FakeColumn = Struct.new(:name) do
def type; :integer; end
end
def setup
@klass = Class.new(Class.new { def self.initialize_generated_modules; end }) do
def self.superclass; Base; end
def self.base_class?; true; end
def self.decorate_matching_attribute_types(*); end
include ActiveRecord::DefineCallbacks
include ActiveRecord::AttributeMethods
def self.attribute_names
%w{ one two three }
end
def self.primary_key
end
def self.columns
attribute_names.map { FakeColumn.new(name) }
end
def self.columns_hash
Hash[attribute_names.map { |name|
[name, FakeColumn.new(name)]
}]
end
end
end
def test_define_attribute_methods
instance = @klass.new
@klass.attribute_names.each do |name|
assert_not_includes instance.methods.map(&:to_s), name
end
@klass.define_attribute_methods
@klass.attribute_names.each do |name|
assert_includes instance.methods.map(&:to_s), name, "#{name} is not defined"
end
end
def test_attribute_methods_generated?
assert_not @klass.method_defined?(:one)
@klass.define_attribute_methods
assert @klass.method_defined?(:one)
end
end
end
end
|
