Patch for rewriting of @protocol.

llvm-svn: 44681

1 files changed, 48 insertions, 0 deletions

diff --git a/clang/test/Sema/protocol-rewrite-1.m b/clang/test/Sema/protocol-rewrite-1.m
new file mode 100644
index 00000000000..a9df2ec7936
--- /dev/null
+++ b/clang/test/Sema/protocol-rewrite-1.m
@@ -0,0 +1,48 @@
+// RUN: clang -rewrite-test %s
+
+typedef struct MyWidget {
+  int a;
+} MyWidget;
+
+MyWidget gWidget = { 17 };
+
+@protocol MyProto
+- (MyWidget *)widget;
+@end
+
+@interface Foo 
+@end
+
+@interface Bar: Foo <MyProto>
+@end
+
+@interface Container 
++ (MyWidget *)elementForView:(Foo *)view;
+@end
+
+@implementation Foo
+@end
+
+@implementation Bar
+- (MyWidget *)widget {
+  return &gWidget;
+}
+@end
+
+@implementation Container
++ (MyWidget *)elementForView:(Foo *)view
+{
+  MyWidget *widget = (void*)0;
+  if (@protocol(MyProto)) {
+    widget = [(id <MyProto>)view widget];
+  }
+  return widget;
+}
+@end
+
+int main(void) {
+  id view;
+  MyWidget *w = [Container elementForView: view];
+
+  return 0;
+}