adding first draft of tests for all examples, they need comments and clean up, some could be made more rebust

2013-09-11 14:45:46 -07:00
parent 5a646a0d20
commit a46fae0687
20 changed files with 914 additions and 0 deletions
@@ -0,0 +1,113 @@
+#!/usr/bin/env python
+
+"""TEST"""
+
+import unittest
+import pexpect
+import re
+from time import sleep
+from mininet.log import setLogLevel
+from mininet.net import Mininet
+from mininet.node import Node
+from mininet.link import Link, Intf
+
+class testHwintf( unittest.TestCase ):
+    "Test ping with single switch topology (common code)."
+
+    prompt = 'mininet>'
+
+    def _testE2E( self ):
+        results = [ "Results:", pexpect.EOF, pexpect.TIMEOUT ]
+        p = pexpect.spawn( 'python -m mininet.examples.simpleperf' )
+        index = p.expect( results, timeout=600 )
+        self.assertEqual( index, 0 )
+        p.wait()
+
+    def setUp( self ):
+        self.h3 = Node( 't0', ip='10.0.0.3/8' )
+        self.n0 = Node( 't1', inNamespace=False)
+        Link( self.h3, self.n0 )
+        self.h3.configDefault()
+
+    def testLocalPing( self ):
+        p = pexpect.spawn( 'python -m mininet.examples.hwintf %s' % self.n0.intf() )
+        p.expect( self.prompt )
+        p.sendline( 'pingall' )
+        p.expect ( '(\d+)% dropped' )
+        percent = int( p.match.group( 1 ) ) if p.match else -1
+        self.assertEqual( percent, 0 )
+        p.expect( self.prompt )
+        p.sendline( 'exit' )
+        p.wait()
+
+    def testExternalPing( self ):
+        expectStr = '(\d+) packets transmitted, (\d+) received'
+        p = pexpect.spawn( 'python -m mininet.examples.hwintf %s' % self.n0.intf() )
+        p.expect( self.prompt )
+
+        m = re.search( expectStr, self.h3.cmd( 'ping -v -c 1 10.0.0.1' ) )
+        tx = m.group( 1 )
+        rx = m.group( 2 )
+        self.assertEqual( tx, rx )
+
+        p.sendline( 'h1 ping -c 1 10.0.0.3')
+        p.expect( expectStr )
+        tx = p.match.group( 1 )
+        rx = p.match.group( 2 )
+        self.assertEqual( tx, rx )
+        p.expect( self.prompt )
+        p.sendline( 'exit' )
+        p.wait()
+
+    def tearDown( self ):
+        self.h3.terminate()
+        self.n0.terminate()
+
+
+            
+    ''' TAP garbage 
+     def testHwintf( self ):
+        ifname = 'br3'
+        #sudo ip tuntap add mode tap br0
+        #sudo ip tuntap del mode tap br0
+        #sudo ip link add name test0 type veth peer name test1
+        #sudo ip link del test0
+        t0 = Node( 't0', inNamespace=False, ip='10.0.0.3/8' )
+
+        t1 = Node( 't1', inNamespace=False)
+        #t0.cmd( 'ip tuntap add mode tap %s' % ifname )
+        #Intf( ifname, t0 )
+        print Link( t0, t1 )
+        t0.configDefault()
+
+        print t0.cmd( 'ifconfig' )
+        ifname =  t1.intf()
+
+
+        try:
+            foo = pexpect.spawn( 'wireshark' )
+            p = pexpect.spawn( 'python -m mininet.examples.hwintf %s' % ifname )
+            p.expect( self.prompt )
+            #t0.cmd( 'ip link set dev %s up' % ifname )
+            #t0.cmd( "bash -c 'echo 1 > /proc/sys/net/ipv4/ip_forward'" )
+            #t0.cmd( "bash -c 'echo 1 > /proc/sys/net/ipv4/conf/%s/proxy_arp'" % ifname)
+            #t0.cmd( 'arp -Ds 10.0.0.3 s1 pub' )
+
+            #p.sendline( 'x s1 wireshark' )
+            print t0.cmd( 'ifconfig %s' % ifname )
+            print t0.cmd( 'ip route' )
+            print t0.cmd( 'ping -v -c 1 10.0.0.3' )
+            print t0.cmd( 'ping -v -c 1 10.0.0.1' )
+
+            p.interact()
+            #p.wait()
+        finally:
+            #t0.cmd( 'ip tuntap del mode tap %s' % ifname )
+            t0.terminate()
+            t1.terminate()
+        #t0.configDefault()
+    '''
+
+if __name__ == '__main__':
+    setLogLevel( 'warning' )
+    unittest.main()