# # Tests the functionality of the customPrompt plugin # To run this test on your own shell, simply run: # # /web/courses/cs3214/spring2015/projects/student-plugins/banth94_crutcher/customPrompt.py eshoutput.py # # from the directory in which your "esh" and your "eshoutput.py" is located. # It should also be possible to make a copy of that file anywhere, as long as the # 'customPrompt.so' file is located within the same directory. # # NOTE: This conflicts with the given 'prompt.c' for full effect, disable that plugin # # @author Anthony Barbee (banth94) # @author Ian Crutcher (crutcher) # # # Setup Section # import sys, imp, atexit, os sys.path.append("/home/courses/cs3214/software/pexpect-dpty/"); import pexpect, shellio, signal, time, os, re, proc_check # Determine the path this file is in thisdir = os.path.dirname(os.path.realpath(__file__)) #Ensure the shell process is terminated def force_shell_termination(shell_process): c.close(force=True) # pulling in the regular expression and other definitions # this should be the eshoutput.py file of the hosting shell, see usage above definitions_scriptname = sys.argv[1] def_module = imp.load_source('', definitions_scriptname) # you can define logfile=open("log.txt", "w") in your eshoutput.py if you want logging! logfile = None if hasattr(def_module, 'logfile'): logfile = def_module.logfile #spawn an instance of the shell, note the -p flags c = pexpect.spawn(def_module.shell, drainpty=True, logfile=logfile, args=['-p', thisdir]) atexit.register(force_shell_termination, shell_process=c) # set timeout for all following 'expect*' calls to 2 seconds c.timeout = 2 # # Testing Section # # ensure that shell prints expected prompt assert c.expect(def_module.prompt) == 0, "Shell did not print expected prompt (1)" # Run plugin with no specified arguments c.sendline("prompt") assert c.expect('Write your desired prompt line: ') == 0, \ "expected output %s" % ("Write your desired prompt line: ") # Send something to specify the desired prompt line c.sendline("TEST") assert c.expect('Your new prompt line is TEST> ') == 0, \ "expected output %s" % ("Your new prompt line is TEST> ") # Run plugin with arguments specified c.sendline("prompt new test") assert c.expect_exact('Your new prompt line is new test> ') == 0, \ "expected output %s" % ('Your new prompt line is new test> ') shellio.success()