adding sanity check for cluster edition

examples/cluster.py

@@ -129,7 +129,7 @@ def __init__( self, name, server=None, user=None, serverIP=None,
                 serverIP = self.findServerIP( server )
         self.serverIP = serverIP
         if not user:
-            user = quietRun( 'who am i' ).split()[ 0 ]
+            user = quietRun( 'echo $SUDO_USER' )
         self.user = user
         if self.user and self.server:
             self.dest = '%s@%s' % ( self.user, self.serverIP )
@@ -595,7 +595,7 @@ def __init__( self, *args, **kwargs ):
                               for server in self.servers }
         self.user = params.pop( 'user', None )
         if self.servers and not self.user:
-            self.user = quietRun( 'who am i' ).split()[ 0 ]
+            self.user = quietRun( 'echo $SUDO_USER' )
         if params.pop( 'precheck' ):
             self.precheck()
         self.connections = {}

examples/clusterSanity.py

+#!/usr/bin/env python
+
+'''
+A sanity check for cluster edition
+'''
+
+from mininet.examples.cluster import MininetCluster
+from mininet.log import info, setLogLevel
+from mininet.examples.clustercli import DemoCLI as CLI
+from mininet.topo import SingleSwitchTopo
+
+def clusterSanity():
+    "Sanity check for cluster mode"
+    topo = SingleSwitchTopo()
+    net = MininetCluster( topo=topo ) 
+    net.start()
+    CLI( net )
+    net.stop()
+
+if __name__  == '__main__':
+    setLogLevel( 'info' )
+    clusterSanity()

examples/test/test_clusterSanity.py

+#!/usr/bin/env python
+
+'''
+A simple sanity check test for cluster edition
+'''
+
+import unittest
+import pexpect
+
+class clusterSanityCheck( unittest.TestCase ):
+
+    prompt = 'mininet>'
+
+    def testClusterPingAll( self ):
+        p = pexpect.spawn( 'python -m mininet.examples.clusterSanity' )
+        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()
+
+
+if __name__  == '__main__':
+    unittest.main()