query parameter needs import

Author: hauner

src/main/groovy/com/github/hauner/openapi/spring/model/parameters/QueryParameter.groovy

@@ -43,6 +43,10 @@ class QueryParameter {
         "@${annotationName}"
     }
 
+    String getDataTypeImport() {
+        dataType.import
+    }
+
     /**
      * Is the parameter required?
      *

src/main/groovy/com/github/hauner/openapi/spring/writer/InterfaceWriter.groovy

@@ -60,6 +60,8 @@ class InterfaceWriter {
                 if (p.withAnnotation()) {
                     imports.add (p.annotationWithPackage)
                 }
+
+                imports.add (p.dataTypeImport)
             }
 
             if (!ep.response.empty) {

src/test/groovy/com/github/hauner/openapi/spring/writer/InterfaceWriterSpec.groovy

@@ -117,7 +117,7 @@ import org.springframework.http.ResponseEntity;
 """)
     }
 
-    void "writes RequestParam import" () {
+    void "writes @RequestParam import" () {
         def apiItf = new Interface (name: 'name', endpoints: [
             new Endpoint(path: 'path', method: HttpMethod.GET, responses: [new EmptyResponse()],
                 parameters: [
@@ -135,7 +135,7 @@ import org.springframework.web.bind.annotation.RequestParam;
 """)
     }
 
-    void "does not write @RequestParam annotation import of request parameter that does not want the annotation" () {
+    void "does not write @RequestParam annotation import of parameter that does not want the annotation" () {
         def endpoint = new Endpoint (path: '/foo', method: HttpMethod.GET, responses: [
             new Response (contentType: 'application/json', responseType: new NoneDataType())
         ], parameters: [
@@ -159,6 +159,30 @@ import org.springframework.web.bind.annotation.RequestParam;
 """)
     }
 
+    void "writes import of request parameter data type" () {
+        def endpoint = new Endpoint (path: '/foo', method: HttpMethod.GET, responses: [
+            new Response (contentType: 'application/json', responseType: new NoneDataType())
+        ], parameters: [
+            new QueryParameter(name: 'foo', required: false, dataType: new ObjectDataType (
+                pkg: 'model', type: 'Foo', properties: [
+                    foo1: new StringDataType (),
+                    foo2: new StringDataType ()
+                ]
+            ))
+        ])
+
+        def apiItf = new Interface (name: 'name', endpoints: [endpoint])
+
+        when:
+        writer.write (target, apiItf)
+
+        then:
+        def result = extractImports (target.toString ())
+        result.contains("""\
+import model.Foo;
+""")
+    }
+
     void "writes model import"() {
         def pkg = 'model.package'
         def type = 'Model'