test: adds test verifying smartlinking that requires claiming a case

dimagi · Oct 3, 2024 · 9a8c91f · 9a8c91f
1 parent 3e129e7
commit 9a8c91f
Show file tree

Hide file tree

Showing 3 changed files with 55 additions and 0 deletions.
diff --git a/src/test/java/org/commcare/formplayer/tests/EndpointLaunchTest.java b/src/test/java/org/commcare/formplayer/tests/EndpointLaunchTest.java
@@ -16,6 +16,7 @@
 import org.commcare.formplayer.beans.menus.CommandUtils.NavIconState;
 import org.commcare.formplayer.mocks.FormPlayerPropertyManagerMock;
 import org.commcare.formplayer.utils.FileUtils;
+import org.commcare.formplayer.utils.MockRequestUtils;
 import org.commcare.formplayer.utils.WithHqUser;
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.BeforeEach;
@@ -34,13 +35,16 @@ public class EndpointLaunchTest extends BaseTestClass {
 
     private final String APP_NAME = "endpoint";
 
+    private MockRequestUtils mockRequest;
+
     @Override
     @BeforeEach
     public void setUp() throws Exception {
         super.setUp();
         configureRestoreFactory("endpointdomain", "endpointusername");
         storageFactoryMock.configure("endpointusername", "endpointdomain", "app_id", "asUser");
         FormPlayerPropertyManagerMock.mockAutoAdvanceMenu(storageFactoryMock);
+        mockRequest = new MockRequestUtils(webClientMock, restoreFactoryMock);
     }
 
     @Override
@@ -275,6 +279,20 @@ public void testPersistentMenuForEndpointLaunchWithoutRespectRelevancy() throws
         assertEquals(expectedMenu, formResponse.getPersistentMenu());
     }
 
+    @Test
+    @WithHqUser(enabledToggles = {TOGGLE_SESSION_ENDPOINTS})
+    public void testEndpointsWithSync() throws Exception {
+        mockRequest.mockPostandUpdateRestore("restores/caseclaim3.xml");
+        HashMap<String, String> endpointArgs = new HashMap<>();
+        endpointArgs.put("case_id", "0156fa3e-093e-4136-b95c-01b13dae66c6"); // from case claim response
+
+        NewFormResponse formResponse = sessionNavigateWithEndpoint(APP_NAME,
+        "add_child",
+        endpointArgs,
+        NewFormResponse.class);
+        assert formResponse.getTitle().contentEquals("Add Child");
+    }
+
     private void configureQueryMock() {
         when(webClientMock.postFormData(anyString(), any(Multimap.class)))
                 .thenReturn(FileUtils.getFile(this.getClass(),

diff --git a/src/test/java/org/commcare/formplayer/utils/MockRequestUtils.java b/src/test/java/org/commcare/formplayer/utils/MockRequestUtils.java
@@ -12,6 +12,7 @@
 import org.commcare.formplayer.services.RestoreFactory;
 import org.commcare.formplayer.web.client.WebClient;
 import org.mockito.Mockito;
+import org.mockito.stubbing.Answer;
 import org.mockito.verification.VerificationMode;
 
 /**
@@ -43,6 +44,22 @@ public VerifiedMock mockPost(boolean returnValue, int times) {
         };
     }
 
+    /**
+     * Mock the post request, verifies it happened, and updates the mocked restore.
+     */
+    public VerifiedMock mockPostandUpdateRestore(String restoreFile) {
+        Mockito.reset(webClientMock);
+        Answer<Boolean> answer = invocation -> {
+            mockRestore(restoreFile);
+            return true;
+        };
+        Mockito.doAnswer(answer).when(webClientMock).caseClaimPost(anyString(), any());
+
+        return () -> {
+            verify(webClientMock, Mockito.times(1)).caseClaimPost(anyString(), any());
+        };
+    }
+
     /**
      * Mock the restore and verify it happened
      */

diff --git a/src/test/resources/archives/endpoint/suite.xml b/src/test/resources/archives/endpoint/suite.xml
@@ -671,6 +671,22 @@
       </push>
     </stack>
   </remote-request>
+  <remote-request>
+    <post url="http://localhost:8000/a/test-1/phone/claim-case/" relevant="count(instance('casedb')/casedb/case[@case_id=instance('commcaresession')/session/data/case_id]) = 0">
+      <data key="case_id" ref="instance('commcaresession')/session/data/case_id"/>
+    </post>
+    <command id="claim_command.add_child.case_id">
+      <display>
+        <text/>
+      </display>
+    </command>
+    <instance id="casedb" src="jr://instance/casedb"/>
+    <instance id="commcaresession" src="jr://instance/session"/>
+    <session>
+      <datum id="case_id" function="instance('commcaresession')/session/data/case_id"/>
+    </session>
+    <stack/>
+  </remote-request>
   <endpoint id="caselist">
     <stack>
       <push>
@@ -708,6 +724,10 @@
   <endpoint id="add_child">
     <argument id="case_id"/>
     <stack>
+      <push>
+        <datum id="case_id" value="$case_id"/>
+        <command value="'claim_command.add_child.case_id'"/>
+      </push>
       <push>
         <command value="'m1'"/>
         <datum id="case_id" value="$case_id"/>