mininet/examples/test/test_linuxrouter.py

   1 #!/usr/bin/env python
   2
   3 """
   4 Test for linuxrouter.py
   5 """
   6
   7 import unittest
   8 from mininet.util import pexpect
   9 from mininet.util import quietRun
  10
  11 class testLinuxRouter( unittest.TestCase ):
  12
  13     prompt = 'mininet>'
  14
  15     def testPingall( self ):
  16         "Test connectivity between hosts"
  17         p = pexpect.spawn( 'python -m mininet.examples.linuxrouter' )
  18         p.expect( self.prompt )
  19         p.sendline( 'pingall' )
  20         p.expect ( '(\d+)% dropped' )
  21         percent = int( p.match.group( 1 ) ) if p.match else -1
  22         p.expect( self.prompt )
  23         p.sendline( 'exit' )
  24         p.wait()
  25         self.assertEqual( percent, 0 )
  26
  27     def testRouterPing( self ):
  28         "Test connectivity from h1 to router"
  29         p = pexpect.spawn( 'python -m mininet.examples.linuxrouter' )
  30         p.expect( self.prompt )
  31         p.sendline( 'h1 ping -c 1 r0' )
  32         p.expect ( '(\d+)% packet loss' )
  33         percent = int( p.match.group( 1 ) ) if p.match else -1
  34         p.expect( self.prompt )
  35         p.sendline( 'exit' )
  36         p.wait()
  37         self.assertEqual( percent, 0 )
  38
  39     def testTTL( self ):
  40         "Verify that the TTL is decremented"
  41         p = pexpect.spawn( 'python -m mininet.examples.linuxrouter' )
  42         p.expect( self.prompt )
  43         p.sendline( 'h1 ping -c 1 h2' )
  44         p.expect ( 'ttl=(\d+)' )
  45         ttl = int( p.match.group( 1 ) ) if p.match else -1
  46         p.expect( self.prompt )
  47         p.sendline( 'exit' )
  48         p.wait()
  49         self.assertEqual( ttl, 63 ) # 64 - 1
  50
  51 if __name__ == '__main__':
  52     unittest.main()