Make sure to work with subclass of subclass of Yuba::ViewModel as exp…

…ected

lib/yuba/view_model/rendering.rb

@@ -19,8 +19,6 @@ def _protected_ivars
 
       def view_model_assigns
         return {} unless defined?(@_view_model)
-        # TODO: get all public methods between self and Yuba::ViewModel
-        #       now get only in self
         methods = @_view_model.public_methods(false)
         methods.reject! do |method_name|
           %i[initialize].include?(method_name) ||

test/yuba/view_model/rendering_test.rb

@@ -23,6 +23,10 @@ def _protected_ivars
     property :name, public: true
   end
 
+  inherited_view_model_class = Class.new(simple_view_model_class) do
+    property :email, public: true
+  end
+
   view_model_class_with_predicate_method = Class.new(Yuba::ViewModel) do
     property :name, public: true
 
@@ -44,4 +48,11 @@ def enable?
     action_controller.render(view_model: view_model)
     assert_equal({ name: 'willnet' }, action_controller.view_assigns)
   end
+
+  test 'subclass of subclass of Yuba::ViewModel works' do
+    action_controller = action_controller_class.new
+    view_model = inherited_view_model_class.new(name: 'willnet', email: 'test@example.com')
+    action_controller.render(view_model: view_model)
+    assert_equal({ name: 'willnet', email: 'test@example.com' }, action_controller.view_assigns)
+  end
 end