core/src/test/java/org/jclouds/predicates/RetryablePredicateTest.java

/**
 *
 * Copyright (C) 2009 Cloud Conscious, LLC. <info@cloudconscious.com>
 *
 * ====================================================================
 * Licensed 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.predicates;

import java.util.Date;
import java.util.concurrent.TimeUnit;

import org.testng.annotations.Test;

import com.google.common.base.Predicate;
import com.google.common.base.Predicates;

/**
 * 
 * @author Adrian Cole
 * 
 */
@Test(groups = "unit", sequential = true, testName = "jclouds.RetryablePredicateTest")
public class RetryablePredicateTest {

   @Test
   void testAlwaysTrue() {
      RetryablePredicate<String> predicate = new RetryablePredicate<String>(Predicates
               .<String> alwaysTrue(), 3, 1, TimeUnit.SECONDS);
      Date startPlusSecond = new Date(System.currentTimeMillis() + 1000);
      predicate.apply("");
      Date now = new Date();
      assert now.compareTo(startPlusSecond) < 0 : String.format("%s should be less than %s", now,
               startPlusSecond);
   }

   @Test
   void testAlwaysFalseMillis() {
      RetryablePredicate<String> predicate = new RetryablePredicate<String>(Predicates
               .<String> alwaysFalse(), 3, 1, TimeUnit.SECONDS);
      Date startPlus3Seconds = new Date(System.currentTimeMillis() + 3000);
      Date startPlus4Seconds = new Date(System.currentTimeMillis() + 4000);
      predicate.apply("");
      Date now = new Date();
      assert now.compareTo(startPlus3Seconds) >= 0 : String.format("%s should be less than %s",
               startPlus3Seconds, now);
      assert now.compareTo(startPlus4Seconds) <= 0 : String.format("%s should be greater than %s",
               startPlus4Seconds, now);

   }

   private static class SecondTimeTrue implements Predicate<String> {

      private int count = 0;

      @Override
      public boolean apply(String input) {
         return count++ == 1;
      }

   }

   @Test
   void testSecondTimeTrue() {
      RetryablePredicate<String> predicate = new RetryablePredicate<String>(new SecondTimeTrue(),
               3, 1, TimeUnit.SECONDS);

      Date startPlusSecond = new Date(System.currentTimeMillis() + 1000);
      Date startPlus2Seconds = new Date(System.currentTimeMillis() + 2000);

      predicate.apply("");
      Date now = new Date();
      assert now.compareTo(startPlusSecond) >= 0 : String.format("%s should be greater than %s",
               now, startPlusSecond);
      assert now.compareTo(startPlus2Seconds) <= 0 : String.format("%s should be greater than %s",
               startPlus2Seconds, now);
   }

}