TestUser connection

Author: labkey-tchad

src/org/labkey/test/util/TestUser.java

@@ -4,6 +4,7 @@
 import org.labkey.remoteapi.Connection;
 import org.labkey.remoteapi.security.CreateUserResponse;
 import org.labkey.test.WebDriverWrapper;
+import org.labkey.test.WebTestHelper;
 
 import java.io.IOException;
 
@@ -102,6 +103,15 @@ public String getPassword()
         return _password;
     }
 
+    /**
+     * Create a non-impersonating API connection for the user.
+     * @return new Connection
+     */
+    public Connection getUserConnection()
+    {
+        return new Connection(WebTestHelper.getBaseURL(), getEmail(), getPassword());
+    }
+
     public TestUser addPermission(String role, String containerPath)
     {
         new ApiPermissionsHelper(getWrapper()).addMemberToRole(getEmail(), role, PermissionsHelper.MemberType.user, containerPath);

src/org/labkey/test/util/exp/SampleTypeAPIHelper.java

@@ -20,7 +20,9 @@
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.NoSuchElementException;
 import java.util.TreeMap;
+import java.util.stream.Collectors;
 
 public class SampleTypeAPIHelper
 {
@@ -177,4 +179,30 @@ public static Map<String, Integer> getRowIdsForSamples(String containerPath, Str
         return rowIds;
     }
 
+    /**
+     * Get sample state IDs defined in the specified folder. Useful for updating sample status via API
+     * @param containerPath Path to the folder where the sample statuses are defined
+     */
+    public static Map<String, Integer> getSampleStateIds(String containerPath) throws IOException, CommandException
+    {
+        Connection cn = WebTestHelper.getRemoteApiConnection();
+        SelectRowsCommand insertCmd = new SelectRowsCommand("core", "DataStates");
+        return insertCmd.execute(cn, containerPath).getRows().stream().collect(Collectors.toMap(row ->
+            (String) row.get("label"), row -> (Integer) row.get("rowId")));
+    }
+
+    /**
+     * Get sample state ID defined in the specified folder. Useful for updating sample status via API
+     * @param name Label of the sample state
+     * @param containerPath Path to the folder where the sample statuses are defined
+     */
+    public static Integer getSampleStateId(String name, String containerPath) throws IOException, CommandException
+    {
+        Map<String, Integer> sampleStateIds = getSampleStateIds(containerPath);
+        if(sampleStateIds.containsKey(name))
+            return sampleStateIds.get(name);
+        else
+            throw new NoSuchElementException("Sample state '%s' not defined in '%s': %s".formatted(name, containerPath, sampleStateIds.keySet()));
+    }
+
 }

src/org/labkey/test/util/query/QueryApiHelper.java

@@ -105,7 +105,7 @@ public <T> RowsResponse updateRows(List<Map<String, T>> rows) throws IOException
         return  updateRowsCommand.execute(_connection, _containerPath);
     }
 
-    public <T> RowsResponse moveRows(String targetContainerPath, List<Map<String, T>> rows) throws IOException, CommandException
+    public <T> RowsResponse moveRows(List<Map<String, T>> rows, String targetContainerPath) throws IOException, CommandException
     {
         MoveRowsCommand moveRowsCommand = new MoveRowsCommand(targetContainerPath, _schema, _query);
         moveRowsCommand.setRows(makeApiRows(rows));