Add missing extends to PrivateAPIInstance

thatch-health · Jul 15, 2024 · d3eca47 · d3eca47
1 parent 0a29bb9
commit d3eca47
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/lib/tapioca/dsl/compilers/grape_endpoints.rb b/lib/tapioca/dsl/compilers/grape_endpoints.rb
@@ -140,6 +140,8 @@ def create_api_class
           superclass = "::Grape::API::Instance"
 
           api.create_class(APIInstanceClassName, superclass_name: superclass) do |klass|
+            klass.create_extend(CallbacksMethodsModuleName)
+            klass.create_extend(RequestResponseMethodsModuleName)
             klass.create_extend(RoutingMethodsModuleName)
           end
         end

diff --git a/spec/tapioca/dsl/compilers/grape_endpoints_spec.rb b/spec/tapioca/dsl/compilers/grape_endpoints_spec.rb
@@ -117,6 +117,8 @@ def route_param(param, options = {}, &block); end
                   end
 
                   class PrivateAPIInstance < ::Grape::API::Instance
+                    extend GeneratedCallbacksMethods
+                    extend GeneratedRequestResponseMethods
                     extend GeneratedRoutingMethods
                   end
 
@@ -210,6 +212,8 @@ def route_param(param, options = {}, &block); end
                   end
 
                   class PrivateAPIInstance < ::Grape::API::Instance
+                    extend GeneratedCallbacksMethods
+                    extend GeneratedRequestResponseMethods
                     extend GeneratedRoutingMethods
                   end