sandbox-apis/simpledb/src/test/java/org/jclouds/simpledb/SimpleDBClientLiveTest.java

/**
 * 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.simpledb;

import static com.google.common.base.Preconditions.checkNotNull;
import static org.testng.Assert.assertNotNull;

import java.util.Properties;
import java.util.Set;
import java.util.SortedSet;

import org.jclouds.Constants;
import org.jclouds.logging.log4j.config.Log4JLoggingModule;
import org.jclouds.rest.RestContext;
import org.jclouds.rest.RestContextFactory;
import org.jclouds.simpledb.domain.ListDomainsResponse;
import org.testng.annotations.AfterTest;
import org.testng.annotations.BeforeGroups;
import org.testng.annotations.Test;

import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Sets;
import com.google.inject.Module;

/**
 * Tests behavior of {@code SimpleDBClient}
 * 
 * @author Adrian Cole
 */
@Test(groups = "live", sequential = true)
public class SimpleDBClientLiveTest {

   protected SimpleDBClient client;

   private RestContext<SimpleDBClient, SimpleDBAsyncClient> context;

   protected Set<String> domains = Sets.newHashSet();
   protected String provider = "simpledb";
   protected String identity;
   protected String credential;
   protected String endpoint;
   protected String apiVersion;

   protected void setupCredentials() {
      identity = checkNotNull(System.getProperty("test." + provider + ".identity"), "test." + provider + ".identity");
      credential = checkNotNull(System.getProperty("test." + provider + ".credential"), "test." + provider
            + ".credential");
      endpoint = System.getProperty("test." + provider + ".endpoint");
     apiVersion = System.getProperty("test." + provider + ".api-version");
   }

   protected Properties setupProperties() {
      Properties overrides = new Properties();
      overrides.setProperty(Constants.PROPERTY_TRUST_ALL_CERTS, "true");
      overrides.setProperty(Constants.PROPERTY_RELAX_HOSTNAME, "true");
      overrides.setProperty(provider + ".identity", identity);
      if (credential != null)
         overrides.setProperty(provider + ".credential", credential);
      if (endpoint != null)
         overrides.setProperty(provider + ".endpoint", endpoint);
      if (apiVersion != null)
         overrides.setProperty(provider + ".api-version", apiVersion);
      return overrides;
   }

   @BeforeGroups(groups = { "live" })
   public void setupClient() {
      setupCredentials();
      Properties overrides = setupProperties();
      context = new RestContextFactory().createContext(provider, ImmutableSet.<Module> of(new Log4JLoggingModule()),
            overrides);
      this.client = context.getApi();
   }

   @Test
   protected void testListDomains() throws InterruptedException {
      listDomainInRegion(null);
   }

   protected void listDomainInRegion(String region) throws InterruptedException {
      SortedSet<String> allResults = Sets.newTreeSet(client.listDomainsInRegion(region));
      assertNotNull(allResults);
      if (allResults.size() >= 1) {
         String domain = allResults.last();
         assertDomainInList(region, domain);
      }
   }

   public static final String PREFIX = System.getProperty("user.name") + "-simpledb";

   protected String createDomainInRegion(String region, String domainName) throws InterruptedException {
      try {
         SortedSet<String> result = Sets.newTreeSet(client.listDomainsInRegion(region));
         if (result.size() >= 1) {
            client.deleteDomainInRegion(region, result.last());
            domainName += 1;// cannot recreate a domain within 60 seconds
         }
      } catch (Exception e) {

      }
      client.createDomainInRegion(region, domainName);

      // TODO get the domain metadata and ensure the region is correct

      // if (region != null)
      // assertEquals(domain.getRegion(), region);
      // assertEquals(domain.getName(), domainName);
      assertDomainInList(region, domainName);
      domains.add(domainName);
      return domainName;
   }

   @Test
   protected void testCreateDomain() throws InterruptedException {
      createDomainInRegion(null, PREFIX + "1");
   }

   protected void assertDomainInList(final String region, final String domain) throws InterruptedException {
      assertEventually(new Runnable() {
         public void run() {
            ListDomainsResponse domains = client.listDomainsInRegion(region);
            assert domains.contains(domain) : domain + " not in " + domains;
         }
      });
   }

   private static final int INCONSISTENCY_WINDOW = 10000;

   /**
    * Due to eventual consistency, container commands may not return correctly immediately. Hence,
    * we will try up to the inconsistency window to see if the assertion completes.
    */
   protected static void assertEventually(Runnable assertion) throws InterruptedException {
      long start = System.currentTimeMillis();
      AssertionError error = null;
      for (int i = 0; i < 30; i++) {
         try {
            assertion.run();
            if (i > 0)
               System.err.printf("%d attempts and %dms asserting %s%n", i + 1, System.currentTimeMillis() - start,
                     assertion.getClass().getSimpleName());
            return;
         } catch (AssertionError e) {
            error = e;
         }
         Thread.sleep(INCONSISTENCY_WINDOW / 30);
      }
      if (error != null)
         throw error;
   }

   @AfterTest
   public void shutdown() {
      context.close();
   }
}