Use array items type from validator

Apipie · Dec 19, 2023 · acd8fc4 · acd8fc4
1 parent 05e3e0e
commit acd8fc4
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/lib/apipie/generator/swagger/param_description/type.rb b/lib/apipie/generator/swagger/param_description/type.rb
@@ -67,7 +67,15 @@ def for_array_type
         '$ref' => reference_name
       }
     else
-      items = { type: 'string' }
+      type = Apipie::Generator::Swagger::TypeExtractor::TYPES[items_type.downcase.to_sym]
+
+      type = if type.present?
+               type
+             else
+               Apipie::Generator::Swagger::TypeExtractor::TYPES[:string]
+             end
+
+      items = { type: type.to_s }
     end
 
     enum = @param_description.options[:in]

diff --git a/spec/lib/apipie/generator/swagger/param_description/type_spec.rb b/spec/lib/apipie/generator/swagger/param_description/type_spec.rb
@@ -115,10 +115,15 @@
           it 'returns the reference' do
             expect(subject).to eq({ '$ref' => reference })
           end
-
         end
       end
 
+      context 'of a Swagger type' do
+        let(:validator_options) { { of: Integer } }
+
+        it { is_expected.to eq({ type: 'integer' }) }
+      end
+
       describe 'enum' do
         subject { items[:enum] }