"""
Tests that C++ templates work as expected
"""
import lldb
from lldbsuite.test.decorators import *
from lldbsuite.test.lldbtest import *
from lldbsuite.test import lldbutil

class TemplateIntegerArgsTestCase(TestBase):
    
    mydir = TestBase.compute_mydir(__file__)

    def test_with_run_command(self):
        """Test that C++ template classes that have integer parameters work correctly.
           
           We must reconstruct the types correctly so the template types are correct 
           and display correctly, and also make sure the expression parser works and
           is able the find all needed functions when evaluating expressions"""
        self.build()
        
        # Set debugger into synchronous mode
        self.dbg.SetAsync(False)

        # Create a target by the debugger.
        exe = os.path.join(os.getcwd(), "a.out")
        target = self.dbg.CreateTarget(exe)
        self.assertTrue(target, VALID_TARGET)

        # Set breakpoints inside and outside methods that take pointers to the containing struct.
        line = line_number('main.cpp', '// Breakpoint 1')
        lldbutil.run_break_set_by_file_and_line (self, "main.cpp", line, num_expected_locations=1, loc_exact=True)

        arguments = None
        environment = None 

        # Now launch the process, and do not stop at entry point.
        process = target.LaunchSimple (arguments, environment, self.get_process_working_directory())
        self.assertTrue(process, PROCESS_IS_VALID)
 
        # Get the thread of the process
        self.assertTrue(process.GetState() == lldb.eStateStopped, PROCESS_STOPPED)
        thread = lldbutil.get_stopped_thread(process, lldb.eStopReasonBreakpoint)

        # Get frame for current thread
        frame = thread.GetSelectedFrame()
        
        testpos = frame.FindVariable('testpos') 
        self.assertTrue(testpos.IsValid(), 'make sure we find a local variabble named "testpos"')
        self.assertTrue(testpos.GetType().GetName() == 'TestObj<1>')
        expr_result = frame.EvaluateExpression("testpos.getArg()")
        self.assertTrue(expr_result.IsValid(), 'got a valid expression result from expression "testpos.getArg()"');
        self.assertTrue(expr_result.GetValue() == "1", "testpos.getArg() == 1")
        self.assertTrue(expr_result.GetType().GetName() == "int", 'expr_result.GetType().GetName() == "int"')     
        
        testneg = frame.FindVariable('testneg') 
        self.assertTrue(testneg.IsValid(), 'make sure we find a local variabble named "testneg"')
        self.assertTrue(testneg.GetType().GetName() == 'TestObj<-1>')        
        
        expr_result = frame.EvaluateExpression("testneg.getArg()")
        self.assertTrue(expr_result.IsValid(), 'got a valid expression result from expression "testneg.getArg()"');
        self.assertTrue(expr_result.GetValue() == "-1", "testneg.getArg() == -1")
        self.assertTrue(expr_result.GetType().GetName() == "int", 'expr_result.GetType().GetName() == "int"')