4570d3eba0
This has led to many test suite failures because of copy and paste where new test cases were based off of other test cases and the "mydir" variable wasn't updated. Now you can call your superclasses "compute_mydir()" function with "__file__" as the sole argument and the relative path will be computed for you. llvm-svn: 196985
77 lines
3.1 KiB
Python
77 lines
3.1 KiB
Python
"""Test custom import command to import files by path."""
|
|
|
|
import os, sys, time
|
|
import unittest2
|
|
import lldb
|
|
from lldbtest import *
|
|
|
|
class ImportTestCase(TestBase):
|
|
|
|
mydir = TestBase.compute_mydir(__file__)
|
|
|
|
@python_api_test
|
|
def test_import_command(self):
|
|
"""Import some Python scripts by path and test them"""
|
|
self.run_test()
|
|
|
|
def setUp(self):
|
|
# Call super's setUp().
|
|
TestBase.setUp(self)
|
|
|
|
def run_test(self):
|
|
"""Import some Python scripts by path and test them."""
|
|
|
|
# This is the function to remove the custom commands in order to have a
|
|
# clean slate for the next test case.
|
|
def cleanup():
|
|
self.runCmd('command script delete foo2cmd', check=False)
|
|
self.runCmd('command script delete foocmd', check=False)
|
|
self.runCmd('command script delete foobarcmd', check=False)
|
|
self.runCmd('command script delete barcmd', check=False)
|
|
self.runCmd('command script delete barothercmd', check=False)
|
|
self.runCmd('command script delete TPcommandA', check=False)
|
|
self.runCmd('command script delete TPcommandB', check=False)
|
|
|
|
# Execute the cleanup function during test case tear down.
|
|
self.addTearDownHook(cleanup)
|
|
|
|
self.runCmd("command script import ./foo/foo.py --allow-reload")
|
|
self.runCmd("command script import ./foo/foo2.py --allow-reload")
|
|
self.runCmd("command script import ./foo/bar/foobar.py --allow-reload")
|
|
self.runCmd("command script import ./bar/bar.py --allow-reload")
|
|
|
|
self.expect("command script import ./nosuchfile.py",
|
|
error=True, startstr='error: module importing failed')
|
|
self.expect("command script import ./nosuchfolder/",
|
|
error=True, startstr='error: module importing failed')
|
|
self.expect("command script import ./foo/foo.py", error=False)
|
|
|
|
self.runCmd("command script import --allow-reload ./thepackage")
|
|
self.expect("TPcommandA",substrs=["hello world A"])
|
|
self.expect("TPcommandB",substrs=["hello world B"])
|
|
|
|
self.runCmd("script import dummymodule")
|
|
self.expect("command script import ./dummymodule.py", error=False)
|
|
self.expect("command script import --allow-reload ./dummymodule.py", error=False)
|
|
|
|
self.runCmd("command script add -f foo.foo_function foocmd")
|
|
self.runCmd("command script add -f foobar.foo_function foobarcmd")
|
|
self.runCmd("command script add -f bar.bar_function barcmd")
|
|
self.expect("foocmd hello",
|
|
substrs = ['foo says', 'hello'])
|
|
self.expect("foo2cmd hello",
|
|
substrs = ['foo2 says', 'hello'])
|
|
self.expect("barcmd hello",
|
|
substrs = ['barutil says', 'bar told me', 'hello'])
|
|
self.expect("barothercmd hello",
|
|
substrs = ['barutil says', 'bar told me', 'hello'])
|
|
self.expect("foobarcmd hello",
|
|
substrs = ['foobar says', 'hello'])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import atexit
|
|
lldb.SBDebugger.Initialize()
|
|
atexit.register(lambda: lldb.SBDebugger.Terminate())
|
|
unittest2.main()
|