From 1a7ee1dc238bd76b25fecc6850a07ce796ec8fed Mon Sep 17 00:00:00 2001
From: "bhawna.arora" <bhawna.arora@docusign.com>
Date: Mon, 8 Jul 2024 18:54:12 +0530
Subject: [PATCH] added cleanup method for solr based on custom query

---
 .../injection/plugin/junit/solr/SolrAssertions.java         | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/plugin/inject-solr-plugin/src/main/java/org/hrodberaht/injection/plugin/junit/solr/SolrAssertions.java b/plugin/inject-solr-plugin/src/main/java/org/hrodberaht/injection/plugin/junit/solr/SolrAssertions.java
index c0da9f1..425324e 100644
--- a/plugin/inject-solr-plugin/src/main/java/org/hrodberaht/injection/plugin/junit/solr/SolrAssertions.java
+++ b/plugin/inject-solr-plugin/src/main/java/org/hrodberaht/injection/plugin/junit/solr/SolrAssertions.java
@@ -74,6 +74,12 @@ public void cleanDataFromCollection() throws IOException, SolrServerException {
     }
 
 
+    public void cleanDataFromCollection(String query) throws IOException, SolrServerException {
+        processDelete(query);
+        processCommit();
+    }
+
+
     public void waitForAsyncCommit() {
         // TODO: Async needs a callback solution
         try {