9 files changed, 49 insertions, 16 deletions
diff --git a/lldb/packages/Python/lldbsuite/test/api/check_public_api_headers/TestPublicAPIHeaders.py b/lldb/packages/Python/lldbsuite/test/api/check_public_api_headers/TestPublicAPIHeaders.py
index dd6dbe0a37e..0b202a092f2 100644
--- a/lldb/packages/Python/lldbsuite/test/api/check_public_api_headers/TestPublicAPIHeaders.py
+++ b/lldb/packages/Python/lldbsuite/test/api/check_public_api_headers/TestPublicAPIHeaders.py
@@ -19,9 +19,9 @@ class SBDirCheckerCase(TestBase):
 
     def setUp(self):
         TestBase.setUp(self)
+        self.template = 'main.cpp.template'
         self.source = 'main.cpp'
         self.exe_name = 'a.out'
-        self.generateSource(self.source)
 
     @skipIfNoSBHeaders
     def test_sb_api_directory(self):
@@ -34,9 +34,40 @@ class SBDirCheckerCase(TestBase):
             self.skipTest(
                 "LLDB is 64-bit and cannot be linked to 32-bit test program.")
 
+        # Generate main.cpp, build it, and execute.
+        self.generate_main_cpp()
         self.buildDriver(self.source, self.exe_name)
         self.sanity_check_executable(self.exe_name)
 
+    def generate_main_cpp(self):
+        """Generate main.cpp from main.cpp.template."""
+        temp = os.path.join(os.getcwd(), self.template)
+        with open(temp, 'r') as f:
+            content = f.read()
+
+        public_api_dir = os.path.join(
+            os.environ["LLDB_SRC"], "include", "lldb", "API")
+
+        # Look under the include/lldb/API directory and add #include statements
+        # for all the SB API headers.
+        public_headers = os.listdir(public_api_dir)
+        # For different platforms, the include statement can vary.
+        if self.platformIsDarwin():
+            include_stmt = "'#include <%s>' % os.path.join('LLDB', header)"
+        if self.getPlatform() == "freebsd" or self.getPlatform(
+        ) == "linux" or os.environ.get('LLDB_BUILD_TYPE') == 'Makefile':
+            include_stmt = "'#include <%s>' % os.path.join(public_api_dir, header)"
+        list = [eval(include_stmt) for header in public_headers if (
+            header.startswith("SB") and header.endswith(".h"))]
+        includes = '\n'.join(list)
+        new_content = content.replace('%include_SB_APIs%', includes)
+        src = os.path.join(os.getcwd(), self.source)
+        with open(src, 'w') as f:
+            f.write(new_content)
+
+        # The main.cpp has been generated, add a teardown hook to remove it.
+        self.addTearDownHook(lambda: os.remove(src))
+
     def sanity_check_executable(self, exe_name):
         """Sanity check executable compiled from the auto-generated program."""
         exe = os.path.join(os.getcwd(), exe_name)
diff --git a/lldb/packages/Python/lldbsuite/test/api/multithreaded/TestMultithreaded.py b/lldb/packages/Python/lldbsuite/test/api/multithreaded/TestMultithreaded.py
index abd252b1e5b..970c25107f6 100644
--- a/lldb/packages/Python/lldbsuite/test/api/multithreaded/TestMultithreaded.py
+++ b/lldb/packages/Python/lldbsuite/test/api/multithreaded/TestMultithreaded.py
@@ -15,15 +15,6 @@ import subprocess
 
 class SBBreakpointCallbackCase(TestBase):
 
-    def setUp(self):
-        TestBase.setUp(self)
-        self.generateSource('driver.cpp')
-        self.generateSource('listener_test.cpp')
-        self.generateSource('test_breakpoint_callback.cpp')
-        self.generateSource('test_listener_event_description.cpp')
-        self.generateSource('test_listener_event_process_state.cpp')
-        self.generateSource('test_listener_resume.cpp')
-
     mydir = TestBase.compute_mydir(__file__)
 
     @skipIfRemote
