commit 88d158c631c4c08831855f601990f36f2234f975
author Fariborz Jahanian <fjahanian@apple.com> Thu Dec 20 02:20:09 2012 +0000
committer Fariborz Jahanian <fjahanian@apple.com> Thu Dec 20 02:20:09 2012 +0000
tree 11dbbeda6451d7eb2ca0ff542bc360b32bf64207
parent d538ed9b2a617239d5bd56357663de62f6f9224c

Provide couple of DeclPrint tests for my last patch.



git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@170635 91177308-0d34-0410-b5e6-96231b3b80d8

unittests/AST/DeclPrinterTest.cpp

diff --git a/unittests/AST/DeclPrinterTest.cpp b/unittests/AST/DeclPrinterTest.cpp
index 844b9a4..44fa742 100644
--- a/unittests/AST/DeclPrinterTest.cpp
+++ b/unittests/AST/DeclPrinterTest.cpp
@@ -1237,3 +1237,21 @@
     "- (int) A:(id)anObject inRange:(long)range"));
 }
 
+TEST(DeclPrinter, TestObjCProtocol1) {
+  ASSERT_TRUE(PrintedDeclObjCMatches(
+    "@protocol P1, P2;",
+    namedDecl(hasName("P1")).bind("id"),
+    "@protocol P1;\n"));
+  ASSERT_TRUE(PrintedDeclObjCMatches(
+    "@protocol P1, P2;",
+    namedDecl(hasName("P2")).bind("id"),
+    "@protocol P2;\n"));
+}
+
+TEST(DeclPrinter, TestObjCProtocol2) {
+  ASSERT_TRUE(PrintedDeclObjCMatches(
+    "@protocol P2 @end"
+    "@protocol P1<P2> @end",
+    namedDecl(hasName("P1")).bind("id"),
+    "@protocol P1<P2>\n@end"));
+}