1 /* 2 * Copyright (C) 2018 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.codegen; 18 19 import com.google.common.collect.ImmutableMap; 20 import dagger.model.BindingGraph; 21 import dagger.spi.BindingGraphPlugin; 22 import dagger.spi.DiagnosticReporter; 23 24 /** 25 * A testing plugin that captures {@link dagger.model.BindingGraph}s for tests to make assertions 26 * about. 27 */ 28 // TODO(dpb): Move to dagger.spi.testing? 29 final class BindingGraphCapturer implements BindingGraphPlugin { 30 31 private final ImmutableMap.Builder<String, BindingGraph> bindingGraphs = ImmutableMap.builder(); 32 33 @Override visitGraph(BindingGraph bindingGraph, DiagnosticReporter diagnosticReporter)34 public void visitGraph(BindingGraph bindingGraph, DiagnosticReporter diagnosticReporter) { 35 bindingGraphs.put( 36 bindingGraph 37 .rootComponentNode() 38 .componentPath() 39 .currentComponent() 40 .getQualifiedName() 41 .toString(), 42 bindingGraph); 43 } 44 45 /** Returns a map of binding graphs, indexed by the canonical name of the root component type. */ bindingGraphs()46 public ImmutableMap<String, BindingGraph> bindingGraphs() { 47 return bindingGraphs.build(); 48 } 49 } 50