Added has_seer_info parameter to NDC search

Author: ctmay4

src/main/java/com/imsweb/seerapi/client/ndc/NdcSearch.java

@@ -12,6 +12,7 @@ public class NdcSearch {
 
     private String _query;
     private Category _category;
+    private Boolean _hasSeerInfo;
     private Boolean _includeRemoved;
     private String _addedSince;
     private String _modifiedSince;
@@ -36,6 +37,14 @@ public void setCategory(Category category) {
         _category = category;
     }
 
+    public Boolean getHasSeerInfo() {
+        return _hasSeerInfo;
+    }
+
+    public void setHasSeerInfo(Boolean hasSeerInfo) {
+        _hasSeerInfo = hasSeerInfo;
+    }
+
     public Boolean getIncludeRemoved() {
         return _includeRemoved;
     }
@@ -103,6 +112,8 @@ public Map<String, String> paramMap() {
             params.put("q", getQuery());
         if (getCategory() != null)
             params.put("category", getCategory().toString());
+        if (getHasSeerInfo() != null)
+            params.put("has_seer_info", getHasSeerInfo().toString());
         if (getIncludeRemoved() != null)
             params.put("include_removed", getIncludeRemoved() ? "true" : "false");
         if (getAddedSince() != null)

src/test/java/com/imsweb/seerapi/client/ndc/NdcTest.java

@@ -116,4 +116,29 @@ public void testNdcSearchByCategory() throws IOException {
         assertThat(product.getSeerInfo().getCategories()).contains(Category.CHEMOTHERAPY);
     }
 
+    @Test
+    public void testNdcHasSeerInfo() throws IOException {
+        NdcSearch search = new NdcSearch();
+        search.setIncludeRemoved(true);
+
+        // get count of all drugs
+        Response<List<NdcProduct>> response = _NDC.search(search.paramMap()).execute();
+        Integer totalCount = Integer.parseInt(response.headers().get("X-Total-Count"));
+        assertThat(totalCount).isGreaterThan(0);
+
+        // get count of seer-info drugs
+        search.setHasSeerInfo(true);
+        response = _NDC.search(search.paramMap()).execute();
+        Integer withCount = Integer.parseInt(response.headers().get("X-Total-Count"));
+        assertThat(withCount).isGreaterThan(0).isLessThan(totalCount);
+
+        // get count of non seer-info drugs
+        search.setHasSeerInfo(false);
+        response = _NDC.search(search.paramMap()).execute();
+        Integer withoutCount = Integer.parseInt(response.headers().get("X-Total-Count"));
+        assertThat(withCount).isGreaterThan(0).isLessThan(totalCount);
+
+        assertThat(totalCount).isEqualTo(withCount + withoutCount);
+    }
+
 }