mininet/examples/test/test_nat.py

   1 #!/usr/bin/env python
   2
   3 """
   4 Test for nat.py
   5 """
   6
   7 import unittest
   8 from mininet.util import pexpect
   9 from mininet.util import quietRun
  10
  11 destIP = '8.8.8.8' # Google DNS
  12
  13 class testNAT( unittest.TestCase ):
  14
  15     prompt = 'mininet>'
  16
  17     @unittest.skipIf( '0 received' in quietRun( 'ping -c 1 %s' % destIP ),
  18                       'Destination IP is not reachable' )
  19     def testNAT( self ):
  20         "Attempt to ping an IP on the Internet and verify 0% packet loss"
  21         p = pexpect.spawn( 'python -m mininet.examples.nat' )
  22         p.expect( self.prompt )
  23         p.sendline( 'h1 ping -c 1 %s' % destIP )
  24         p.expect ( '(\d+)% packet loss' )
  25         percent = int( p.match.group( 1 ) ) if p.match else -1
  26         p.expect( self.prompt )
  27         p.sendline( 'exit' )
  28         p.wait()
  29         self.assertEqual( percent, 0 )
  30
  31 if __name__ == '__main__':
  32     unittest.main()