#!/usr/bin/python # # Tests the functionality of hello world plugin # # Written for CS 3214 Spring 2015. # # To run this test on your own shell, simply run: # # # @author Lance Chao, Alice Wang # 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 = 5 ############################################################################# # Now the real test starts! # ################################################################# # Step 1. Run hello java # c.sendline("hello java") assert c.expect("\npublic class HelloWorld{") == 0, "Incorrect output" ################################################################# # Step 2. Run hello c++ # c.sendline("hello c++") assert c.expect("\n#include ") == 0, "Incorrect output" ################################################################# # Step 3. Run hello c # c.sendline("hello c") assert c.expect("\n#include ") == 0, "Incorrect output" ################################################################# # Step 4. Run hello python # c.sendline("hello python") assert c.expect("\nprint") == 0, "Incorrect output" shellio.success()