Adding basic test for thread state (See PR-15824)

llvm-svn: 180149
diff --git a/lldb/test/functionalities/thread/state/TestThreadStates.py b/lldb/test/functionalities/thread/state/TestThreadStates.py
new file mode 100644
index 0000000..13b7bc1
--- /dev/null
+++ b/lldb/test/functionalities/thread/state/TestThreadStates.py
@@ -0,0 +1,140 @@
+"""
+Test thread states.
+"""
+
+import os, time
+import unittest2
+import lldb
+from lldbtest import *
+import lldbutil
+
+class StopThreadsTestCase(TestBase):
+
+    mydir = os.path.join("functionalities", "thread", "state")
+
+    @unittest2.skipUnless(sys.platform.startswith("darwin"), "requires Darwin")
+    @dsym_test
+    @expectedFailure("PR-15824") # thread states not properly maintained
+    def test_with_dsym(self):
+        """Test thread states."""
+        self.buildDsym()
+        self.thread_states_test()
+
+    @dwarf_test
+    @expectedFailure("PR-15824") # thread states not properly maintained
+    def test_with_dwarf(self):
+        """Test thread states."""
+        self.buildDwarf()
+        self.thread_states_test()
+
+    def setUp(self):
+        # Call super's setUp().
+        TestBase.setUp(self)
+        # Find the line numbers for our breakpoints.
+        self.break_1 = line_number('main.c', '// Set first breakpoint here')
+        self.break_2 = line_number('main.c', '// Set second breakpoint here')
+
+    def thread_states_test(self):
+        """Test thread states."""
+        exe = os.path.join(os.getcwd(), "a.out")
+        self.runCmd("file " + exe, CURRENT_EXECUTABLE_SET)
+
+        # This should create a breakpoint in the main thread.
+        lldbutil.run_break_set_by_file_and_line (self, "main.c", self.break_1, num_expected_locations=1)
+        lldbutil.run_break_set_by_file_and_line (self, "main.c", self.break_2, num_expected_locations=1)
+
+        # The breakpoint list should show 2 breakpoints with 1 location each.
+        self.expect("breakpoint list -f", "Breakpoint location shown correctly",
+            substrs = ["1: file ='main.c', line = %d, locations = 1" % self.break_1,
+                       "2: file ='main.c', line = %d, locations = 1" % self.break_2])
+
+        # Run the program.
+        self.runCmd("run", RUN_SUCCEEDED)
+
+        # The stop reason of the thread should be breakpoint.
+        self.expect("thread list", STOPPED_DUE_TO_BREAKPOINT,
+            substrs = ['stopped',
+                       '* thread #1',
+                       'stop reason = breakpoint'])
+
+        # Get the target process
+        target = self.dbg.GetSelectedTarget()
+        process = target.GetProcess()
+
+        # Get the number of threads
+        num_threads = process.GetNumThreads()
+
+        self.assertTrue(num_threads == 1, 'Number of expected threads and actual threads do not match.')
+
+        # Get the thread object
+        thread = process.GetThreadAtIndex(0)
+
+        # Make sure the thread is in the stopped state.
+        self.assertTrue(thread.IsStopped(), "Thread state isn't \'stopped\' during breakpoint 1.")
+        self.assertFalse(thread.IsSuspended(), "Thread state is \'suspended\' during breakpoint 1.")
+
+        # Continue, the inferior will go into an infinite loop waiting for 'g_test' to change.
+        self.dbg.SetAsync(True)
+        self.runCmd("continue")
+        time.sleep(1)
+
+        # Check the thread state. It should be running.
+        self.assertFalse(thread.IsStopped(), "Thread state is \'stopped\' when it should be running.")
+        self.assertFalse(thread.IsSuspended(), "Thread state is \'suspended\' when it should be running.")
+
+        # Suspend the thread
+        thread.Suspend()
+
+        # Make sure the thread is in the suspended state (which means it is also stopped).
+        self.assertTrue(thread.IsSuspended(), "Thread state isn't \'suspended\' during suspend.")
+        self.assertTrue(thread.IsStopped(), "Thread state isn't \'stopped\' during suspend.")
+
+        # Resume the thread
+        thread.Resume()
+
+        # Check the thread state. It should be running.
+        self.assertFalse(thread.IsStopped(), "Thread state is \'stopped\' after thread resume.")
+        self.assertFalse(thread.IsSuspended(), "Thread state is \'suspended\' after thread resume.")
+
+        # Go back to synchronous interactions
+        self.dbg.SetAsync(False)
+
+        # Stop the process
+        self.runCmd("process interrupt")
+
+        # The stop reason of the thread should be signal.
+        self.expect("process status", STOPPED_DUE_TO_SIGNAL,
+            substrs = ['stopped',
+                       '* thread #1',
+                       'stop reason = signal'])
+
+
+        # Check the thread state
+        self.assertTrue(thread.IsStopped(), "Thread state isn't \'stopped\' after process stop.")
+        self.assertFalse(thread.IsSuspended(), "Thread state is \'suspended\' after process stop.")
+
+        # Get the inferior out of its loop
+        self.runCmd("expression g_test = 1")
+
+        # Check the thread state
+        self.assertTrue(thread.IsStopped(), "Thread state isn't \'stopped\' after expression evaluation.")
+        self.assertFalse(thread.IsSuspended(), "Thread state is \'suspended\' after expression evaluation.")
+
+        # Run to breakpoint 2
+        self.runCmd("continue")
+
+        # Make sure both threads are stopped
+        self.assertTrue(thread.IsStopped(), "Thread state isn't \'stopped\' during breakpoint 2.")
+        self.assertFalse(thread.IsSuspended(), "Thread state is \'suspended\' during breakpoint 2.")
+
+        # Run to completion
+        self.runCmd("continue")
+
+        # At this point, the inferior process should have exited.
+        self.assertTrue(process.GetState() == lldb.eStateExited, PROCESS_EXITED)
+
+if __name__ == '__main__':
+    import atexit
+    lldb.SBDebugger.Initialize()
+    atexit.register(lambda: lldb.SBDebugger.Terminate())
+    unittest2.main()