summaryrefslogtreecommitdiff
path: root/tests/pexpectTest.py
diff options
context:
space:
mode:
authorNoah Spurrier <noah@squaretrade.com>2012-10-26 11:19:10 -0700
committerNoah Spurrier <noah@squaretrade.com>2012-10-26 11:19:10 -0700
commit7999ca657997e78febfb3fb89cfcc066d50bf788 (patch)
tree7ff33465bb4f8f79b92add505d11d4b731dfe6a7 /tests/pexpectTest.py
parent6b65a76c26d72caf0a5b11725750861bf89f2b75 (diff)
downloadpexpect-7999ca657997e78febfb3fb89cfcc066d50bf788.tar.gz
Moved everything up one directory level.
Diffstat (limited to 'tests/pexpectTest.py')
-rwxr-xr-xtests/pexpectTest.py67
1 files changed, 67 insertions, 0 deletions
diff --git a/tests/pexpectTest.py b/tests/pexpectTest.py
new file mode 100755
index 0000000..62ac422
--- /dev/null
+++ b/tests/pexpectTest.py
@@ -0,0 +1,67 @@
+#!/usr/bin/env python
+'''
+PEXPECT LICENSE
+
+ This license is approved by the OSI and FSF as GPL-compatible.
+ http://opensource.org/licenses/isc-license.txt
+
+ Copyright (c) 2012, Noah Spurrier <noah@noah.org>
+ PERMISSION TO USE, COPY, MODIFY, AND/OR DISTRIBUTE THIS SOFTWARE FOR ANY
+ PURPOSE WITH OR WITHOUT FEE IS HEREBY GRANTED, PROVIDED THAT THE ABOVE
+ COPYRIGHT NOTICE AND THIS PERMISSION NOTICE APPEAR IN ALL COPIES.
+ THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+
+'''
+
+import os, time, pexpect, sys
+
+def getProcessResults(cmd, timeLimit=20):
+ '''
+ executes 'cmd' as a child process and returns the child's output,
+ the duration of execution, and the process exit status. Aborts if
+ child process does not generate output for 'timeLimit' seconds.
+ '''
+ output = ""
+ startTime = time.time()
+ child = pexpect.spawn(cmd, timeout=10)
+ child.logfile = sys.stdout
+
+ while 1:
+ try:
+ # read_nonblocking will add to 'outout' one byte at a time
+ # newlines can show up as '\r\n' so we kill any '\r's which
+ # will mess up the formatting for the viewer
+ output += child.read_nonblocking(timeout=timeLimit).replace("\r","")
+ except pexpect.EOF, e:
+ print str(e)
+ # process terminated normally
+ break
+ except pexpect.TIMEOUT, e:
+ print str(e)
+ output += "\nProcess aborted by FlashTest after %s seconds.\n" % timeLimit
+ print child.isalive()
+ child.kill(9)
+ break
+
+ endTime = time.time()
+ child.close(force=True)
+
+ duration = endTime - startTime
+ exitStatus = child.exitstatus
+
+ return (output, duration, exitStatus)
+
+cmd = "./ticker.py"
+
+result, duration, exitStatus = getProcessResults(cmd)
+
+print "result: %s" % result
+print "duration: %s" % duration
+print "exit-status: %s" % exitStatus
+