Bob Lantz
e37afe996b
Mostly we just use mininet.util.pexpect so that we can use unicode strings without changing calls to pexpect.spawn()
51 lines
1.8 KiB
Python
Executable File
51 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
"""
|
|
Test for multiping.py
|
|
"""
|
|
|
|
import unittest
|
|
from mininet.util import pexpect
|
|
from collections import defaultdict
|
|
|
|
class testMultiPing( unittest.TestCase ):
|
|
|
|
def testMultiPing( self ):
|
|
"""Verify that each target is pinged at least once, and
|
|
that pings to 'real' targets are successful and unknown targets fail"""
|
|
p = pexpect.spawn( 'python -m mininet.examples.multiping' )
|
|
opts = [ "Host (h\d+) \(([\d.]+)\) will be pinging ips: ([\d\. ]+)",
|
|
"(h\d+): ([\d.]+) -> ([\d.]+) \d packets transmitted, (\d) received",
|
|
pexpect.EOF ]
|
|
pings = defaultdict( list )
|
|
while True:
|
|
index = p.expect( opts )
|
|
if index == 0:
|
|
name = p.match.group(1)
|
|
ip = p.match.group(2)
|
|
targets = p.match.group(3).split()
|
|
pings[ name ] += targets
|
|
elif index == 1:
|
|
name = p.match.group(1)
|
|
ip = p.match.group(2)
|
|
target = p.match.group(3)
|
|
received = int( p.match.group(4) )
|
|
if target == '10.0.0.200':
|
|
self.assertEqual( received, 0, p.match.group(0) + '\n' +
|
|
target + ' received %d != 0 packets' % received )
|
|
else:
|
|
self.assertEqual( received, 1, p.match.group(0) + '\n' +
|
|
target + ' received %d != 1 packets' % received )
|
|
try:
|
|
pings[ name ].remove( target )
|
|
except:
|
|
pass
|
|
else:
|
|
break
|
|
self.assertTrue( len( pings ) > 0, 'too few pings' )
|
|
for t in pings.values():
|
|
self.assertEqual( len( t ), 0, 'missed ping target(s): %s' % t )
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|