diff --git a/test/unit/disabled_enterprise_message_block_test.rb b/test/unit/disabled_enterprise_message_block_test.rb index 16ddb4a..c6997a2 100644 --- a/test/unit/disabled_enterprise_message_block_test.rb +++ b/test/unit/disabled_enterprise_message_block_test.rb @@ -19,12 +19,17 @@ class DisabledEnterpriseMessageBlockTest < ActiveSupport::TestCase block = enterprise.boxes.first.blocks.first ApplicationHelper.class_eval do + alias_method :original_profile, :profile def profile return Enterprise['disabled-enterprise'] end end assert_match 'This message is for disabled enterprises', render_block_content(block) + + ApplicationHelper.class_eval do + alias_method :profile, :original_profile + end end should 'not be editable' do -- libgit2 0.21.2