64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
|
#! /usr/bin/python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
# Copyright (c) 2009-2011 by Bjoern Kolbeck, Minor Gordon, Zuse Institute Berlin
|
||
|
|
# Licensed under the BSD License, see LICENSE file for details.
|
||
|
|
|
||
|
|
import unittest, subprocess, time, os, sys
|
||
|
|
from glob import glob
|
||
|
|
|
||
|
|
|
||
|
|
class ErichsddwriteTest(unittest.TestCase):
|
||
|
|
def __init__( self, stdout=sys.stdout, stderr=sys.stderr, *args, **kwds ):
|
||
|
|
unittest.TestCase.__init__( self )
|
||
|
|
self.stdout = stdout
|
||
|
|
self.stderr = stderr
|
||
|
|
|
||
|
|
def setUp( self ):
|
||
|
|
self.client_processes = []
|
||
|
|
|
||
|
|
def tearDown( self ):
|
||
|
|
for client_process in self.client_processes:
|
||
|
|
client_process.terminate()
|
||
|
|
client_process.wait()
|
||
|
|
|
||
|
|
for file_name in glob( self.__class__.__name__ + "*" ):
|
||
|
|
os.unlink( file_name )
|
||
|
|
|
||
|
|
def runTest( self ):
|
||
|
|
class_name = self.__class__.__name__
|
||
|
|
for clients_count in xrange( 2, 4, 2 ):
|
||
|
|
for client_i in xrange( clients_count ):
|
||
|
|
args = "dd if=/dev/zero of=%(class_name)s_%(client_i)u bs=1MB count=10" % locals()
|
||
|
|
client_process = subprocess.Popen( args, shell=True, stdout=self.stdout, stderr=self.stderr )
|
||
|
|
self.client_processes.append( client_process )
|
||
|
|
|
||
|
|
while len( self.client_processes ) > 0:
|
||
|
|
client_i = 0
|
||
|
|
while client_i < len( self.client_processes ):
|
||
|
|
retcode = self.client_processes[client_i].poll()
|
||
|
|
if retcode is not None:
|
||
|
|
if retcode != 0:
|
||
|
|
self.fail()
|
||
|
|
else:
|
||
|
|
del self.client_processes[client_i]
|
||
|
|
else:
|
||
|
|
client_i ++ 1
|
||
|
|
|
||
|
|
time.sleep( 0.5 )
|
||
|
|
|
||
|
|
|
||
|
|
def createTestSuite( *args, **kwds ):
|
||
|
|
if not sys.platform.startswith( "win" ):
|
||
|
|
return unittest.TestSuite( [ErichsddwriteTest( *args, **kwds )] )
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
if not sys.platform.startswith( "win" ):
|
||
|
|
result = unittest.TextTestRunner( verbosity=2 ).run( createTestSuite() )
|
||
|
|
if not result.wasSuccessful():
|
||
|
|
sys.exit(1)
|
||
|
|
else:
|
||
|
|
print sys.modules[__name__].__file__.split( os.sep )[-1], "not supported on Windows"
|
||
|
|
|