Json-ify procs in #to_return_json

bblimke · May 19, 2023 · cc4644f · cc4644f
1 parent d1e55fa
commit cc4644f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/lib/webmock/request_stub.rb b/lib/webmock/request_stub.rb
@@ -29,9 +29,13 @@ def to_return_json(*response_hashes)
 
       json_response_hashes = [*response_hashes].flatten.map do |resp_h|
         headers, body = resp_h.values_at(:headers, :body)
+
+        body = body.call if body.respond_to?(:call)
+        body = body.to_json unless body.is_a?(String)
+
         resp_h.merge(
           headers: {content_type: 'application/json'}.merge(headers.to_h),
-          body: body.is_a?(String) ? body : body.to_json
+          body: body
         )
       end
 

diff --git a/spec/unit/request_stub_spec.rb b/spec/unit/request_stub_spec.rb
@@ -87,6 +87,14 @@
       expect(@request_stub.response.body).to eq('{"abc":"def"}')
     end
 
+    it "should json-ify any callable proc or lambda to body" do
+      record = double("SomeRecord")
+      allow(record).to receive_messages(to_json: '{"what":"something callable"}.')
+
+      @request_stub.to_return_json(body: -> { record })
+      expect(@request_stub.response.body).to eq('{"what":"something callable"}.')
+    end
+
     it "should apply the content_type header" do
       @request_stub.to_return_json(body: {abc: "def"}, status: 500)
       expect(@request_stub.response.headers).to eq({"Content-Type"=>"application/json"})