Add tests for distributed sets and counters
diff --git a/TestON/drivers/common/cli/emulator/remotemininetdriver.py b/TestON/drivers/common/cli/emulator/remotemininetdriver.py
index 6e7d0a4..e95e5c1 100644
--- a/TestON/drivers/common/cli/emulator/remotemininetdriver.py
+++ b/TestON/drivers/common/cli/emulator/remotemininetdriver.py
@@ -403,8 +403,8 @@
# Close the ssh connection
self.handle.sendline( "" )
# self.handle.expect( "\$" )
- i = self.handle.expect( [ '\$', 'mininet>', pexpect.TIMEOUT ],
- timeout=2)
+ i = self.handle.expect( [ '\$', 'mininet>', pexpect.TIMEOUT,
+ pexpect.EOF ], timeout=2 )
if i == 0:
self.handle.sendline( "exit" )
self.handle.expect( "closed" )
@@ -415,7 +415,6 @@
self.handle.sendline( "exit" )
self.handle.expect( "exit" )
self.handle.expect( "closed" )
-
else:
main.log.error( "Connection failed to the host" )
return main.TRUE