#2893 Move builders from dspace-server-webapp to dspace-api.

dspace-api/src/test/java/org/dspace/builder/ProcessBuilder.java

@@ -0,0 +1,97 @@
+/**
+ * The contents of this file are subject to the license and copyright
+ * detailed in the LICENSE and NOTICE files at the root of the source
+ * tree and available online at
+ *
+ * http://www.dspace.org/license/
+ */
+package org.dspace.builder;
+
+import java.io.IOException;
+import java.sql.SQLException;
+import java.util.List;
+
+import org.dspace.authorize.AuthorizeException;
+import org.dspace.content.ProcessStatus;
+import org.dspace.core.Context;
+import org.dspace.eperson.EPerson;
+import org.dspace.scripts.DSpaceCommandLineParameter;
+import org.dspace.scripts.Process;
+import org.dspace.scripts.service.ProcessService;
+
+public class ProcessBuilder extends AbstractBuilder<Process, ProcessService> {
+
+    private Process process;
+
+    protected ProcessBuilder(Context context) {
+        super(context);
+    }
+
+    public static ProcessBuilder createProcess(Context context, EPerson ePerson, String scriptName,
+                                               List<DSpaceCommandLineParameter> parameters)
+        throws SQLException {
+        ProcessBuilder processBuilder = new ProcessBuilder(context);
+        return processBuilder.create(context, ePerson, scriptName, parameters);
+    }
+
+    private ProcessBuilder create(Context context, EPerson ePerson, String scriptName,
+                                  List<DSpaceCommandLineParameter> parameters)
+        throws SQLException {
+        this.context = context;
+        this.process = processService.create(context, ePerson, scriptName, parameters);
+        this.process.setProcessStatus(ProcessStatus.SCHEDULED);
+        return this;
+    }
+
+    @Override
+    public void cleanup() throws Exception {
+        try (Context c = new Context()) {
+            c.turnOffAuthorisationSystem();
+            // Ensure object and any related objects are reloaded before checking to see what needs cleanup
+            process = c.reloadEntity(process);
+            if (process != null) {
+                delete(c, process);
+            }
+            c.complete();
+            indexingService.commit();
+        }
+    }
+
+    public Process build() {
+        try {
+            processService.update(context, process);
+            context.dispatchEvents();
+            indexingService.commit();
+        } catch (Exception e) {
+            return null;
+        }
+        return process;
+    }
+
+    protected ProcessService getService() {
+        return processService;
+    }
+
+    @Override
+    public void delete(Context c, Process dso) throws Exception {
+        if (dso != null) {
+            getService().delete(c, dso);
+        }
+    }
+
+    public static void deleteProcess(Integer integer) throws SQLException, IOException {
+        try (Context c = new Context()) {
+            c.turnOffAuthorisationSystem();
+            Process process = processService.find(c, integer);
+            if (process != null) {
+                try {
+                    processService.delete(c, process);
+                } catch (AuthorizeException e) {
+                    // cannot occur, just wrap it to make the compiler happy
+                    throw new RuntimeException(e);
+                }
+            }
+            c.complete();
+        }
+    }
+}