Brian O'Connor
58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
#!/usr/bin/env python
|
|
|
|
"""
|
|
Test for natnet.py
|
|
"""
|
|
|
|
import unittest
|
|
import pexpect
|
|
from mininet.util import quietRun
|
|
|
|
class testNATNet( unittest.TestCase ):
|
|
|
|
prompt = 'mininet>'
|
|
|
|
def setUp( self ):
|
|
self.net = pexpect.spawn( 'python -m mininet.examples.natnet' )
|
|
self.net.expect( self.prompt )
|
|
|
|
def testPublicPing( self ):
|
|
"Attempt to ping the public server (h0) from h1 and h2"
|
|
self.net.sendline( 'h1 ping -c 1 h0' )
|
|
self.net.expect ( '(\d+)% packet loss' )
|
|
percent = int( self.net.match.group( 1 ) ) if self.net.match else -1
|
|
self.assertEqual( percent, 0 )
|
|
self.net.expect( self.prompt )
|
|
|
|
self.net.sendline( 'h2 ping -c 1 h0' )
|
|
self.net.expect ( '(\d+)% packet loss' )
|
|
percent = int( self.net.match.group( 1 ) ) if self.net.match else -1
|
|
self.assertEqual( percent, 0 )
|
|
self.net.expect( self.prompt )
|
|
|
|
def testPrivatePing( self ):
|
|
"Attempt to ping h1 and h2 from public server"
|
|
self.net.sendline( 'h0 ping -c 1 -t 1 h1' )
|
|
result = self.net.expect ( [ 'unreachable', 'loss' ] )
|
|
self.assertEqual( result, 0 )
|
|
self.net.expect( self.prompt )
|
|
|
|
self.net.sendline( 'h0 ping -c 1 -t 1 h2' )
|
|
result = self.net.expect ( [ 'unreachable', 'loss' ] )
|
|
self.assertEqual( result, 0 )
|
|
self.net.expect( self.prompt )
|
|
|
|
def testPrivateToPrivatePing( self ):
|
|
"Attempt to ping from NAT'ed host h1 to NAT'ed host h2"
|
|
self.net.sendline( 'h1 ping -c 1 -t 1 h2' )
|
|
result = self.net.expect ( [ '[Uu]nreachable', 'loss' ] )
|
|
self.assertEqual( result, 0 )
|
|
self.net.expect( self.prompt )
|
|
|
|
def tearDown( self ):
|
|
self.net.sendline( 'exit' )
|
|
self.net.wait()
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|