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

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

@@ -0,0 +1,169 @@
+/**
+ * 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.Date;
+
+import org.apache.log4j.Logger;
+import org.dspace.authorize.AuthorizeException;
+import org.dspace.authorize.ResourcePolicy;
+import org.dspace.authorize.service.ResourcePolicyService;
+import org.dspace.content.DSpaceObject;
+import org.dspace.core.Context;
+import org.dspace.discovery.SearchServiceException;
+import org.dspace.eperson.EPerson;
+import org.dspace.eperson.Group;
+
+public class ResourcePolicyBuilder extends AbstractBuilder<ResourcePolicy, ResourcePolicyService> {
+
+    /* Log4j logger*/
+    private static final Logger log = Logger.getLogger(ResourcePolicyBuilder.class);
+
+    private ResourcePolicy resourcePolicy;
+
+    protected ResourcePolicyBuilder(Context context) {
+        super(context);
+    }
+
+    @Override
+    protected ResourcePolicyService getService() {
+        return resourcePolicyService;
+    }
+
+    @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
+            resourcePolicy = c.reloadEntity(resourcePolicy);
+            if (resourcePolicy != null) {
+                delete(c, resourcePolicy);
+            }
+            c.complete();
+            indexingService.commit();
+        }
+    }
+
+    @Override
+    public void delete(Context c, ResourcePolicy dso) throws Exception {
+        if (dso != null) {
+            getService().delete(c, dso);
+        }
+    }
+
+    @Override
+    public ResourcePolicy build() {
+        try {
+
+            resourcePolicyService.update(context, resourcePolicy);
+            context.dispatchEvents();
+
+            indexingService.commit();
+        } catch (SearchServiceException e) {
+            log.error(e);
+        } catch (SQLException e) {
+            log.error(e);
+        } catch (AuthorizeException e) {
+            log.error(e);
+            ;
+        }
+        return resourcePolicy;
+    }
+
+    public void delete(ResourcePolicy rp) throws Exception {
+        try (Context c = new Context()) {
+            c.turnOffAuthorisationSystem();
+            ResourcePolicy attachedDso = c.reloadEntity(rp);
+            if (attachedDso != null) {
+                getService().delete(c, attachedDso);
+            }
+            c.complete();
+        }
+
+        indexingService.commit();
+    }
+
+    public static void delete(Integer id)
+            throws SQLException, IOException, SearchServiceException {
+        try (Context c = new Context()) {
+            c.turnOffAuthorisationSystem();
+            ResourcePolicy rp = resourcePolicyService.find(c, id);
+            if (rp != null) {
+                try {
+                    resourcePolicyService.delete(c, rp);
+                } catch (AuthorizeException e) {
+                    throw new RuntimeException(e.getMessage(), e);
+                }
+            }
+            c.complete();
+        }
+        indexingService.commit();
+    }
+
+    public static ResourcePolicyBuilder createResourcePolicy(Context context)
+            throws SQLException, AuthorizeException {
+        ResourcePolicyBuilder resourcePolicyBuilder = new ResourcePolicyBuilder(context);
+        return resourcePolicyBuilder.create(context);
+    }
+
+    private ResourcePolicyBuilder create(Context context)
+            throws SQLException, AuthorizeException {
+        this.context = context;
+
+        resourcePolicy = resourcePolicyService.create(context);
+
+        return this;
+    }
+
+    public ResourcePolicyBuilder withUser(EPerson ePerson) throws SQLException {
+        resourcePolicy.setEPerson(ePerson);
+        return this;
+    }
+
+    public ResourcePolicyBuilder withGroup(Group epersonGroup) throws SQLException {
+        resourcePolicy.setGroup(epersonGroup);
+        return this;
+    }
+
+    public ResourcePolicyBuilder withAction(int action) throws SQLException {
+        resourcePolicy.setAction(action);
+        return this;
+    }
+
+    public ResourcePolicyBuilder withDspaceObject(DSpaceObject dspaceObject) throws SQLException {
+        resourcePolicy.setdSpaceObject(dspaceObject);
+        return this;
+    }
+
+    public ResourcePolicyBuilder withPolicyType(String policyType) {
+        resourcePolicy.setRpType(policyType);
+        return this;
+    }
+
+    public ResourcePolicyBuilder withStartDate(Date data) throws SQLException {
+        resourcePolicy.setStartDate(data);
+        return this;
+    }
+
+    public ResourcePolicyBuilder withEndDate(Date data) throws SQLException {
+        resourcePolicy.setEndDate(data);
+        return this;
+    }
+
+    public ResourcePolicyBuilder withDescription(String description) throws SQLException {
+        resourcePolicy.setRpDescription(description);
+        return this;
+    }
+
+    public ResourcePolicyBuilder withName(String name) throws SQLException {
+        resourcePolicy.setRpName(name);
+        return this;
+    }
+}