@@ -2125,6 +2125,13 @@ TEST_F(TokenAnnotatorTest, UnderstandsTrailingReturnArrow) {
2125
2125
ASSERT_EQ (Tokens.size (), 21u ) << Tokens;
2126
2126
EXPECT_TOKEN (Tokens[13 ], tok::arrow, TT_Unknown);
2127
2127
2128
+ auto Style = getLLVMStyle ();
2129
+ Style.StatementAttributeLikeMacros .push_back (" emit" );
2130
+ Tokens = annotate (" emit foo()->bar;" , Style);
2131
+ ASSERT_EQ (Tokens.size (), 8u ) << Tokens;
2132
+ EXPECT_TOKEN (Tokens[0 ], tok::identifier, TT_StatementAttributeLikeMacro);
2133
+ EXPECT_TOKEN (Tokens[4 ], tok::arrow, TT_Unknown);
2134
+
2128
2135
// Mixed
2129
2136
Tokens = annotate (" auto f() -> int { auto a = b()->c; }" );
2130
2137
ASSERT_EQ (Tokens.size (), 18u ) << Tokens;
@@ -2950,6 +2957,13 @@ TEST_F(TokenAnnotatorTest, StartOfName) {
2950
2957
ASSERT_EQ (Tokens.size (), 7u ) << Tokens;
2951
2958
EXPECT_TOKEN (Tokens[0 ], tok::at, TT_ObjCDecl);
2952
2959
EXPECT_TOKEN (Tokens[2 ], tok::identifier, TT_StartOfName);
2960
+
2961
+ auto Style = getLLVMStyle ();
2962
+ Style.StatementAttributeLikeMacros .push_back (" emit" );
2963
+ Tokens = annotate (" emit foo = 0;" , Style);
2964
+ ASSERT_EQ (Tokens.size (), 6u ) << Tokens;
2965
+ EXPECT_TOKEN (Tokens[0 ], tok::identifier, TT_StatementAttributeLikeMacro);
2966
+ EXPECT_TOKEN (Tokens[1 ], tok::identifier, TT_Unknown);
2953
2967
}
2954
2968
2955
2969
TEST_F (TokenAnnotatorTest, BraceKind) {
0 commit comments