Remote ns-3 with ping working

[nepi.git] / test / resources / ns3 / ns3wrapper.py

diff --git a/test/resources/ns3/ns3wrapper.py b/test/resources/ns3/ns3wrapper.py
index 3843864..8765761 100755 (executable)
--- a/test/resources/ns3/ns3wrapper.py
+++ b/test/resources/ns3/ns3wrapper.py
@@ -32,7 +32,9 @@
 
 from nepi.resources.ns3.ns3wrapper import NS3Wrapper
 
+import StringIO
 import subprocess
+import sys
 import time
 import unittest
 
@@ -218,6 +220,7 @@ class NS3WrapperTest(unittest.TestCase):
         p.communicate()
 
     def test_start(self):
+        # Instantiate ns-3
         wrapper = NS3Wrapper()
 
         ### create 2  nodes
@@ -291,6 +294,8 @@ class NS3WrapperTest(unittest.TestCase):
         # wait until simulation is over
         wrapper.shutdown()
 
+        # TODO: Add assertions !!
+
     def test_runtime_attr_modify(self):
         wrapper = NS3Wrapper()
        
@@ -416,6 +421,8 @@ class NS3WrapperTest(unittest.TestCase):
 
         p = subprocess.Popen("rm /tmp/trace-p2p-*",  shell = True)
         p.communicate()
+        
+        # TODO: Add assertions !!
 
 if __name__ == '__main__':
     unittest.main()