fix: Remove content field if no content (#193)

exoego · web-flow · commit efcca3be27b9 · 2024-03-25T16:44:06.000+09:00
diff --git a/lib/rspec/openapi/schema_builder.rb b/lib/rspec/openapi/schema_builder.rb
@@ -14,12 +14,15 @@ def build(record)
     if record.response_body
       disposition = normalize_content_disposition(record.response_content_disposition)
 
-      response[:content] = {
-        normalize_content_type(record.response_content_type) => {
-          schema: build_property(record.response_body, disposition: disposition),
-          example: response_example(record, disposition: disposition),
-        }.compact,
-      }
+      has_content = !normalize_content_type(record.response_content_type).nil?
+      if has_content
+        response[:content] = {
+          normalize_content_type(record.response_content_type) => {
+            schema: build_property(record.response_body, disposition: disposition),
+            example: response_example(record, disposition: disposition),
+          }.compact,
+        }
+      end
     end
 
     {
diff --git a/spec/rails/doc/smart/expected.yaml b/spec/rails/doc/smart/expected.yaml
@@ -19,6 +19,13 @@ paths:
       summary: get
       tags:
       - Page
+      parameters:
+        - name: head
+          in: query
+          required: false
+          schema:
+            type: integer
+          example: 1
       responses:
         '200':
           description: return HTML
@@ -27,8 +34,8 @@ paths:
               schema:
                 type: string
               example: '<!DOCTYPE html><html lang="en"><head><title>Hello</title></head><body>Hello</body></html>'
-#        '204':
-#          description: return no content
+        '204':
+          description: return no content
   "/tables":
     get:
       summary: index
diff --git a/spec/requests/rails_smart_merge_spec.rb b/spec/requests/rails_smart_merge_spec.rb
@@ -117,5 +117,10 @@
       get '/pages'
       expect(response.status).to eq(200)
     end
+
+    it 'return no content' do
+      get '/pages?head=1'
+      expect(response.status).to eq(204)
+    end
   end
 end
