adds tests for queryData initiatedBy

dimagi · Jan 22, 2024 · 8b600f5 · 8b600f5
1 parent 7bee616
commit 8b600f5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/src/main/java/org/commcare/formplayer/objects/QueryData.java b/src/main/java/org/commcare/formplayer/objects/QueryData.java
@@ -52,6 +52,11 @@ public String getInitiatedBy(String key) {
         return null;
     }
 
+    public void setInitiatedBy(String key, String value) {
+        this.initKey(key);
+        ((Map<String, Object>) this.get(key)).put(INITIATED_BY, value);
+    }
+
     public Hashtable<String, String> getInputs(String key) {
         Map<String, Object> value = (Map<String, Object>) this.get(key);
         if (value != null) {

diff --git a/src/test/java/org/commcare/formplayer/tests/CaseClaimTests.java b/src/test/java/org/commcare/formplayer/tests/CaseClaimTests.java
@@ -490,6 +490,13 @@ public void testQueryPromptGrouped() throws Exception {
         assertEquals(queryResponseBean.getDisplays()[4].getGroupKey(), groupKey2);
     }
 
+    @Test
+    public void testQueryInitiatedBy() throws Exception {
+        QueryData queryData = new QueryData();
+        queryData.setInitiatedBy("search_command.m1_results", "field_change");
+        assertEquals(queryData.getInitiatedBy("search_command.m1_results"), "field_change");
+    }
+
     @Test
     public void testDependentItemsets_DependentChoicesChangeWithSelection() throws Exception {
         Hashtable<String, String> inputs = new Hashtable<>();