Fixed MininetFacade and moved main logic into run. First shot at "test" function

examples/controlnet.py

@@ -30,18 +30,24 @@ class DataController( Controller ):
         "Ignore spurious error"
         pass
 
-
+class MininetFacade( object ):
-class MininetFacade:
     "TODO: CLI that can talk to two or more networks"
     
-    def __init__( self, *args ):
+    def __init__( self, net, *args, **kwargs ):
-        self.nets = args
+        self.net = net
+        self.nets = [ net ] + list( args ) + kwargs.values()
+        self.nameToNet = kwargs
+        self.nameToNet['net'] = net
 
     # default is first net
     def __getattr__( self, name ):
-        return getattr( self.nets[ 0 ], name )
+        return getattr( self.net, name )
 
     def __getitem__( self, key ):
+        #search kwargs for net named key
+        if key in self.nameToNet:
+            return self.nameToNet[ key ]
+        #search each net for node named key
         for net in self.nets:
             if key in net:
                 return net[ key ]
@@ -91,9 +97,7 @@ class ControlNetwork( Topo ):
 
 
 # Make it Happen!!
-
+def run( func=CLI ):
-setLogLevel( 'info' )
-
     info( '* Creating Control Network\n' )
     ctopo = ControlNetwork( n=4, dataController=DataController )
     cnet = Mininet( topo=ctopo, ipBase='192.168.123.0/24', controller=None )
@@ -113,8 +117,10 @@ for host in cnet.hosts:
     info( '* Starting Data Network\n')
     net.start()
 
-mn = MininetFacade( net, cnet )
+    mn = MininetFacade( net, cnet=cnet )
-CLI( mn )
+
+    # run the function passed as an argument
+    func( mn )
 
     info( '* Stopping Data Network\n' )
     net.stop()
@@ -124,8 +130,19 @@ info( '* Stopping Control Network\n' )
     #cnet.hosts = list( set( cnet.hosts ) - set( dataControllers ) )
     cnet.stop()
 
+def test( net ):
+    netLoss = net.pingAll()
+    cnetLoss = net['cnet'].pingAll()
 
+if __name__ == '__main__':
+    setLogLevel( 'info' )
 
+    import argparse
+    parser = argparse.ArgumentParser(description='TODO:description')
+    parser.add_argument('--test', dest='func', action='store_const',
+                        const=test, default=CLI,
+                        help='TODO: test help')
 
+    args = parser.parse_args()
 
-
+    run( func=args.func )