/*
 * Licensed to jclouds, Inc. (jclouds) under one or more
 * contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  jclouds licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.jclouds.vcloud.director.v1_5.features;

import static com.google.common.base.Objects.equal;
import static org.jclouds.vcloud.director.v1_5.VCloudDirectorLiveTestConstants.OBJ_FIELD_UPDATABLE;
import static org.jclouds.vcloud.director.v1_5.VCloudDirectorLiveTestConstants.REF_REQ_LIVE;
import static org.testng.Assert.assertNotNull;
import static org.testng.Assert.assertTrue;

import org.jclouds.vcloud.director.v1_5.domain.AdminOrg;
import org.jclouds.vcloud.director.v1_5.domain.Checks;
import org.jclouds.vcloud.director.v1_5.domain.OrgEmailSettings;
import org.jclouds.vcloud.director.v1_5.domain.OrgGeneralSettings;
import org.jclouds.vcloud.director.v1_5.domain.OrgLdapSettings;
import org.jclouds.vcloud.director.v1_5.domain.OrgLeaseSettings;
import org.jclouds.vcloud.director.v1_5.domain.OrgPasswordPolicySettings;
import org.jclouds.vcloud.director.v1_5.domain.OrgSettings;
import org.jclouds.vcloud.director.v1_5.domain.OrgVAppTemplateLeaseSettings;
import org.jclouds.vcloud.director.v1_5.domain.Reference;
import org.jclouds.vcloud.director.v1_5.domain.SmtpServerSettings;
import org.jclouds.vcloud.director.v1_5.features.admin.AdminOrgClient;
import org.jclouds.vcloud.director.v1_5.internal.BaseVCloudDirectorClientLiveTest;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

import com.google.common.collect.Iterables;

/**
 * Tests live behavior of {@link AdminGroupClient}.
 * 
 * @author danikov
 */
@Test(groups = { "live", "admin", "org" }, singleThreaded = true, testName = "AdminOrgClientLiveTest")
public class AdminOrgClientLiveTest extends BaseVCloudDirectorClientLiveTest {
   
   public static final String ORG = "admin org";

   /*
    * Convenience references to API clients.
    */

   private AdminOrgClient orgClient;

   /*
    * Shared state between dependant tests.
    */
   private Reference orgRef;
   private OrgSettings settings;
   private OrgEmailSettings emailSettings;
   private OrgGeneralSettings generalSettings;
   private OrgLdapSettings ldapSettings;
   private OrgPasswordPolicySettings passwordPolicy;
   private OrgLeaseSettings vAppLeaseSettings;
   private OrgVAppTemplateLeaseSettings vAppTemplateLeaseSettings;

   @Override
   @BeforeClass(alwaysRun = true)
   public void setupRequiredClients() {
      orgClient = adminContext.getApi().getOrgClient();
      orgRef = Iterables.getFirst(orgClient.getOrgList().getOrgs(), null).toAdminReference(endpoint);
      assertNotNull(orgRef, String.format(REF_REQ_LIVE, "admin org"));
   }
   
   @Test(description = "GET /admin/org/{id}")
   public void testGetAdminOrg() {
      AdminOrg adminOrg = orgClient.getOrg(orgRef.getHref());
      
      Checks.checkAdminOrg(adminOrg);
   }
   
   @Test(description = "GET /admin/org/{id}/settings/email")
   public void testGetEmailSettings() {
      emailSettings = orgClient.getEmailSettings(orgRef.getHref());
      
      Checks.checkEmailSettings(emailSettings);
   }
   
