Add ActionView::TestCase DSL generator

Author: vinistock

lib/tapioca/compilers/dsl/action_view_test_helpers.rb

@@ -0,0 +1,83 @@
+# typed: strict
+# frozen_string_literal: true
+
+begin
+  require "action_view/test_case"
+rescue LoadError
+  return
+end
+
+module Tapioca
+  module Compilers
+    module Dsl
+      # `Tapioca::Compilers::Dsl::ActionViewTestHelpers` decorates RBI files for all
+      # subclasses of `ActionView::TestCase` with the helpers that are included dynamically.
+      #
+      # For example, considering the `UsersHelper` module:
+      #
+      # ~~~rb
+      # module UsersHelper
+      #   def current_user_name
+      #     # ...
+      #   end
+      # end
+      # ~~~
+      #
+      # and its respective test:
+      #
+      # ~~~rb
+      # class UsersHelperTest < ActionView::TestCase
+      #   test "current_user_name works" do
+      #     assert_equal("John", current_user_name)
+      #   end
+      # end
+      # ~~~
+      #
+      # this generator will produce an RBI file `users_helper_test.rbi` with the following content:
+      #
+      # ~~~rbi
+      # # users_helper_test.rbi
+      # # typed: strong
+      # class UsersHelperTest
+      #   include HelperMethods
+      #
+      #   module HelperMethods
+      #     include ::UsersHelper
+      #   end
+      # end
+      # ~~~
+      class ActionViewTestHelpers < Base
+        extend T::Sig
+
+        sig do
+          override
+            .params(root: RBI::Tree, constant: T.class_of(ActionView::TestCase))
+            .void
+        end
+        def decorate(root, constant)
+          # The helpers are dynamically included in the class when the test is instantiated. The argument for
+          # initializing the method is just the test name. All of the helper modules are included in the module inside
+          # _helpers, so we can filter the ancestors based on their names and find all helpers
+          fake_test = constant.new("fake")
+          helpers_included = fake_test._helpers.ancestors.select { |mod| mod.name&.end_with?("Helper") }
+          helper_methods_name = "HelperMethods"
+
+          root.create_path(constant) do |klass|
+            klass.create_include(helper_methods_name)
+
+            klass.create_module(helper_methods_name) do |helper_methods|
+              helpers_included.each do |helper|
+                helper_methods.create_include(T.must(qualified_name_of(helper)))
+              end
+            end
+          end
+        end
+
+        sig { override.returns(T::Enumerable[Module]) }
+        def gather_constants
+          descendants_of(ActionView::TestCase)
+        end
+      end
+    end
+  end
+end

manual/generator_actionviewtesthelpers.md

@@ -0,0 +1,38 @@
+## ActionViewTestHelpers
+
+`Tapioca::Compilers::Dsl::ActionViewTestHelpers` decorates RBI files for all
+subclasses of `ActionView::TestCase` with the helpers that are included dynamically.
+
+For example, considering the `UsersHelper` module:
+
+~~~rb
+module UsersHelper
+  def current_user_name
+    # ...
+  end
+end
+~~~
+
+and its respective test:
+
+~~~rb
+class UsersHelperTest < ActionView::TestCase
+  test "current_user_name works" do
+    assert_equal("John", current_user_name)
+  end
+end
+~~~
+
+this generator will produce an RBI file `users_helper_test.rbi` with the following content:
+
+~~~rbi
+# users_helper_test.rbi
+# typed: strong
+class UsersHelperTest
+  include HelperMethods
+
+  module HelperMethods
+    include ::UsersHelper
+  end
+end
+~~~

manual/generators.md

@@ -6,6 +6,7 @@ In the following section you will find all available DSL generators:
 * [AASM](generator_aasm.md)
 * [ActionControllerHelpers](generator_actioncontrollerhelpers.md)
 * [ActionMailer](generator_actionmailer.md)
+* [ActionViewTestHelpers](generator_actionviewtesthelpers.md)
 * [ActiveJob](generator_activejob.md)
 * [ActiveModelAttributes](generator_activemodelattributes.md)
 * [ActiveModelSecurePassword](generator_activemodelsecurepassword.md)

spec/tapioca/compilers/dsl/action_view_test_helpers_spec.rb

@@ -0,0 +1,60 @@
+# typed: strict
+# frozen_string_literal: true
+
+require "spec_helper"
+
+class Tapioca::Compilers::Dsl::ActionViewTestHelpersSpec < DslSpec
+  describe("#initialize") do
+    after(:each) do
+      T.unsafe(self).assert_no_generated_errors
+    end
+
+    it("doesn't gather anything if there are no ActionView::TestCase tests") do
+      assert_empty(gathered_constants)
+    end
+
+    it("gathers ActionView::TestCase tests") do
+      add_ruby_file("users_helper_test.rb", <<~RUBY)
+        class UsersHelperTest < ActionView::TestCase
+        end
+      RUBY
+
+      assert_equal(["UsersHelperTest"], gathered_constants)
+    end
+  end
+
+  describe("#decorate") do
+    after(:each) do
+      T.unsafe(self).assert_no_generated_errors
+    end
+
+    it("generates a module with all dynamic helper inclusions") do
+      add_ruby_file("users_helper.rb", <<~RUBY)
+        module UsersHelper
+          def current_user_name
+            "John"
+          end
+        end
+      RUBY
+
+      add_ruby_file("users_helper_test.rb", <<~RUBY)
+        class UsersHelperTest < ActionView::TestCase
+        end
+      RUBY
+
+      expected = <<~RBI
+        # typed: strong
+
+        class UsersHelperTest
+          include HelperMethods
+
+          module HelperMethods
+            include ::UsersHelper
+          end
+        end
+      RBI
+
+      assert_equal(expected, rbi_for(:UsersHelperTest))
+    end
+  end
+end