| /* |
| * Copyright (C) 2014 The Dagger Authors. |
| * |
| * 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 dagger.internal.codegen; |
| |
| import static com.google.testing.compile.CompilationSubject.assertThat; |
| import static dagger.internal.codegen.Compilers.compilerWithOptions; |
| import static dagger.internal.codegen.Compilers.daggerCompiler; |
| import static dagger.internal.codegen.TestUtils.message; |
| |
| import com.google.common.collect.ImmutableList; |
| import com.google.testing.compile.Compilation; |
| import com.google.testing.compile.JavaFileObjects; |
| import javax.tools.JavaFileObject; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.junit.runners.JUnit4; |
| |
| @RunWith(JUnit4.class) |
| public class MapMultibindingValidationTest { |
| @Test |
| public void duplicateMapKeys_UnwrappedMapKey() { |
| JavaFileObject module = |
| JavaFileObjects.forSourceLines( |
| "test.MapModule", |
| "package test;", |
| "", |
| "import dagger.Module;", |
| "import dagger.Provides;", |
| "import dagger.multibindings.StringKey;", |
| "import dagger.multibindings.IntoMap;", |
| "", |
| "@Module", |
| "final class MapModule {", |
| " @Provides @IntoMap @StringKey(\"AKey\") Object provideObjectForAKey() {", |
| " return \"one\";", |
| " }", |
| "", |
| " @Provides @IntoMap @StringKey(\"AKey\") Object provideObjectForAKeyAgain() {", |
| " return \"one again\";", |
| " }", |
| "}"); |
| |
| // If they're all there, report only Map<K, V>. |
| Compilation compilation = |
| daggerCompiler() |
| .compile( |
| module, |
| component( |
| "Map<String, Object> objects();", |
| "Map<String, Provider<Object>> objectProviders();", |
| "Producer<Map<String, Producer<Object>>> objectProducers();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "The same map key is bound more than once for " |
| + "Map<String,Object>"); |
| assertThat(compilation).hadErrorContaining("provideObjectForAKey()"); |
| assertThat(compilation).hadErrorContaining("provideObjectForAKeyAgain()"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| compilation = |
| compilerWithOptions("-Adagger.fullBindingGraphValidation=ERROR") |
| .compile(module); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "The same map key is bound more than once for " |
| + "Map<String,Provider<Object>>") |
| .inFile(module) |
| .onLineContaining("class MapModule"); |
| assertThat(compilation).hadErrorContaining("provideObjectForAKey()"); |
| assertThat(compilation).hadErrorContaining("provideObjectForAKeyAgain()"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| // If there's Map<K, V> and Map<K, Provider<V>>, report only Map<K, V>. |
| compilation = |
| daggerCompiler() |
| .compile( |
| module, |
| component( |
| "Map<String, Object> objects();", |
| "Map<String, Provider<Object>> objectProviders();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "The same map key is bound more than once for " |
| + "Map<String,Object>"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| // If there's Map<K, V> and Map<K, Producer<V>>, report only Map<K, V>. |
| compilation = |
| daggerCompiler() |
| .compile( |
| module, |
| component( |
| "Map<String, Object> objects();", |
| "Producer<Map<String, Producer<Object>>> objectProducers();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "The same map key is bound more than once for " |
| + "Map<String,Object>"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| // If there's Map<K, Provider<V>> and Map<K, Producer<V>>, report only Map<K, Provider<V>>. |
| compilation = |
| daggerCompiler() |
| .compile( |
| module, |
| component( |
| "Map<String, Provider<Object>> objectProviders();", |
| "Producer<Map<String, Producer<Object>>> objectProducers();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "The same map key is bound more than once for " |
| + "Map<String,Provider<Object>>"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| compilation = daggerCompiler().compile(module, component("Map<String, Object> objects();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "The same map key is bound more than once for " |
| + "Map<String,Object>"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| compilation = |
| daggerCompiler() |
| .compile(module, component("Map<String, Provider<Object>> objectProviders();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "The same map key is bound more than once for " |
| + "Map<String,Provider<Object>>"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| compilation = |
| daggerCompiler() |
| .compile( |
| module, component("Producer<Map<String, Producer<Object>>> objectProducers();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "The same map key is bound more than once for " |
| + "Map<String,Producer<Object>>"); |
| assertThat(compilation).hadErrorCount(1); |
| } |
| |
| @Test |
| public void duplicateMapKeys_WrappedMapKey() { |
| JavaFileObject module = |
| JavaFileObjects.forSourceLines( |
| "test.MapModule", |
| "package test;", |
| "", |
| "import dagger.Module;", |
| "import dagger.Provides;", |
| "import dagger.multibindings.IntoMap;", |
| "import dagger.MapKey;", |
| "", |
| "@Module", |
| "abstract class MapModule {", |
| "", |
| " @MapKey(unwrapValue = false)", |
| " @interface WrappedMapKey {", |
| " String value();", |
| " }", |
| "", |
| " @Provides", |
| " @IntoMap", |
| " @WrappedMapKey(\"foo\")", |
| " static String stringMapEntry1() { return \"\"; }", |
| "", |
| " @Provides", |
| " @IntoMap", |
| " @WrappedMapKey(\"foo\")", |
| " static String stringMapEntry2() { return \"\"; }", |
| "}"); |
| |
| JavaFileObject component = component("Map<test.MapModule.WrappedMapKey, String> objects();"); |
| |
| Compilation compilation = daggerCompiler().compile(component, module); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| message( |
| "\033[1;31m[Dagger/MapKeys]\033[0m The same map key is bound more than once for " |
| + "Map<MapModule.WrappedMapKey,String>", |
| " @Provides @IntoMap @MapModule.WrappedMapKey(\"foo\") String " |
| + "MapModule.stringMapEntry1()", |
| " @Provides @IntoMap @MapModule.WrappedMapKey(\"foo\") String " |
| + "MapModule.stringMapEntry2()")) |
| .inFile(component) |
| .onLineContaining("interface TestComponent"); |
| } |
| |
| @Test |
| public void inconsistentMapKeyAnnotations() { |
| JavaFileObject module = |
| JavaFileObjects.forSourceLines( |
| "test.MapModule", |
| "package test;", |
| "", |
| "import dagger.Module;", |
| "import dagger.Provides;", |
| "import dagger.multibindings.StringKey;", |
| "import dagger.multibindings.IntoMap;", |
| "", |
| "@Module", |
| "final class MapModule {", |
| " @Provides @IntoMap @StringKey(\"AKey\") Object provideObjectForAKey() {", |
| " return \"one\";", |
| " }", |
| "", |
| " @Provides @IntoMap @StringKeyTwo(\"BKey\") Object provideObjectForBKey() {", |
| " return \"two\";", |
| " }", |
| "}"); |
| JavaFileObject stringKeyTwoFile = |
| JavaFileObjects.forSourceLines( |
| "test.StringKeyTwo", |
| "package test;", |
| "", |
| "import dagger.MapKey;", |
| "", |
| "@MapKey(unwrapValue = true)", |
| "public @interface StringKeyTwo {", |
| " String value();", |
| "}"); |
| |
| // If they're all there, report only Map<K, V>. |
| Compilation compilation = |
| daggerCompiler() |
| .compile( |
| module, |
| stringKeyTwoFile, |
| component( |
| "Map<String, Object> objects();", |
| "Map<String, Provider<Object>> objectProviders();", |
| "Producer<Map<String, Producer<Object>>> objectProducers();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "Map<String,Object>" |
| + " uses more than one @MapKey annotation type"); |
| assertThat(compilation).hadErrorContaining("provideObjectForAKey()"); |
| assertThat(compilation).hadErrorContaining("provideObjectForBKey()"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| compilation = |
| compilerWithOptions("-Adagger.fullBindingGraphValidation=ERROR") |
| .compile(module, stringKeyTwoFile); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "Map<String,Provider<Object>>" |
| + " uses more than one @MapKey annotation type") |
| .inFile(module) |
| .onLineContaining("class MapModule"); |
| assertThat(compilation).hadErrorContaining("provideObjectForAKey()"); |
| assertThat(compilation).hadErrorContaining("provideObjectForBKey()"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| // If there's Map<K, V> and Map<K, Provider<V>>, report only Map<K, V>. |
| compilation = |
| daggerCompiler() |
| .compile( |
| module, |
| stringKeyTwoFile, |
| component( |
| "Map<String, Object> objects();", |
| "Map<String, Provider<Object>> objectProviders();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "Map<String,Object>" |
| + " uses more than one @MapKey annotation type"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| // If there's Map<K, V> and Map<K, Producer<V>>, report only Map<K, V>. |
| compilation = |
| daggerCompiler() |
| .compile( |
| module, |
| stringKeyTwoFile, |
| component( |
| "Map<String, Object> objects();", |
| "Producer<Map<String, Producer<Object>>> objectProducers();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "Map<String,Object>" |
| + " uses more than one @MapKey annotation type"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| // If there's Map<K, Provider<V>> and Map<K, Producer<V>>, report only Map<K, Provider<V>>. |
| compilation = |
| daggerCompiler() |
| .compile( |
| module, |
| stringKeyTwoFile, |
| component( |
| "Map<String, Provider<Object>> objectProviders();", |
| "Producer<Map<String, Producer<Object>>> objectProducers();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "Map<String,Provider<Object>>" |
| + " uses more than one @MapKey annotation type"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| compilation = |
| daggerCompiler() |
| .compile(module, stringKeyTwoFile, component("Map<String, Object> objects();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "Map<String,Object>" |
| + " uses more than one @MapKey annotation type"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| compilation = |
| daggerCompiler() |
| .compile( |
| module, |
| stringKeyTwoFile, |
| component("Map<String, Provider<Object>> objectProviders();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "Map<String,Provider<Object>>" |
| + " uses more than one @MapKey annotation type"); |
| assertThat(compilation).hadErrorCount(1); |
| |
| compilation = |
| daggerCompiler() |
| .compile( |
| module, |
| stringKeyTwoFile, |
| component("Producer<Map<String, Producer<Object>>> objectProducers();")); |
| assertThat(compilation).failed(); |
| assertThat(compilation) |
| .hadErrorContaining( |
| "Map<String,Producer<Object>>" |
| + " uses more than one @MapKey annotation type"); |
| assertThat(compilation).hadErrorCount(1); |
| } |
| |
| private static JavaFileObject component(String... entryPoints) { |
| return JavaFileObjects.forSourceLines( |
| "test.TestComponent", |
| ImmutableList.<String>builder() |
| .add( |
| "package test;", |
| "", |
| "import dagger.Component;", |
| "import dagger.producers.Producer;", |
| "import java.util.Map;", |
| "import javax.inject.Provider;", |
| "", |
| "@Component(modules = {MapModule.class})", |
| "interface TestComponent {") |
| .add(entryPoints) |
| .add("}") |
| .build()); |
| } |
| } |