1*f585d8a3SJacky Wang /* 2*f585d8a3SJacky Wang * Copyright (C) 2018 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.functional.multibindings; 18*f585d8a3SJacky Wang 19*f585d8a3SJacky Wang import static com.google.common.truth.Truth.assertThat; 20*f585d8a3SJacky Wang import static java.lang.annotation.RetentionPolicy.RUNTIME; 21*f585d8a3SJacky Wang 22*f585d8a3SJacky Wang import com.google.auto.value.AutoAnnotation; 23*f585d8a3SJacky Wang import dagger.Component; 24*f585d8a3SJacky Wang import dagger.MapKey; 25*f585d8a3SJacky Wang import dagger.Module; 26*f585d8a3SJacky Wang import dagger.Provides; 27*f585d8a3SJacky Wang import dagger.multibindings.IntoMap; 28*f585d8a3SJacky Wang import java.lang.annotation.Retention; 29*f585d8a3SJacky Wang import java.util.Map; 30*f585d8a3SJacky Wang import org.junit.Test; 31*f585d8a3SJacky Wang import org.junit.runner.RunWith; 32*f585d8a3SJacky Wang import org.junit.runners.JUnit4; 33*f585d8a3SJacky Wang 34*f585d8a3SJacky Wang @RunWith(JUnit4.class) 35*f585d8a3SJacky Wang public final class MapKeyWithDefaultTest { 36*f585d8a3SJacky Wang @Retention(RUNTIME) 37*f585d8a3SJacky Wang @MapKey(unwrapValue = false) 38*f585d8a3SJacky Wang @interface MapKeyWithDefault { hasDefault()39*f585d8a3SJacky Wang boolean hasDefault() default true; required()40*f585d8a3SJacky Wang boolean required(); 41*f585d8a3SJacky Wang } 42*f585d8a3SJacky Wang 43*f585d8a3SJacky Wang @Module 44*f585d8a3SJacky Wang interface TestModule { 45*f585d8a3SJacky Wang @Provides 46*f585d8a3SJacky Wang @IntoMap 47*f585d8a3SJacky Wang @MapKeyWithDefault(required = false) justRequired()48*f585d8a3SJacky Wang static int justRequired() { 49*f585d8a3SJacky Wang return 1; 50*f585d8a3SJacky Wang } 51*f585d8a3SJacky Wang 52*f585d8a3SJacky Wang @Provides 53*f585d8a3SJacky Wang @IntoMap 54*f585d8a3SJacky Wang @MapKeyWithDefault(required = false, hasDefault = false) both()55*f585d8a3SJacky Wang static int both() { 56*f585d8a3SJacky Wang return 2; 57*f585d8a3SJacky Wang } 58*f585d8a3SJacky Wang } 59*f585d8a3SJacky Wang 60*f585d8a3SJacky Wang @Component(modules = TestModule.class) 61*f585d8a3SJacky Wang interface TestComponent { map()62*f585d8a3SJacky Wang Map<MapKeyWithDefault, Integer> map(); 63*f585d8a3SJacky Wang } 64*f585d8a3SJacky Wang 65*f585d8a3SJacky Wang @Test test()66*f585d8a3SJacky Wang public void test() { 67*f585d8a3SJacky Wang Map<MapKeyWithDefault, Integer> map = DaggerMapKeyWithDefaultTest_TestComponent.create().map(); 68*f585d8a3SJacky Wang assertThat(map).hasSize(2); 69*f585d8a3SJacky Wang assertThat(map.get(mapKey(true, false))).isEqualTo(1); 70*f585d8a3SJacky Wang assertThat(map.get(mapKey(false, false))).isEqualTo(2); 71*f585d8a3SJacky Wang } 72*f585d8a3SJacky Wang 73*f585d8a3SJacky Wang @AutoAnnotation mapKey(boolean hasDefault, boolean required)74*f585d8a3SJacky Wang static MapKeyWithDefault mapKey(boolean hasDefault, boolean required) { 75*f585d8a3SJacky Wang return new AutoAnnotation_MapKeyWithDefaultTest_mapKey(hasDefault, required); 76*f585d8a3SJacky Wang } 77*f585d8a3SJacky Wang } 78