Brian O'Connor
49 lines
1.6 KiB
Python
Executable File
49 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
"""
|
|
Test for multiping.py
|
|
"""
|
|
|
|
import unittest
|
|
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 )
|
|
else:
|
|
self.assertEqual( received, 1 )
|
|
try:
|
|
pings[ name ].remove( target )
|
|
except:
|
|
pass
|
|
else:
|
|
break
|
|
self.assertTrue( len( pings ) > 0 )
|
|
for t in pings.values():
|
|
self.assertEqual( len( t ), 0 )
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|