1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
/*
* Copyright (C) 2021 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.spi.model.testing;
import static com.google.common.collect.Iterables.getOnlyElement;
import static com.google.common.truth.Truth.assertAbout;
import static dagger.internal.codegen.extension.DaggerStreams.toImmutableSet;
import com.google.common.collect.ImmutableSet;
import com.google.common.truth.FailureMetadata;
import com.google.common.truth.Subject;
import dagger.spi.model.Binding;
import dagger.spi.model.BindingGraph;
import dagger.spi.model.DaggerType;
import javax.lang.model.type.TypeMirror;
import org.checkerframework.checker.nullness.compatqual.NullableDecl;
/** A Truth subject for making assertions on a {@link BindingGraph}. */
public final class BindingGraphSubject extends Subject {
/** Starts a fluent assertion about a {@link BindingGraph}. */
public static BindingGraphSubject assertThat(BindingGraph bindingGraph) {
return assertAbout(BindingGraphSubject::new).that(bindingGraph);
}
private final BindingGraph actual;
private BindingGraphSubject(FailureMetadata metadata, @NullableDecl BindingGraph actual) {
super(metadata, actual);
this.actual = actual;
}
/**
* Asserts that the graph has at least one binding with an unqualified key.
*
* @param type the canonical name of the type, as returned by {@link TypeMirror#toString()}
*/
public void hasBindingWithKey(String type) {
bindingWithKey(type);
}
/**
* Asserts that the graph has at least one binding with a qualified key.
*
* @param qualifier the canonical string form of the qualifier, as returned by {@link
* javax.lang.model.element.AnnotationMirror AnnotationMirror.toString()}
* @param type the canonical name of the type, as returned by {@link TypeMirror#toString()}
*/
public void hasBindingWithKey(String qualifier, String type) {
bindingWithKey(qualifier, type);
}
/**
* Returns a subject for testing the binding for an unqualified key.
*
* @param type the canonical name of the type, as returned by {@link TypeMirror#toString()}
*/
public BindingSubject bindingWithKey(String type) {
return bindingWithKeyString(type);
}
/**
* Returns a subject for testing the binding for a qualified key.
*
* @param qualifier the canonical string form of the qualifier, as returned by {@link
* javax.lang.model.element.AnnotationMirror AnnotationMirror.toString()}
* @param type the canonical name of the type, as returned by {@link TypeMirror#toString()}
*/
public BindingSubject bindingWithKey(String qualifier, String type) {
return bindingWithKeyString(keyString(qualifier, type));
}
private BindingSubject bindingWithKeyString(String keyString) {
ImmutableSet<Binding> bindings = getBindingNodes(keyString);
// TODO(dpb): Handle multiple bindings for the same key.
check("bindingsWithKey(%s)", keyString).that(bindings).hasSize(1);
return check("bindingWithKey(%s)", keyString)
.about(BindingSubject::new)
.that(getOnlyElement(bindings));
}
private ImmutableSet<Binding> getBindingNodes(String keyString) {
return actual.bindings().stream()
.filter(binding -> keyString(binding).equals(keyString))
.collect(toImmutableSet());
}
public static String keyString(Binding binding) {
return binding.key().qualifier().isPresent()
? keyString(binding.key().qualifier().get().toString(), formattedType(binding.key().type()))
: formattedType(binding.key().type());
}
private static String keyString(String qualifier, String type) {
return String.format("%s %s", qualifier, type);
}
private static String formattedType(DaggerType type) {
switch (type.backend()) {
case JAVAC:
return type.javac().toString();
case KSP:
return type.ksp().getDeclaration().getQualifiedName().asString();
}
throw new AssertionError("Unsupported backend");
}
/** A Truth subject for a {@link Binding}. */
public final class BindingSubject extends Subject {
private final Binding actual;
BindingSubject(FailureMetadata metadata, @NullableDecl Binding actual) {
super(metadata, actual);
this.actual = actual;
}
/**
* Asserts that the binding depends on a binding with an unqualified key.
*
* @param type the canonical name of the type, as returned by {@link TypeMirror#toString()}
*/
public void dependsOnBindingWithKey(String type) {
dependsOnBindingWithKeyString(type);
}
/**
* Asserts that the binding depends on a binding with a qualified key.
*
* @param qualifier the canonical string form of the qualifier, as returned by {@link
* javax.lang.model.element.AnnotationMirror AnnotationMirror.toString()}
* @param type the canonical name of the type, as returned by {@link TypeMirror#toString()}
*/
public void dependsOnBindingWithKey(String qualifier, String type) {
dependsOnBindingWithKeyString(keyString(qualifier, type));
}
private void dependsOnBindingWithKeyString(String keyString) {
if (actualBindingGraph().requestedBindings(actual).stream()
.noneMatch(binding -> keyString(binding).equals(keyString))) {
failWithActual("expected to depend on binding with key", keyString);
}
}
private BindingGraph actualBindingGraph() {
return BindingGraphSubject.this.actual;
}
}
}
|
