Bob Lantz
e37afe996b
Mostly we just use mininet.util.pexpect so that we can use unicode strings without changing calls to pexpect.spawn()
46 lines
1.3 KiB
Python
Executable File
46 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
"""
|
|
Test for popen.py and popenpoll.py
|
|
"""
|
|
|
|
import unittest
|
|
from mininet.util import pexpect
|
|
|
|
class testPopen( unittest.TestCase ):
|
|
|
|
def pingTest( self, name ):
|
|
"Verify that there are no dropped packets for each host"
|
|
p = pexpect.spawn( 'python -m %s' % name )
|
|
opts = [ "<(h\d+)>: PING ",
|
|
"<(h\d+)>: (\d+) packets transmitted, (\d+) received",
|
|
pexpect.EOF ]
|
|
pings = {}
|
|
while True:
|
|
index = p.expect( opts )
|
|
if index == 0:
|
|
name = p.match.group(1)
|
|
pings[ name ] = 0
|
|
elif index == 1:
|
|
name = p.match.group(1)
|
|
transmitted = p.match.group(2)
|
|
received = p.match.group(3)
|
|
# verify no dropped packets
|
|
self.assertEqual( received, transmitted )
|
|
pings[ name ] += 1
|
|
else:
|
|
break
|
|
self.assertTrue( len(pings) > 0 )
|
|
# verify that each host has gotten results
|
|
for count in pings.values():
|
|
self.assertEqual( count, 1 )
|
|
|
|
def testPopen( self ):
|
|
self.pingTest( 'mininet.examples.popen' )
|
|
|
|
def testPopenPoll( self ):
|
|
self.pingTest( 'mininet.examples.popenpoll' )
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|