Add a test that checks that the formatter doesn't discard '@' with ObjC1 set.

Also set ObjC1 in the formatter tests.

The only effect of this flag in the lexer is that '@' now gets turned into
tok::at instead of tok::unknown.

llvm-svn: 171742

1 files changed, 6 insertions, 0 deletions

diff --git a/clang/unittests/Format/FormatTest.cpp b/clang/unittests/Format/FormatTest.cpp
index 2d6a1426db4..3a67e2dfa18 100644
--- a/clang/unittests/Format/FormatTest.cpp
+++ b/clang/unittests/Format/FormatTest.cpp
@@ -29,6 +29,7 @@ protected:
     LangOptions LangOpts;
     LangOpts.CPlusPlus = 1;
     LangOpts.CPlusPlus11 = 1;
+    LangOpts.ObjC1 = 1;
     Lexer Lex(ID, Context.Sources.getBuffer(ID), Context.Sources, LangOpts);
     tooling::Replacements Replace = reformat(Style, Lex, Context.Sources,
                                              Ranges);
@@ -1020,5 +1021,10 @@ TEST_F(FormatTest, FormatForObjectiveCMethodDecls) {
           "outRange8:(NSRange) out_range8  outRange9:(NSRange) out_range9;"));
 }
 
+TEST_F(FormatTest, DoNotDropAt) {
+  verifyFormat("@interface");
+  verifyFormat("@dynamic");
+}
+
 } // end namespace tooling
 } // end namespace clang