1 files changed, 22 insertions, 4 deletions
diff --git a/eidas_modules/authmodule-eIDAS-v2/src/main/java/at/asitplus/eidas/specific/modules/auth/eidas/v2/dao/CountrySpecificDetailSearchResult.java b/eidas_modules/authmodule-eIDAS-v2/src/main/java/at/asitplus/eidas/specific/modules/auth/eidas/v2/dao/CountrySpecificDetailSearchResult.java
index 710e286c..b74172f9 100644
--- a/eidas_modules/authmodule-eIDAS-v2/src/main/java/at/asitplus/eidas/specific/modules/auth/eidas/v2/dao/CountrySpecificDetailSearchResult.java
+++ b/eidas_modules/authmodule-eIDAS-v2/src/main/java/at/asitplus/eidas/specific/modules/auth/eidas/v2/dao/CountrySpecificDetailSearchResult.java
@@ -1,14 +1,32 @@
 package at.asitplus.eidas.specific.modules.auth.eidas.v2.dao;
 
+import at.asitplus.eidas.specific.modules.auth.eidas.v2.exception.WorkflowException;
+
+import java.util.ArrayList;
+
 public class CountrySpecificDetailSearchResult {
 
   //TODO is the result the same as the one form the initial search?
-  int resultsZmr;
-  int resultsErnb;
+  ArrayList<RegisterResult> resultsZmr = new ArrayList<>();
+  ArrayList<RegisterResult> resultsErnb = new ArrayList<>();
 
   public int getResultCount() {
-    return resultsZmr + resultsErnb;
+    return resultsZmr.size() + resultsErnb.size();
   }
 
-
+  /**
+   * Verfies that there is only one match and retunrs the bpk.
+   * @return bpk bpk of the match
+   * @throws WorkflowException if multiple results have been found
+   */
+  public String getBpk() throws WorkflowException {
+    if (getResultCount() != 1) {
+      throw new WorkflowException("getResultCount() != 1");
+    }
+    if (resultsZmr.size() == 1) {
+      return resultsZmr.get(0).getBpk();
+    } else {
+      return resultsErnb.get(0).getBpk();
+    }
+  }
 }