From 75fdf7fd1516090651c0c3ffba4869cba9f3a879 Mon Sep 17 00:00:00 2001
From: Dave Lee <davelee.com@gmail.com>
Date: Thu, 16 Mar 2023 19:15:38 -0700
Subject: [PATCH] [lldb] Test direct ivar access in objc++ (NFC)

Add an Objective-C++ specific test for direct ivar access. This adds to the existing C++ and ObjC tests, and tests against regression for future refactoring.

Differential Revision: https://reviews.llvm.org/D146320
---
 .../commands/frame/var/direct-ivar/objcpp/Makefile |  4 +++
 .../objcpp/TestFrameVarDirectIvarObjCPlusPlus.py   | 24 +++++++++++++++
 .../commands/frame/var/direct-ivar/objcpp/main.mm  | 35 ++++++++++++++++++++++
 3 files changed, 63 insertions(+)
 create mode 100644 lldb/test/API/commands/frame/var/direct-ivar/objcpp/Makefile
 create mode 100644 lldb/test/API/commands/frame/var/direct-ivar/objcpp/TestFrameVarDirectIvarObjCPlusPlus.py
 create mode 100644 lldb/test/API/commands/frame/var/direct-ivar/objcpp/main.mm

diff --git a/lldb/test/API/commands/frame/var/direct-ivar/objcpp/Makefile b/lldb/test/API/commands/frame/var/direct-ivar/objcpp/Makefile
new file mode 100644
index 0000000..e987754
--- /dev/null
+++ b/lldb/test/API/commands/frame/var/direct-ivar/objcpp/Makefile
@@ -0,0 +1,4 @@
+OBJCXX_SOURCES := main.mm
+CFLAGS_EXTRAS := -fblocks -fobjc-arc
+LD_EXTRAS := -lobjc
+include Makefile.rules
diff --git a/lldb/test/API/commands/frame/var/direct-ivar/objcpp/TestFrameVarDirectIvarObjCPlusPlus.py b/lldb/test/API/commands/frame/var/direct-ivar/objcpp/TestFrameVarDirectIvarObjCPlusPlus.py
new file mode 100644
index 0000000..e09a8ff
--- /dev/null
+++ b/lldb/test/API/commands/frame/var/direct-ivar/objcpp/TestFrameVarDirectIvarObjCPlusPlus.py
@@ -0,0 +1,24 @@
+import lldb
+from lldbsuite.test.lldbtest import *
+from lldbsuite.test.decorators import *
+from lldbsuite.test import lldbutil
+
+
+class TestCase(TestBase):
+    @skipUnlessDarwin
+    def test_objc_self(self):
+        self.build()
+        lldbutil.run_to_source_breakpoint(self, "check self", lldb.SBFileSpec("main.mm"))
+        self.expect("frame variable _ivar", startstr="(int) _ivar = 30")
+
+    @skipUnlessDarwin
+    def test_objc_explicit_self(self):
+        self.build()
+        lldbutil.run_to_source_breakpoint(self, "check explicit self", lldb.SBFileSpec("main.mm"))
+        self.expect("frame variable _ivar", startstr="(int) _ivar = 30")
+
+    @skipUnlessDarwin
+    def test_cpp_this(self):
+        self.build()
+        lldbutil.run_to_source_breakpoint(self, "check this", lldb.SBFileSpec("main.mm"))
+        self.expect("frame variable m_field", startstr="(int) m_field = 41")
diff --git a/lldb/test/API/commands/frame/var/direct-ivar/objcpp/main.mm b/lldb/test/API/commands/frame/var/direct-ivar/objcpp/main.mm
new file mode 100644
index 0000000..2903d19
--- /dev/null
+++ b/lldb/test/API/commands/frame/var/direct-ivar/objcpp/main.mm
@@ -0,0 +1,35 @@
+#import <objc/NSObject.h>
+#include <stdio.h>
+
+struct Structure {
+  int m_field;
+  void fun() {
+    puts("check this\n");
+  }
+};
+
+@interface Classic : NSObject {
+@public
+  int _ivar;
+}
+@end
+
+@implementation Classic
+- (void)fun {
+  puts("check self\n");
+}
+@end
+
+int main() {
+  Structure s;
+  s.m_field = 41;
+  s.fun();
+
+  Classic *c = [Classic new];
+  c->_ivar = 30;
+  [c fun];
+
+  Classic *self = c;
+  puts("check explicit self\n");
+  (void)self;
+}
-- 
2.7.4