From 01c0ef001332fb89892cbbde573dca195f3380d3 Mon Sep 17 00:00:00 2001
From: Brian O'Connor <bocon13@gmail.com>
Date: Tue, 10 Sep 2013 14:07:46 -0700
Subject: [PATCH] added comments to test_baresshd.py

---
 examples/test/test_baresshd.py | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/examples/test/test_baresshd.py b/examples/test/test_baresshd.py
index d57436eb..cec2c812 100755
--- a/examples/test/test_baresshd.py
+++ b/examples/test/test_baresshd.py
@@ -1,6 +1,8 @@
 #!/usr/bin/env python
 
-"""TEST"""
+"""
+Tests for baresshd.py
+"""
 
 import unittest
 import pexpect
@@ -9,13 +11,12 @@
 from mininet.clean import cleanup, sh
 
 class testBareSSHD( unittest.TestCase ):
-    "Test ping with single switch topology (common code)."
 
     opts = [ '\(yes/no\)\?', 'Welcome to h1', 'refused', pexpect.EOF, pexpect.TIMEOUT ]
 
     def connected( self ):
-        "check connected"
-        p = pexpect.spawn( 'ssh 10.0.0.1 -i /tmp/ssh/test_rsa ' )
+        "Log into ssh server, check banner, then exit"
+        p = pexpect.spawn( 'ssh 10.0.0.1 -i /tmp/ssh/test_rsa exit' )
         while True:
             index = p.expect( self.opts )
             if index == 0:
@@ -26,19 +27,22 @@ def connected( self ):
                 return False
 
     def setUp( self ):
+        # verify that sshd is not running
         self.assertFalse( self.connected() )
         # create public key pair for testing
         sh( 'mkdir /tmp/ssh' )
         sh( "ssh-keygen -t rsa -P '' -f /tmp/ssh/test_rsa" )
         sh( 'cat /tmp/ssh/test_rsa.pub >> /tmp/ssh/authorized_keys' )
+        # run example with custom sshd args
         cmd = ( 'python -m mininet.examples.baresshd '
                 '-o AuthorizedKeysFile=/tmp/ssh/authorized_keys '
                 '-o StrictModes=no' )
         sh( cmd )
 
     def testSSH( self ):
+        "Simple test to verify that we can ssh into h1"
         result = False
-        # try to connect up to 3 times
+        # try to connect up to 3 times; sshd can take a while to start
         for _ in range( 3 ):
             result = self.connected()
             if result:
@@ -54,7 +58,6 @@ def tearDown( self ):
         # remove public key pair
         sh( 'rm -rf /tmp/ssh' )
 
-
 if __name__ == '__main__':
     setLogLevel( 'warning' )
     unittest.main()
-- 
GitLab