@@ -220,13 +220,13 @@ TEST(IncludeCleaner, ComputeMissingHeaders) {
220
220
TEST (IncludeCleaner, GenerateMissingHeaderDiags) {
221
221
Annotations MainFile (R"cpp(
222
222
#include "a.h"
223
- #include "a_angled .h"
223
+ #include "angled_wrapper .h"
224
224
#include "all.h"
225
225
$insert_b[[]]#include "baz.h"
226
226
#include "dir/c.h"
227
227
$insert_d[[]]$insert_foo[[]]#include "fuzz.h"
228
228
#include "header.h"
229
- $insert_foobar[[]]$insert_b_angled [[]]#include <e.h>
229
+ $insert_foobar[[]]$insert_angled [[]]#include <e.h>
230
230
$insert_f[[]]$insert_vector[[]]
231
231
232
232
#define DEF(X) const Foo *X;
@@ -238,7 +238,7 @@ TEST(IncludeCleaner, GenerateMissingHeaderDiags) {
238
238
239
239
void foo() {
240
240
$b[[b]]();
241
- $b_angled[[b_angled ]]();
241
+ $angled[[angled ]]();
242
242
243
243
ns::$bar[[Bar]] bar;
244
244
bar.d();
@@ -264,8 +264,10 @@ TEST(IncludeCleaner, GenerateMissingHeaderDiags) {
264
264
TU.Filename = " main.cpp" ;
265
265
TU.AdditionalFiles [" a.h" ] = guard (" #include \" b.h\" " );
266
266
TU.AdditionalFiles [" b.h" ] = guard (" void b();" );
267
- TU.AdditionalFiles [" a_angled.h" ] = guard (" #include \" b_angled.h\" " );
268
- TU.AdditionalFiles [" b_angled.h" ] = guard (" void b_angled();" );
267
+
268
+ TU.AdditionalFiles [" angled_wrapper.h" ] = guard (" #include <angled.h>" );
269
+ TU.AdditionalFiles [" angled.h" ] = guard (" void angled();" );
270
+ TU.ExtraArgs .push_back (" -I" + testPath (" ." ));
269
271
270
272
TU.AdditionalFiles [" dir/c.h" ] = guard (" #include \" d.h\" " );
271
273
TU.AdditionalFiles [" dir/d.h" ] =
@@ -305,7 +307,7 @@ TEST(IncludeCleaner, GenerateMissingHeaderDiags) {
305
307
return Header.ends_with (" buzz.h" );
306
308
}},
307
309
/* AngledHeaders=*/ {[](llvm::StringRef Header) {
308
- return Header.contains (" b_angled .h" );
310
+ return Header.contains (" angled .h" );
309
311
}});
310
312
EXPECT_THAT (
311
313
Diags,
@@ -315,12 +317,11 @@ TEST(IncludeCleaner, GenerateMissingHeaderDiags) {
315
317
withFix ({Fix (MainFile.range (" insert_b" ), " #include \" b.h\"\n " ,
316
318
" #include \" b.h\" " ),
317
319
FixMessage (" add all missing includes" )})),
318
- AllOf (
319
- Diag (MainFile.range (" b_angled" ),
320
- " No header providing \" b_angled\" is directly included" ),
321
- withFix ({Fix (MainFile.range (" insert_b_angled" ),
322
- " #include <b_angled.h>\n " , " #include <b_angled.h>" ),
323
- FixMessage (" add all missing includes" )})),
320
+ AllOf (Diag (MainFile.range (" angled" ),
321
+ " No header providing \" angled\" is directly included" ),
322
+ withFix ({Fix (MainFile.range (" insert_angled" ),
323
+ " #include <angled.h>\n " , " #include <angled.h>" ),
324
+ FixMessage (" add all missing includes" )})),
324
325
AllOf (Diag (MainFile.range (" bar" ),
325
326
" No header providing \" ns::Bar\" is directly included" ),
326
327
withFix ({Fix (MainFile.range (" insert_d" ),
0 commit comments