11d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/* 21d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Copyright (C) 2008 The Guava Authors 31d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 41d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Licensed under the Apache License, Version 2.0 (the "License"); 51d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * you may not use this file except in compliance with the License. 61d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * You may obtain a copy of the License at 71d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 81d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * http://www.apache.org/licenses/LICENSE-2.0 91d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 101d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Unless required by applicable law or agreed to in writing, software 111d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * distributed under the License is distributed on an "AS IS" BASIS, 121d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 131d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * See the License for the specific language governing permissions and 141d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * limitations under the License. 151d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 161d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 171d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpackage com.google.common.util.concurrent; 181d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 191d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport com.google.common.base.Function; 201d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 211d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.lang.reflect.UndeclaredThrowableException; 221d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 231d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/** 241d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Unit tests for {@link Futures#transform(ListenableFuture, Function)}. 251d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 261d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * @author Nishant Thakkar 271d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 281d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpublic class FuturesTransformTest 291d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert extends AbstractChainedListenableFutureTest<String> { 301d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert private static final String RESULT_DATA = "SUCCESS"; 311d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 321d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override protected ListenableFuture<String> buildChainingFuture( 331d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ListenableFuture<Integer> inputFuture) { 341d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return Futures.transform(inputFuture, 351d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert new ComposeFunction()); 361d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 371d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 381d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override protected String getSuccessfulResult() { 391d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return RESULT_DATA; 401d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 411d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 421d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert private class ComposeFunction 431d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert implements Function<Integer, String> { 441d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @Override 451d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public String apply(Integer input) { 461d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert if (input.intValue() == VALID_INPUT_DATA) { 471d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert return RESULT_DATA; 481d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } else { 491d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert throw new UndeclaredThrowableException(EXCEPTION); 501d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 511d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 521d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 531d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 541d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testFutureGetThrowsFunctionException() throws Exception { 551d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert inputFuture.set(EXCEPTION_DATA); 561d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert listener.assertException(EXCEPTION); 571d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 581d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert} 59