34 lines
864 B
Python
34 lines
864 B
Python
|
import os
|
||
|
import re
|
||
|
import shutil
|
||
|
|
||
|
def checkOutput(output):
|
||
|
"""
|
||
|
Ensure hello world is in string.
|
||
|
return: whether hello world is present
|
||
|
"""
|
||
|
return output == 'Hello world!\n'
|
||
|
|
||
|
def runCommand(command):
|
||
|
"""
|
||
|
Define a function to run a shell commmand and return the output
|
||
|
param command: command to run
|
||
|
return: Whether retcode is 0
|
||
|
"""
|
||
|
fh = os.popen(command, mode='r', buffering=-1)
|
||
|
output = fh.read()
|
||
|
retcode = fh.close()
|
||
|
assert retcode == None
|
||
|
|
||
|
|
||
|
def runCommandAndOutputCheck(command):
|
||
|
"""
|
||
|
Define a function to run a shell command and ensure output is correct
|
||
|
param command: command
|
||
|
return: Whether retcode is 0 && output matches test.
|
||
|
"""
|
||
|
fh = os.popen(command, mode='r', buffering=-1)
|
||
|
output = fh.read()
|
||
|
retcode = fh.close()
|
||
|
assert retcode == None and CheckOutput(output)
|