Imported Upstream version 1.5.1

tests/test_scripts/02_erichs_ddwrite.py

@@ -0,0 +1,63 @@
+#! /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"
+