Extend the SearchBuilder to support selecting the DataType(s)

Change-Id: I8d41a7e7cb0dd948b0ce699aaa04beb510d5dbc7

Author: msomhorst

subprojects/client/src/integrationTest/groovy/com/bol/openapi/OpenApiClientIntegrationSpec.groovy

@@ -101,6 +101,33 @@ class OpenApiClientIntegrationSpec extends Specification {
         product.offerData.offers.size() >= 1
     }
 
+    def 'Return products and categories'() {
+        def results = OpenApiClient.withDefaultClient(apiKey).searchBuilder()
+                .term('harry potter')
+                .dataType(QueryDataType.DataType.PRODUCTS)
+                .dataType(QueryDataType.DataType.CATEGORIES)
+                .search()
+
+        expect:
+        results.totalResultSize >= 1
+        results.products.size() > 0
+        results.categories.size() > 0
+        results.refinementGroups.size() == 0
+    }
+
+    def 'Return only refinements'() {
+        def results = OpenApiClient.withDefaultClient(apiKey).searchBuilder()
+                .term('harry potter')
+                .dataType(QueryDataType.DataType.REFINEMENTS)
+                .search()
+
+        expect:
+        results.totalResultSize >= 1
+        results.products.size() == 0
+        results.categories.size() == 0
+        results.refinementGroups.size() > 0
+    }
+
     def 'Can find out if the OpenAPI is healthy'() {
         def status = OpenApiClient.withDefaultClient(apiKey).getHealthStatus()
 

subprojects/client/src/main/java/com/bol/openapi/SearchBuilder.java

@@ -55,6 +55,11 @@ public SearchBuilder category(String id) {
         return new SearchBuilder(this);
     }
 
+    public SearchBuilder dataType(QueryDataType.DataType dataType) {
+        dataTypes.add(dataType);
+        return new SearchBuilder(this);
+    }
+
     public SearchBuilder allOffers() {
         offerTypes.add(ALL);
         return new SearchBuilder(this);