File tree 2 files changed +13
-1
lines changed
2 files changed +13
-1
lines changed Original file line number Diff line number Diff line change @@ -221,7 +221,7 @@ def primary_key_is_an_integer?
221
221
if defined? ( @primary_key_is_an_integer )
222
222
@primary_key_is_an_integer
223
223
else
224
- @primary_key_is_an_integer = !ANCESTRY_UNCAST_TYPES . include? ( type_for_attribute ( primary_key ) )
224
+ @primary_key_is_an_integer = !ANCESTRY_UNCAST_TYPES . include? ( type_for_attribute ( primary_key ) . type )
225
225
end
226
226
end
227
227
end
Original file line number Diff line number Diff line change @@ -123,4 +123,16 @@ def test_setup_test_nodes
123
123
end
124
124
end
125
125
end
126
+
127
+ def test_primary_key_is_an_integer
128
+ AncestryTestDatabase . with_model ( extra_columns : { string_id : :string } ) do |model |
129
+ model . primary_key = :string_id
130
+
131
+ assert !model . primary_key_is_an_integer?
132
+ end
133
+
134
+ AncestryTestDatabase . with_model do |model |
135
+ assert model . primary_key_is_an_integer?
136
+ end
137
+ end
126
138
end
You can’t perform that action at this time.
0 commit comments