1*f585d8a3SJacky Wang /* 2*f585d8a3SJacky Wang * Copyright (C) 2016 The Dagger Authors. 3*f585d8a3SJacky Wang * 4*f585d8a3SJacky Wang * Licensed under the Apache License, Version 2.0 (the "License"); 5*f585d8a3SJacky Wang * you may not use this file except in compliance with the License. 6*f585d8a3SJacky Wang * You may obtain a copy of the License at 7*f585d8a3SJacky Wang * 8*f585d8a3SJacky Wang * http://www.apache.org/licenses/LICENSE-2.0 9*f585d8a3SJacky Wang * 10*f585d8a3SJacky Wang * Unless required by applicable law or agreed to in writing, software 11*f585d8a3SJacky Wang * distributed under the License is distributed on an "AS IS" BASIS, 12*f585d8a3SJacky Wang * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13*f585d8a3SJacky Wang * See the License for the specific language governing permissions and 14*f585d8a3SJacky Wang * limitations under the License. 15*f585d8a3SJacky Wang */ 16*f585d8a3SJacky Wang 17*f585d8a3SJacky Wang package dagger.producers.internal; 18*f585d8a3SJacky Wang 19*f585d8a3SJacky Wang import static com.google.common.truth.Truth.assertThat; 20*f585d8a3SJacky Wang import static org.junit.Assert.fail; 21*f585d8a3SJacky Wang 22*f585d8a3SJacky Wang import dagger.producers.Producer; 23*f585d8a3SJacky Wang import dagger.producers.Producers; 24*f585d8a3SJacky Wang import java.util.Map; 25*f585d8a3SJacky Wang import java.util.concurrent.ExecutionException; 26*f585d8a3SJacky Wang import org.junit.Test; 27*f585d8a3SJacky Wang import org.junit.runner.RunWith; 28*f585d8a3SJacky Wang import org.junit.runners.JUnit4; 29*f585d8a3SJacky Wang 30*f585d8a3SJacky Wang @RunWith(JUnit4.class) 31*f585d8a3SJacky Wang public final class MapProducerTest { 32*f585d8a3SJacky Wang @Test success()33*f585d8a3SJacky Wang public void success() throws Exception { 34*f585d8a3SJacky Wang Producer<Map<Integer, String>> mapProducer = 35*f585d8a3SJacky Wang MapProducer.<Integer, String>builder(2) 36*f585d8a3SJacky Wang .put(15, Producers.immediateProducer("fifteen")) 37*f585d8a3SJacky Wang .put(42, Producers.immediateProducer("forty two")) 38*f585d8a3SJacky Wang .build(); 39*f585d8a3SJacky Wang Map<Integer, String> map = mapProducer.get().get(); 40*f585d8a3SJacky Wang assertThat(map).hasSize(2); 41*f585d8a3SJacky Wang assertThat(map).containsEntry(15, "fifteen"); 42*f585d8a3SJacky Wang assertThat(map).containsEntry(42, "forty two"); 43*f585d8a3SJacky Wang } 44*f585d8a3SJacky Wang 45*f585d8a3SJacky Wang @Test failingContribution()46*f585d8a3SJacky Wang public void failingContribution() throws Exception { 47*f585d8a3SJacky Wang RuntimeException cause = new RuntimeException("monkey"); 48*f585d8a3SJacky Wang Producer<Map<Integer, String>> mapProducer = 49*f585d8a3SJacky Wang MapProducer.<Integer, String>builder(2) 50*f585d8a3SJacky Wang .put(15, Producers.immediateProducer("fifteen")) 51*f585d8a3SJacky Wang // TODO(ronshapiro): remove the type parameter when we drop java7 support 52*f585d8a3SJacky Wang .put(42, Producers.<String>immediateFailedProducer(cause)) 53*f585d8a3SJacky Wang .build(); 54*f585d8a3SJacky Wang try { 55*f585d8a3SJacky Wang mapProducer.get().get(); 56*f585d8a3SJacky Wang fail(); 57*f585d8a3SJacky Wang } catch (ExecutionException e) { 58*f585d8a3SJacky Wang assertThat(e).hasCauseThat().isSameInstanceAs(cause); 59*f585d8a3SJacky Wang } 60*f585d8a3SJacky Wang } 61*f585d8a3SJacky Wang } 62