Commit db23b672 authored by Stan Hu's avatar Stan Hu Committed by Toon Claes

Fix failing spec in spec/controllers/admin/hooks_controller_spec.rb

attr_encrypted expects models to have their attribute methods defined,
or it will override them with standard Ruby accessors. Migration specs
that rolled back the state of the database after columns were migrated
to encrypted values were interfering with these definitions. To ensure
that the SystemHook specs pass, we need to call
`SystemHook.define_attribute_methods` to ensure that attr_encrypted sees
the right methods that reflect the latest state of the database.

Closes https://gitlab.com/gitlab-org/gitlab-ee/issues/8234
parent f79ece0a
...@@ -47,10 +47,17 @@ module MigrationsHelpers ...@@ -47,10 +47,17 @@ module MigrationsHelpers
# migrated the schema up, otherwise, column information could be # migrated the schema up, otherwise, column information could be
# outdated. We have a separate method for this so we can override it in EE. # outdated. We have a separate method for this so we can override it in EE.
active_record_base.descendants.each(&method(:reset_column_information)) active_record_base.descendants.each(&method(:reset_column_information))
end
# Without that, we get errors because of missing attributes, e.g. def refresh_attribute_methods
# Without this, we get errors because of missing attributes, e.g.
# super: no superclass method `elasticsearch_indexing' for #<ApplicationSetting:0x00007f85628508d8> # super: no superclass method `elasticsearch_indexing' for #<ApplicationSetting:0x00007f85628508d8>
ApplicationSetting.define_attribute_methods # attr_encrypted also expects ActiveRecord attribute methods to be
# defined, or it will override the accessors:
# https://gitlab.com/gitlab-org/gitlab-ee/issues/8234#note_113976421
[ApplicationSetting, SystemHook].each do |model|
model.define_attribute_methods
end
end end
def reset_column_information(klass) def reset_column_information(klass)
...@@ -90,6 +97,7 @@ module MigrationsHelpers ...@@ -90,6 +97,7 @@ module MigrationsHelpers
end end
reset_column_in_all_models reset_column_in_all_models
refresh_attribute_methods
end end
def disable_migrations_output def disable_migrations_output
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment