q3k
/
khepri-tftpy
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Adding testcases for new file-like input and output

master
Michael P. Soulier 2012-09-30 22:07:20 -04:00
parent 792e849d20
commit b57e583798
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
t/test.py
View File

@ -141,12 +141,14 @@ class TestTftpyState(unittest.TestCase):
def setUp(self):
tftpy.setLogLevel(logging.DEBUG)
def clientServerUploadOptions(self, options, transmitname=None):
def clientServerUploadOptions(self, options, input=None, transmitname=None):
"""Fire up a client and a server and do an upload."""
root = '/tmp'
home = os.path.dirname(os.path.abspath(__file__))
filename = '100KBFILE'
input_path = os.path.join(home, filename)
if not input:
input = input_path
if transmitname:
filename = transmitname
server = tftpy.TftpServer(root)
@ -160,7 +162,7 @@ class TestTftpyState(unittest.TestCase):
try:
time.sleep(1)
client.upload(filename,
input_path)
input)
finally:
os.kill(child_pid, 15)
os.waitpid(child_pid, 0)
@ -168,7 +170,7 @@ class TestTftpyState(unittest.TestCase):
else:
server.listen('localhost', 20001)
def clientServerDownloadOptions(self, options):
def clientServerDownloadOptions(self, options, output='/tmp/out'):
"""Fire up a client and a server and do a download."""
root = os.path.dirname(os.path.abspath(__file__))
server = tftpy.TftpServer(root)
@ -182,7 +184,7 @@ class TestTftpyState(unittest.TestCase):
try:
time.sleep(1)
client.download('100KBFILE',
'/tmp/out')
output)
finally:
os.kill(child_pid, 15)
os.waitpid(child_pid, 0)
@ -193,6 +195,10 @@ class TestTftpyState(unittest.TestCase):
def testClientServerNoOptions(self):
self.clientServerDownloadOptions({})
def testClientFileObject(self):
output = open('/tmp/out', 'w')
self.clientServerDownloadOptions({}, output)
def testClientServerBlksize(self):
for blksize in [512, 1024, 2048, 4096]:
self.clientServerDownloadOptions({'blksize': blksize})
@ -200,6 +206,10 @@ class TestTftpyState(unittest.TestCase):
def testClientServerUploadNoOptions(self):
self.clientServerUploadOptions({})
def testClientServerUploadFileObj(self):
fileobj = open('/tmp/100KBFILE', 'r')
self.clientServerUploadOptions({}, input=fileobj)
def testClientServerUploadWithSubdirs(self):
self.clientServerUploadOptions({}, transmitname='foo/bar/100KBFILE')