Small changes to the visibility variable assignment

hanami · Jun 21, 2016 · 088ebd6 · 088ebd6
1 parent 1d37fa0
commit 088ebd6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/lib/hanami/view/escape.rb b/lib/hanami/view/escape.rb
@@ -165,8 +165,9 @@ def self.extended(base)
       # @since 0.4.0
       # @api private
       def method_added(method_name)
+        visibility = :public
         visibility = :private if private_method_defined? method_name
-        visibility ||= :protected if protected_method_defined? method_name
+        visibility = :protected if protected_method_defined? method_name
 
         unless autoescape_methods[method_name]
           prepend Module.new {

diff --git a/test/escape_test.rb b/test/escape_test.rb
@@ -52,5 +52,6 @@
   it "does not alter the method visibility" do
     Users::Show.private_instance_methods.must_include(:private_username)
     Users::Show.protected_instance_methods.must_include(:protected_username)
+    Users::Show.public_instance_methods.must_include(:raw_username)
   end
 end