mininet/examples/test/runner.py

   1 #!/usr/bin/env python
   2
   3 """
   4 Run all mininet.examples tests
   5  -v : verbose output
   6  -quick : skip tests that take more than ~30 seconds
   7 """
   8
   9 import unittest
  10 import os
  11 import sys
  12 from mininet.util import ensureRoot
  13 from mininet.clean import cleanup
  14
  15 class MininetTestResult( unittest.TextTestResult ):
  16     def addFailure( self, test, err ):
  17         super( MininetTestResult, self ).addFailure( test, err )
  18         cleanup()
  19     def addError( self,test, err ):
  20         super( MininetTestResult, self ).addError( test, err )
  21         cleanup()
  22
  23 class MininetTestRunner( unittest.TextTestRunner ):
  24     def _makeResult( self ):
  25         return MininetTestResult( self.stream, self.descriptions, self.verbosity )
  26
  27 def runTests( testDir, verbosity=1 ):
  28     "discover and run all tests in testDir"
  29     # ensure root and cleanup before starting tests
  30     ensureRoot()
  31     cleanup()
  32     # discover all tests in testDir
  33     testSuite = unittest.defaultTestLoader.discover( testDir )
  34     # run tests
  35     success = MininetTestRunner( verbosity=verbosity ).run( testSuite ).wasSuccessful()
  36     sys.exit( 0 if success else 1 )
  37
  38 if __name__ == '__main__':
  39     # get the directory containing example tests
  40     testDir = os.path.dirname( os.path.realpath( __file__ ) )
  41     verbosity = 2 if '-v' in sys.argv else 1
  42     runTests( testDir, verbosity )