Fix built-in parsers and formatters

ref 93e6998
ruby-grape · Mar 17, 2016 · 2534984 · 2534984
1 parent 21b58b1
commit 2534984
Show file tree

Hide file tree

Showing 5 changed files with 54 additions and 41 deletions.
diff --git a/lib/grape.rb b/lib/grape.rb
@@ -77,28 +77,6 @@ module Exceptions
     autoload :MethodNotAllowed
   end
 
-  module ErrorFormatter
-    extend ActiveSupport::Autoload
-    autoload :Base
-    autoload :Json
-    autoload :Txt
-    autoload :Xml
-  end
-
-  module Formatter
-    extend ActiveSupport::Autoload
-    autoload :Json
-    autoload :SerializableHash
-    autoload :Txt
-    autoload :Xml
-  end
-
-  module Parser
-    extend ActiveSupport::Autoload
-    autoload :Json
-    autoload :Xml
-  end
-
   module Middleware
     extend ActiveSupport::Autoload
     autoload :Base

diff --git a/lib/grape/error_formatter.rb b/lib/grape/error_formatter.rb
@@ -1,14 +1,21 @@
+require 'grape/error_formatter/base'
+require 'grape/error_formatter/json'
+require 'grape/error_formatter/txt'
+require 'grape/error_formatter/xml'
+
 module Grape
   module ErrorFormatter
     class << self
+      FORMATTERS = {
+        serializable_hash: Grape::ErrorFormatter::Json,
+        json: Grape::ErrorFormatter::Json,
+        jsonapi: Grape::ErrorFormatter::Json,
+        txt: Grape::ErrorFormatter::Txt,
+        xml: Grape::ErrorFormatter::Xml
+      }
+
       def builtin_formatters
-        {
-          serializable_hash: Grape::ErrorFormatter::Json,
-          json: Grape::ErrorFormatter::Json,
-          jsonapi: Grape::ErrorFormatter::Json,
-          txt: Grape::ErrorFormatter::Txt,
-          xml: Grape::ErrorFormatter::Xml
-        }
+        FORMATTERS
       end
 
       def formatters(options)

diff --git a/lib/grape/formatter.rb b/lib/grape/formatter.rb
@@ -1,14 +1,21 @@
+require 'grape/formatter/json'
+require 'grape/formatter/serializable_hash'
+require 'grape/formatter/txt'
+require 'grape/formatter/xml'
+
 module Grape
   module Formatter
     class << self
+      FORMATTERS = {
+        json: Grape::Formatter::Json,
+        jsonapi: Grape::Formatter::Json,
+        serializable_hash: Grape::Formatter::SerializableHash,
+        txt: Grape::Formatter::Txt,
+        xml: Grape::Formatter::Xml
+      }
+
       def builtin_formmaters
-        {
-          json: Grape::Formatter::Json,
-          jsonapi: Grape::Formatter::Json,
-          serializable_hash: Grape::Formatter::SerializableHash,
-          txt: Grape::Formatter::Txt,
-          xml: Grape::Formatter::Xml
-        }
+        FORMATTERS
       end
 
       def formatters(options)

diff --git a/lib/grape/parser.rb b/lib/grape/parser.rb
@@ -1,12 +1,17 @@
+require 'grape/parser/json'
+require 'grape/parser/xml'
+
 module Grape
   module Parser
     class << self
+      PARSERS = {
+        json: Grape::Parser::Json,
+        jsonapi: Grape::Parser::Json,
+        xml: Grape::Parser::Xml
+      }
+
       def builtin_parsers
-        {
-          json: Grape::Parser::Json,
-          jsonapi: Grape::Parser::Json,
-          xml: Grape::Parser::Xml
-        }
+        PARSERS
       end
 
       def parsers(options)

diff --git a/spec/grape/middleware/formatter_spec.rb b/spec/grape/middleware/formatter_spec.rb
@@ -292,4 +292,20 @@ def to_xml
       expect(subject.call(env)).to be_a(Grape::ServeFile::SendfileResponse)
     end
   end
+
+  context 'inheritable formatters' do
+    let(:app) { ->(_env) { [200, {}, ['']] } }
+    before do
+      class << Grape::Formatter
+        FORMATTERS[:invalid] = ->(_, _) { { message: 'invalid' }.to_json }
+      end
+      Grape::ContentTypes::CONTENT_TYPES[:invalid] = 'application/x-invalid'
+    end
+
+    it 'returns response by invalid formatter' do
+      env = { 'PATH_INFO' => '/hello.invalid', 'HTTP_ACCEPT' => 'application/x-invalid' }
+      _, _, bodies = *subject.call(env)
+      expect(bodies.body.first).to eq({ message: 'invalid' }.to_json)
+    end
+  end
 end