diff --git a/lldb/packages/Python/lldbsuite/test/api/multithreaded/driver.cpp.template b/lldb/packages/Python/lldbsuite/test/api/multithreaded/driver.cpp
index adb1d200655..fa0c48e0ecf 100644
--- a/lldb/packages/Python/lldbsuite/test/api/multithreaded/driver.cpp.template
+++ b/lldb/packages/Python/lldbsuite/test/api/multithreaded/driver.cpp
@@ -7,7 +7,7 @@
 #include <string>
 #include <vector>
 
-%include_SB_APIs%
+#include "lldb-headers.h"
 
 #include "common.h"
 
diff --git a/lldb/packages/Python/lldbsuite/test/api/multithreaded/listener_test.cpp.template b/lldb/packages/Python/lldbsuite/test/api/multithreaded/listener_test.cpp
index e305d1af489..b20868ff94f 100644
--- a/lldb/packages/Python/lldbsuite/test/api/multithreaded/listener_test.cpp.template
+++ b/lldb/packages/Python/lldbsuite/test/api/multithreaded/listener_test.cpp
@@ -7,7 +7,7 @@
 #include <thread>
 #include <vector>
 
-%include_SB_APIs%
+#include "lldb-headers.h"
 #include "common.h"
 
 using namespace lldb;
diff --git a/lldb/packages/Python/lldbsuite/test/api/multithreaded/lldb-headers.h b/lldb/packages/Python/lldbsuite/test/api/multithreaded/lldb-headers.h
new file mode 100644
index 00000000000..da0914b3b07
--- /dev/null
+++ b/lldb/packages/Python/lldbsuite/test/api/multithreaded/lldb-headers.h
@@ -0,0 +1,11 @@
+
+#ifndef LLDB_HEADERS_H
+#define LLDB_HEADERS_H
+
+#ifdef __APPLE__
+#include <LLDB/LLDB.h>
+#else
+#include "lldb/API/LLDB.h"
+#endif
+
+#endif // LLDB_HEADERS_H
diff --git a/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_breakpoint_callback.cpp.template b/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_breakpoint_callback.cpp
index 4133025aa49..def31f82b0c 100644
--- a/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_breakpoint_callback.cpp.template
+++ b/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_breakpoint_callback.cpp
@@ -7,7 +7,7 @@
 #include <vector>
 #include <string>
 
-%include_SB_APIs%
+#include "lldb-headers.h"
 
 #include "common.h"
 
diff --git a/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_listener_event_description.cpp.template b/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_listener_event_description.cpp
index 63e3f3631e5..0d7844dce6d 100644
--- a/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_listener_event_description.cpp.template
+++ b/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_listener_event_description.cpp
@@ -8,7 +8,7 @@
 #include <string>
 #include <thread>
 
-%include_SB_APIs%
+#include "lldb-headers.h"
 
 #include "common.h"
 
diff --git a/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_listener_event_process_state.cpp.template b/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_listener_event_process_state.cpp
index 2926ece4d8d..574257e98ee 100644
--- a/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_listener_event_process_state.cpp.template
+++ b/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_listener_event_process_state.cpp
@@ -8,7 +8,7 @@
 #include <string>
 #include <thread>
 
-%include_SB_APIs%
+#include "lldb-headers.h"
 
 #include "common.h"
 
diff --git a/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_listener_resume.cpp.template b/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_listener_resume.cpp
index 4adc9b33887..8cf786b1160 100644
--- a/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_listener_resume.cpp.template
+++ b/lldb/packages/Python/lldbsuite/test/api/multithreaded/test_listener_resume.cpp
@@ -8,7 +8,7 @@
 #include <string>
 #include <thread>
 
-%include_SB_APIs%
+#include "lldb-headers.h"
 
 #include "common.h"