update CapitalizeDecorator to handle scenario where the @nameable obj…

…ect is not a Person
kessie2862 · Aug 1, 2023 · 9446402 · 9446402
1 parent 55c53ac
commit 9446402
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/capitalize_decorator.rb b/capitalize_decorator.rb
@@ -2,6 +2,7 @@
 
 class CapitalizeDecorator < Decorator
   def correct_name
-    super.capitalize
+    nameable_name = @nameable.respond_to?(:correct_name) ? @nameable.correct_name : @nameable
+    nameable_name.to_s.capitalize
   end
 end