1 /* 2 * Copyright (C) 2014 The Dagger Authors. 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package dagger.internal; 18 19 import static com.google.common.truth.Truth.assertThat; 20 21 import com.google.common.collect.ImmutableSet; 22 import java.util.Arrays; 23 import java.util.LinkedHashSet; 24 import java.util.Set; 25 import java.util.concurrent.atomic.AtomicInteger; 26 import org.junit.Rule; 27 import org.junit.Test; 28 import org.junit.rules.ExpectedException; 29 import org.junit.runner.RunWith; 30 import org.junit.runners.JUnit4; 31 32 @RunWith(JUnit4.class) 33 @SuppressWarnings("unchecked") 34 public class SetFactoryTest { 35 @Rule public ExpectedException thrown = ExpectedException.none(); 36 37 @Test providerReturnsNull()38 public void providerReturnsNull() { 39 Factory<Set<Integer>> factory = 40 SetFactory.<Integer>builder(0, 1).addCollectionProvider(() -> null).build(); 41 thrown.expect(NullPointerException.class); 42 factory.get(); 43 } 44 45 @Test providerReturnsNullSet()46 public void providerReturnsNullSet() { 47 Factory<Set<Integer>> factory = 48 SetFactory.<Integer>builder(1, 0).addProvider(() -> null).build(); 49 thrown.expect(NullPointerException.class); 50 factory.get(); 51 } 52 53 @Test providerReturnsSetWithNullElement()54 public void providerReturnsSetWithNullElement() { 55 Set<Integer> set = new LinkedHashSet<>(Arrays.asList(1, null, 3)); 56 Factory<Set<Integer>> factory = 57 SetFactory.<Integer>builder(0, 1).addCollectionProvider(() -> set).build(); 58 thrown.expect(NullPointerException.class); 59 factory.get(); 60 } 61 62 @Test invokesProvidersEveryTime()63 public void invokesProvidersEveryTime() { 64 Factory<Set<Integer>> factory = 65 SetFactory.<Integer>builder(2, 2) 66 .addProvider(incrementingIntegerProvider(0)) 67 .addProvider(incrementingIntegerProvider(10)) 68 .addCollectionProvider(incrementingIntegerSetProvider(20)) 69 .addCollectionProvider(incrementingIntegerSetProvider(30)) 70 .build(); 71 assertThat(factory.get()).containsExactly(0, 10, 20, 21, 30, 31); 72 assertThat(factory.get()).containsExactly(1, 11, 22, 23, 32, 33); 73 assertThat(factory.get()).containsExactly(2, 12, 24, 25, 34, 35); 74 } 75 incrementingIntegerProvider(int seed)76 private static Provider<Integer> incrementingIntegerProvider(int seed) { 77 final AtomicInteger value = new AtomicInteger(seed); 78 return value::getAndIncrement; 79 } 80 incrementingIntegerSetProvider(int seed)81 private static Provider<Set<Integer>> incrementingIntegerSetProvider(int seed) { 82 final AtomicInteger value = new AtomicInteger(seed); 83 return () -> ImmutableSet.of(value.getAndIncrement(), value.getAndIncrement()); 84 } 85 } 86