#90 Can now set updates-allowed easily on a database

Author: rjrudin

src/main/java/com/marklogic/mgmt/databases/DatabaseManager.java

@@ -68,6 +68,19 @@ public void deleteReplicaForests(String databaseNameOrId) {
         logger.info(format("Finished deleting replica forests for database %s", databaseNameOrId));
     }
 
+    /**
+     * TODO Not sure, when setting updates-allowed on primary forests, if replica forests need to have their
+     * updates-allowed set as well.
+     *
+     * @param databaseNameOrId
+     */
+    public void setUpdatesAllowedOnPrimaryForests(String databaseNameOrId, String mode) {
+        ForestManager mgr = new ForestManager(getManageClient());
+        for (String forestId : getPrimaryForestIds(databaseNameOrId)) {
+            mgr.setUpdatesAllowed(forestId, mode);
+        }
+    }
+
     @Override
     protected String[] getDeleteResourceParams(String payload) {
         return forestDelete != null ? new String[] { "forest-delete", forestDelete } : new String[] {};

src/main/java/com/marklogic/mgmt/forests/ForestManager.java

@@ -144,6 +144,12 @@ public ForestStatus getForestStatus(String forestIdOrName) {
         return new ForestStatus(getManageClient().getXml(path));
     }
 
+    public void setUpdatesAllowed(String forestIdOrName, String mode) {
+        String path = getPropertiesPath(forestIdOrName);
+        String json = format("{\"updates-allowed\":\"%s\"}", mode);
+        getManageClient().putJson(path, json);
+    }
+
     @Override
     protected String[] getDeleteResourceParams(String payload) {
         return this.deleteLevel != null ? new String[] { "level", deleteLevel } : null;

src/test/java/com/marklogic/appdeployer/command/databases/SetUpdatesAllowedOnDatabaseForestsTest.java

@@ -0,0 +1,37 @@
+package com.marklogic.appdeployer.command.databases;
+
+import com.marklogic.appdeployer.AbstractAppDeployerTest;
+import com.marklogic.mgmt.databases.DatabaseManager;
+import com.marklogic.mgmt.forests.ForestManager;
+import org.junit.After;
+import org.junit.Test;
+
+import java.io.File;
+
+/**
+ * Doesn't actually use a command, but it's nice to extend the parent test class.
+ */
+public class SetUpdatesAllowedOnDatabaseForestsTest extends AbstractAppDeployerTest {
+
+    @After
+    public void teardown() {
+        undeploySampleApp();
+    }
+
+    @Test
+    public void test() {
+        appConfig.getConfigDir().setBaseDir(new File("src/test/resources/sample-app/db-only-config"));
+        initializeAppDeployer(new DeployContentDatabasesCommand(2));
+        appDeployer.deploy(appConfig);
+
+        DatabaseManager dbMgr = new DatabaseManager(this.manageClient);
+        dbMgr.setUpdatesAllowedOnPrimaryForests(appConfig.getContentDatabaseName(), "flash-backup");
+
+        ForestManager forestMgr = new ForestManager(this.manageClient);
+        assertEquals("flash-backup", forestMgr.getPropertiesAsXml("sample-app-content-1").getElementValue("//m:updates-allowed"));
+        assertEquals("flash-backup", forestMgr.getPropertiesAsXml("sample-app-content-2").getElementValue("//m:updates-allowed"));
+
+        // Gotta set it back to all so the database/forests can be deleted
+        dbMgr.setUpdatesAllowedOnPrimaryForests(appConfig.getContentDatabaseName(), "all");
+    }
+}