   @Test(description = "PUT /admin/org/{id}/settings/email", 
         dependsOnMethods = { "testGetEmailSettings" })
   public void testUpdateEmailSettings() {
      boolean isDefaultSmtpServer = emailSettings.isDefaultSmtpServer();
      boolean isDefaultOrgEmail = emailSettings.isDefaultOrgEmail();
      String oldFromEmailAddress = emailSettings.getFromEmailAddress();
      String newFromEmailAddress = "test@test.com";
      String oldDefaultSubjectPrefix = emailSettings.getDefaultSubjectPrefix();
      String newDefaultSubjectPrefix = "new"+oldDefaultSubjectPrefix;
      boolean isAlertEmailToAllAdmins = emailSettings.isAlertEmailToAllAdmins();
      SmtpServerSettings oldSmtpServerSettings = emailSettings.getSmtpServerSettings();
      SmtpServerSettings newSmtpServerSettings = oldSmtpServerSettings.toBuilder()
         .useAuthentication(!oldSmtpServerSettings.useAuthentication())
         .host("new"+oldSmtpServerSettings.getHost())
         .username("new"+oldSmtpServerSettings.getUsername())
         .password("new"+oldSmtpServerSettings.getPassword())
         .build();
      
      try {
         OrgEmailSettings newEmailSettings = emailSettings.toBuilder()
               .isDefaultSmtpServer(!isDefaultSmtpServer)
               .isDefaultOrgEmail(!isDefaultOrgEmail)
               .fromEmailAddress(newFromEmailAddress)
               .defaultSubjectPrefix(newDefaultSubjectPrefix)
               .isAlertEmailToAllAdmins(!isAlertEmailToAllAdmins)
               .smtpServerSettings(newSmtpServerSettings)
               .build();
         
         emailSettings = orgClient.updateEmailSettings(
               orgRef.getHref(), newEmailSettings);
         
         assertTrue(equal(emailSettings.isDefaultSmtpServer(), !isDefaultSmtpServer), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "emailSettings", "isDefaultSmtpServer"));
         assertTrue(equal(emailSettings.isDefaultOrgEmail(), !isDefaultOrgEmail), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "emailSettings", "isDefaultOrgEmail"));
         assertTrue(equal(emailSettings.getFromEmailAddress(), newFromEmailAddress), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "emailSettings", "fromEmailAddress"));
         assertTrue(equal(emailSettings.getDefaultSubjectPrefix(), newDefaultSubjectPrefix), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "emailSettings", "defaultSubjectPrefix"));
         assertTrue(equal(emailSettings.isAlertEmailToAllAdmins(), !isAlertEmailToAllAdmins), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "emailSettings", "isAlertEmailToAllAdmins"));
         assertTrue(equal(emailSettings.getSmtpServerSettings(), newSmtpServerSettings), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "emailSettings", "smtpServerSettings"));
         
         //TODO negative tests?
         
         Checks.checkEmailSettings(emailSettings);
      } finally {
         emailSettings = emailSettings.toBuilder()
               .isDefaultSmtpServer(isDefaultSmtpServer)
               .isDefaultOrgEmail(isDefaultOrgEmail)
               .fromEmailAddress(oldFromEmailAddress)
               .defaultSubjectPrefix(oldDefaultSubjectPrefix)
               .isAlertEmailToAllAdmins(isAlertEmailToAllAdmins)
               .smtpServerSettings(oldSmtpServerSettings)
               .build();
         
         emailSettings = orgClient.updateEmailSettings(
               orgRef.getHref(), emailSettings);
      }
   }
 
   @Test(description = "GET /admin/org/{id}/settings/general")
   public void testGetGeneralSettings() {
      generalSettings = orgClient.getGeneralSettings(orgRef.getHref());
      
      Checks.checkGeneralSettings(generalSettings);
   }
   
   @Test(description = "PUT /admin/org/{id}/settings/general", 
         dependsOnMethods = { "testGetGeneralSettings" } )
   public void testUpdateGeneralSettings() {
      // FIXME: canPublishCatalogs does not update
      //boolean canPublishCatalogs = generalSettings.canPublishCatalogs();
      Integer deployedVMQuota = generalSettings.getDeployedVMQuota();
      Integer storedVmQuota = generalSettings.getStoredVmQuota();
      boolean useServerBootSequence = generalSettings.useServerBootSequence();
      Integer delayAfterPowerOnSeconds = generalSettings.getDelayAfterPowerOnSeconds();
      
      try {
         OrgGeneralSettings newGeneralSettings = generalSettings.toBuilder()
               //.canPublishCatalogs(!canPublishCatalogs)
               .deployedVMQuota(deployedVMQuota+1)
               .storedVmQuota(storedVmQuota+1)
               .useServerBootSequence(!useServerBootSequence)
               .delayAfterPowerOnSeconds(delayAfterPowerOnSeconds+1)
               .build();
         
         generalSettings = orgClient.updateGeneralSettings(
               orgRef.getHref(), newGeneralSettings);
         
//         assertTrue(equal(generalSettings.canPublishCatalogs(), !canPublishCatalogs), 
//               String.format(OBJ_FIELD_UPDATABLE, 
//               "generalSettings", "canPublishCatalogs"));
         assertTrue(equal(generalSettings.getDeployedVMQuota(), deployedVMQuota+1), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "generalSettings", "deployedVMQuota"));
         assertTrue(equal(generalSettings.getStoredVmQuota(), storedVmQuota+1), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "generalSettings", "storedVmQuota"));
         assertTrue(equal(generalSettings.useServerBootSequence(), !useServerBootSequence), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "generalSettings", "useServerBootSequence"));
         assertTrue(equal(generalSettings.getDelayAfterPowerOnSeconds(), delayAfterPowerOnSeconds+1), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "generalSettings", "delayAfterPowerOnSeconds"));
         
         //TODO negative tests?
         
         Checks.checkGeneralSettings(generalSettings);
      } finally {
         generalSettings = generalSettings.toBuilder()
//               .canPublishCatalogs(canPublishCatalogs)
               .deployedVMQuota(deployedVMQuota)
               .storedVmQuota(storedVmQuota)
               .useServerBootSequence(useServerBootSequence)
               .delayAfterPowerOnSeconds(delayAfterPowerOnSeconds)
               .build();
         
         generalSettings = orgClient.updateGeneralSettings(
               orgRef.getHref(), generalSettings);
      }
   }
 
   @Test(description = "GET /admin/org/{id}/settings/ldap")
   public void testGetLdapSettings() {
      ldapSettings = orgClient.getLdapSettings(orgRef.getHref());
      
      Checks.checkLdapSettings(ldapSettings);
   }
 
   @Test(description = "GET /admin/org/{id}/settings/passwordPolicy")
   public void testGetPasswordPolicy() {
      passwordPolicy = orgClient.getPasswordPolicy(orgRef.getHref());
      
      Checks.checkPasswordPolicySettings(passwordPolicy);
   }
   
   @Test(description = "PUT /admin/org/{id}/settings/passwordPolicy", 
         dependsOnMethods = { "testGetPasswordPolicy" })
   public void testUpdatePasswordPolicy() {
      boolean accountLockoutEnabled = passwordPolicy.isAccountLockoutEnabled();
      Integer invalidLoginsBeforeLockout = passwordPolicy.getInvalidLoginsBeforeLockout();
      Integer accountLockoutIntervalMinutes = passwordPolicy.getAccountLockoutIntervalMinutes();
      
      try {
         OrgPasswordPolicySettings newPasswordPolicy = passwordPolicy.toBuilder()
               .accountLockoutEnabled(!accountLockoutEnabled)
               .invalidLoginsBeforeLockout(invalidLoginsBeforeLockout+1)
               .accountLockoutIntervalMinutes(accountLockoutIntervalMinutes+1)
               .build();
         
         passwordPolicy = orgClient.updatePasswordPolicy(
               orgRef.getHref(), newPasswordPolicy);
         
         assertTrue(equal(passwordPolicy.isAccountLockoutEnabled(), !accountLockoutEnabled), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "PasswordPolicySettings", "deleteOnStorageLeaseExpiration"));
         assertTrue(equal(passwordPolicy.getInvalidLoginsBeforeLockout(), invalidLoginsBeforeLockout+1), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "PasswordPolicySettings", "storageLeaseSeconds"));
         assertTrue(equal(passwordPolicy.getAccountLockoutIntervalMinutes(), accountLockoutIntervalMinutes+1), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "PasswordPolicySettings", "deploymentLeaseSeconds"));
         
         //TODO negative tests?
         
         Checks.checkPasswordPolicySettings(passwordPolicy);
      } finally {
         passwordPolicy = passwordPolicy.toBuilder()
               .accountLockoutEnabled(accountLockoutEnabled)
               .invalidLoginsBeforeLockout(invalidLoginsBeforeLockout)
               .accountLockoutIntervalMinutes(accountLockoutIntervalMinutes)
               .build();
         
         passwordPolicy = orgClient.updatePasswordPolicy(
               orgRef.getHref(), passwordPolicy);
      }
   }
   
   @Test(description = "GET /admin/org/{id}/settings/vAppLeaseSettings")
   public void testGetVAppLeaseSettings() {
      vAppLeaseSettings = orgClient.getVAppLeaseSettings(orgRef.getHref());
      
      Checks.checkVAppLeaseSettings(vAppLeaseSettings);
   }
   
   @Test(description = "PUT /admin/org/{id}/settings/vAppLeaseSettings", 
         dependsOnMethods = { "testGetVAppLeaseSettings" } ) // FIXME: fails with 403 forbidden
   public void testUpdateVAppLeaseSettings() {
      boolean deleteOnStorageLeaseExpiration = vAppLeaseSettings.deleteOnStorageLeaseExpiration();
      Integer storageLeaseSeconds = vAppLeaseSettings.getStorageLeaseSeconds();
      Integer deploymentLeaseSeconds = vAppLeaseSettings.getDeploymentLeaseSeconds();
      
      try {
         OrgLeaseSettings newVAppLeaseSettings = vAppLeaseSettings.toBuilder()
               .deleteOnStorageLeaseExpiration(!deleteOnStorageLeaseExpiration)
               .storageLeaseSeconds(storageLeaseSeconds+1)
               .deploymentLeaseSeconds(deploymentLeaseSeconds+1)
               .build();
         
         vAppLeaseSettings = orgClient.updateVAppLeaseSettings(
               orgRef.getHref(), newVAppLeaseSettings);
         
         assertTrue(equal(vAppLeaseSettings.deleteOnStorageLeaseExpiration(), !deleteOnStorageLeaseExpiration), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "vAppLeaseSettings", "deleteOnStorageLeaseExpiration"));
         assertTrue(equal(vAppLeaseSettings.getStorageLeaseSeconds(), storageLeaseSeconds+1), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "vAppLeaseSettings", "storageLeaseSeconds"));
         assertTrue(equal(vAppLeaseSettings.getDeploymentLeaseSeconds(), deploymentLeaseSeconds+1), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "vAppLeaseSettings", "deploymentLeaseSeconds"));
         
         //TODO negative tests?
         
         Checks.checkVAppLeaseSettings(vAppLeaseSettings);
      } finally {
         vAppLeaseSettings = vAppLeaseSettings.toBuilder()
               .deleteOnStorageLeaseExpiration(deleteOnStorageLeaseExpiration)
               .storageLeaseSeconds(storageLeaseSeconds)
               .deploymentLeaseSeconds(deploymentLeaseSeconds)
               .build();
         
         vAppLeaseSettings = orgClient.updateVAppLeaseSettings(
               orgRef.getHref(), vAppLeaseSettings);
      }
   }
 
   @Test(description = "GET /admin/org/{id}/settings/vAppTemplateLeaseSettings")
   public void testGetVAppTemplateLeaseSettings() {
      vAppTemplateLeaseSettings = orgClient.getVAppTemplateLeaseSettings(orgRef.getHref());
      
      Checks.checkVAppTemplateLeaseSettings(vAppTemplateLeaseSettings);
   }
   
   @Test(description = "PUT /admin/org/{id}/settings/vAppTemplateLeaseSettings", 
         dependsOnMethods = { "testGetVAppTemplateLeaseSettings" }) // FIXME: fails with 403 forbidden
   public void testUpdateVAppTemplateLeaseSettings() {
      boolean deleteOnStorageLeaseExpiration = vAppTemplateLeaseSettings.deleteOnStorageLeaseExpiration();
      Integer storageLeaseSeconds = vAppTemplateLeaseSettings.getStorageLeaseSeconds();
      
      try {
         OrgVAppTemplateLeaseSettings newVAppTemplateLeaseSettings = vAppTemplateLeaseSettings.toBuilder()
               .deleteOnStorageLeaseExpiration(!deleteOnStorageLeaseExpiration)
               .storageLeaseSeconds(storageLeaseSeconds+1)
               .build();
         
         vAppTemplateLeaseSettings = orgClient.updateVAppTemplateLeaseSettings(
               orgRef.getHref(), newVAppTemplateLeaseSettings);
         
         assertTrue(equal(vAppTemplateLeaseSettings.deleteOnStorageLeaseExpiration(), !deleteOnStorageLeaseExpiration), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "vAppTemplateLeaseSettings", "deleteOnStorageLeaseExpiration"));
         assertTrue(equal(vAppTemplateLeaseSettings.getStorageLeaseSeconds(), storageLeaseSeconds+1), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "vAppTemplateLeaseSettings", "storageLeaseSeconds"));
         
         //TODO negative tests?
         
         Checks.checkVAppTemplateLeaseSettings(vAppTemplateLeaseSettings);
      } finally {
         vAppTemplateLeaseSettings = vAppTemplateLeaseSettings.toBuilder()
               .deleteOnStorageLeaseExpiration(deleteOnStorageLeaseExpiration)
               .storageLeaseSeconds(storageLeaseSeconds)
               .build();
         
         vAppTemplateLeaseSettings = orgClient.updateVAppTemplateLeaseSettings(
               orgRef.getHref(), vAppTemplateLeaseSettings);
      }
   }
   
   @Test(description = "GET /admin/org/{id}/settings")
   public void testGetSettings() {
      settings = orgClient.getSettings(orgRef.getHref());
      
      Checks.checkOrgSettings(settings);
   }
   
   @Test(description = "PUT /admin/org/{id}/settings",
         dependsOnMethods = { "testGetEmailSettings" } )
   public void testUpdateSettings() throws Exception {
      String newFromEmailAddress = "test"+random.nextInt(Integer.MAX_VALUE)+"@test.com";
      Exception exception = null;
      
      try {
         OrgSettings newSettings = OrgSettings.builder()
               .emailSettings(emailSettings.toBuilder().fromEmailAddress(newFromEmailAddress).build())
               .build();
         
         OrgSettings modified = orgClient.updateSettings(
               orgRef.getHref(), newSettings);
         
         Checks.checkOrgSettings(settings);
         assertTrue(equal(modified.getEmailSettings().getFromEmailAddress(), newFromEmailAddress), 
               String.format(OBJ_FIELD_UPDATABLE, 
               "orgSettings", "emailSettings"));
         
      } catch (Exception e) {
         exception = e;
      } finally {
         try {
            OrgSettings restorableSettings = OrgSettings.builder()
                     .emailSettings(emailSettings)
                     .build();
            
            settings = orgClient.updateSettings(
                     orgRef.getHref(), restorableSettings);
         } catch (Exception e) {
            if (exception != null) {
               logger.warn(e, "Error reseting settings; rethrowing origina test exception...");
               throw exception;
            } else {
               throw e;
            }
         }
      }
   }
}