Added attribute MaxAddresses for interface factories to design
[nepi.git] / test / testbeds / netns / execute.py
index 884a57d..e9591f3 100755 (executable)
@@ -2,53 +2,54 @@
 # -*- coding: utf-8 -*-
 
 import getpass
-from nepi.util.constants import AF_INET, STATUS_FINISHED
+from nepi.util.constants import ApplicationStatus as AS
 from nepi.testbeds import netns
 import os
 import shutil
+import tempfile
 import test_util
 import time
 import unittest
-import uuid
 
 class NetnsExecuteTestCase(unittest.TestCase):
     def setUp(self):
-        self._home_dir = os.path.join(os.getenv("HOME"), ".nepi", 
-                str(uuid.uuid1()))
-        os.makedirs(self._home_dir)
+        self.root_dir = tempfile.mkdtemp()
 
     @test_util.skipUnless(os.getuid() == 0, "Test requires root privileges")
     def test_run_ping_if(self):
         user = getpass.getuser()
         testbed_version = "01"
-        instance = netns.TestbedInstance(testbed_version)
-        instance.configure("homeDirectory", self._home_dir)
-        instance.create(2, "Node")
-        instance.create(3, "Node")
-        instance.create(4, "NodeInterface")
-        instance.create_set(4, "up", True)
-        instance.connect(2, "devs", 4, "node")
-        instance.add_adddress(4, AF_INET, "10.0.0.1", 24, None)
-        instance.create(5, "NodeInterface")
-        instance.create_set(5, "up", True)
-        instance.connect(3, "devs", 5, "node")
-        instance.add_adddress(5, AF_INET, "10.0.0.2", 24, None)
-        instance.create(6, "Switch")
-        instance.create_set(6, "up", True)
-        instance.connect(4, "switch", 6, "devs")
-        instance.connect(5, "switch", 6, "devs")
-        instance.create(7, "Application")
-        instance.create_set(7, "command", "ping -qc1 10.0.0.2")
-        instance.create_set(7, "user", user)
-        instance.add_trace(7, "stdout")
-        instance.connect(7, "node", 2, "apps")
+        instance = netns.TestbedController(testbed_version)
+        instance.defer_configure("homeDirectory", self.root_dir)
+        instance.defer_create(2, "Node")
+        instance.defer_create(3, "Node")
+        instance.defer_create(4, "NodeInterface")
+        instance.defer_create_set(4, "up", True)
+        instance.defer_connect(2, "devs", 4, "node")
+        instance.defer_add_address(4, "10.0.0.1", 24, None)
+        instance.defer_create(5, "NodeInterface")
+        instance.defer_create_set(5, "up", True)
+        instance.defer_connect(3, "devs", 5, "node")
+        instance.defer_add_address(5, "10.0.0.2", 24, None)
+        instance.defer_create(6, "Switch")
+        instance.defer_create_set(6, "up", True)
+        instance.defer_connect(4, "switch", 6, "devs")
+        instance.defer_connect(5, "switch", 6, "devs")
+        instance.defer_create(7, "Application")
+        instance.defer_create_set(7, "command", "ping -qc1 10.0.0.2")
+        instance.defer_create_set(7, "user", user)
+        instance.defer_add_trace(7, "stdout")
+        instance.defer_connect(7, "node", 2, "apps")
 
         instance.do_setup()
         instance.do_create()
-        instance.do_connect()
+        instance.do_connect_init()
+        instance.do_connect_compl()
+        instance.do_preconfigure()
         instance.do_configure()
+        instance.do_prestart()
         instance.start()
-        while instance.status(7) != STATUS_FINISHED:
+        while instance.status(7) != AS.STATUS_FINISHED:
             time.sleep(0.5)
         ping_result = instance.trace(7, "stdout")
         comp_result = """PING 10.0.0.2 (10.0.0.2) 56(84) bytes of data.
@@ -64,31 +65,34 @@ class NetnsExecuteTestCase(unittest.TestCase):
     def test_run_ping_p2pif(self):
         user = getpass.getuser()
         testbed_version = "01"
-        instance = netns.TestbedInstance(testbed_version)
-        instance.configure("homeDirectory", self._home_dir)
-        instance.create(2, "Node")
-        instance.create(3, "Node")
-        instance.create(4, "P2PNodeInterface")
-        instance.create_set(4, "up", True)
-        instance.connect(2, "devs", 4, "node")
-        instance.add_adddress(4, AF_INET, "10.0.0.1", 24, None)
-        instance.create(5, "P2PNodeInterface")
-        instance.create_set(5, "up", True)
-        instance.connect(3, "devs", 5, "node")
-        instance.add_adddress(5, AF_INET, "10.0.0.2", 24, None)
-        instance.connect(4, "p2p", 5, "p2p")
-        instance.create(6, "Application")
-        instance.create_set(6, "command", "ping -qc1 10.0.0.2")
-        instance.create_set(6, "user", user)
-        instance.add_trace(6, "stdout")
-        instance.connect(6, "node", 2, "apps")
+        instance = netns.TestbedController(testbed_version)
+        instance.defer_configure("homeDirectory", self.root_dir)
+        instance.defer_create(2, "Node")
+        instance.defer_create(3, "Node")
+        instance.defer_create(4, "P2PNodeInterface")
+        instance.defer_create_set(4, "up", True)
+        instance.defer_connect(2, "devs", 4, "node")
+        instance.defer_add_address(4, "10.0.0.1", 24, None)
+        instance.defer_create(5, "P2PNodeInterface")
+        instance.defer_create_set(5, "up", True)
+        instance.defer_connect(3, "devs", 5, "node")
+        instance.defer_add_address(5, "10.0.0.2", 24, None)
+        instance.defer_connect(4, "p2p", 5, "p2p")
+        instance.defer_create(6, "Application")
+        instance.defer_create_set(6, "command", "ping -qc1 10.0.0.2")
+        instance.defer_create_set(6, "user", user)
+        instance.defer_add_trace(6, "stdout")
+        instance.defer_connect(6, "node", 2, "apps")
 
         instance.do_setup()
         instance.do_create()
-        instance.do_connect()
+        instance.do_connect_init()
+        instance.do_connect_compl()
+        instance.do_preconfigure()
         instance.do_configure()
+        instance.do_prestart()
         instance.start()
-        while instance.status(6) != STATUS_FINISHED:
+        while instance.status(6) != AS.STATUS_FINISHED:
             time.sleep(0.5)
         ping_result = instance.trace(6, "stdout")
         comp_result = """PING 10.0.0.2 (10.0.0.2) 56(84) bytes of data.
@@ -104,50 +108,53 @@ class NetnsExecuteTestCase(unittest.TestCase):
     def test_run_ping_routing(self):
         user = getpass.getuser()
         testbed_version = "01"
-        instance = netns.TestbedInstance(testbed_version)
-        instance.configure("homeDirectory", self._home_dir)
-        instance.create(2, "Node")
-        instance.create(3, "Node")
-        instance.create(4, "Node")
-        instance.create(5, "NodeInterface")
-        instance.create_set(5, "up", True)
-        instance.connect(2, "devs", 5, "node")
-        instance.add_adddress(5, AF_INET, "10.0.0.1", 24, None)
-        instance.create(6, "NodeInterface")
-        instance.create_set(6, "up", True)
-        instance.connect(3, "devs", 6, "node")
-        instance.add_adddress(6, AF_INET, "10.0.0.2", 24, None)
-        instance.create(7, "NodeInterface")
-        instance.create_set(7, "up", True)
-        instance.connect(3, "devs", 7, "node")
-        instance.add_adddress(7, AF_INET, "10.0.1.1", 24, None)
-        instance.create(8, "NodeInterface")
-        instance.create_set(8, "up", True)
-        instance.connect(4, "devs", 8, "node")
-        instance.add_adddress(8, AF_INET, "10.0.1.2", 24, None)
-        instance.create(9, "Switch")
-        instance.create_set(9, "up", True)
-        instance.connect(5, "switch", 9, "devs")
-        instance.connect(6, "switch", 9, "devs")
-        instance.create(10, "Switch")
-        instance.create_set(10, "up", True)
-        instance.connect(7, "switch", 10, "devs")
-        instance.connect(8, "switch", 10, "devs")
-        instance.create(11, "Application")
-        instance.create_set(11, "command", "ping -qc1 10.0.1.2")
-        instance.create_set(11, "user", user)
-        instance.add_trace(11, "stdout")
-        instance.connect(11, "node", 2, "apps")
-
-        instance.add_route(2, "10.0.1.0", 24, "10.0.0.2")
-        instance.add_route(4, "10.0.0.0", 24, "10.0.1.1")
+        instance = netns.TestbedController(testbed_version)
+        instance.defer_configure("homeDirectory", self.root_dir)
+        instance.defer_create(2, "Node")
+        instance.defer_create(3, "Node")
+        instance.defer_create(4, "Node")
+        instance.defer_create(5, "NodeInterface")
+        instance.defer_create_set(5, "up", True)
+        instance.defer_connect(2, "devs", 5, "node")
+        instance.defer_add_address(5, "10.0.0.1", 24, None)
+        instance.defer_create(6, "NodeInterface")
+        instance.defer_create_set(6, "up", True)
+        instance.defer_connect(3, "devs", 6, "node")
+        instance.defer_add_address(6, "10.0.0.2", 24, None)
+        instance.defer_create(7, "NodeInterface")
+        instance.defer_create_set(7, "up", True)
+        instance.defer_connect(3, "devs", 7, "node")
+        instance.defer_add_address(7, "10.0.1.1", 24, None)
+        instance.defer_create(8, "NodeInterface")
+        instance.defer_create_set(8, "up", True)
+        instance.defer_connect(4, "devs", 8, "node")
+        instance.defer_add_address(8, "10.0.1.2", 24, None)
+        instance.defer_create(9, "Switch")
+        instance.defer_create_set(9, "up", True)
+        instance.defer_connect(5, "switch", 9, "devs")
+        instance.defer_connect(6, "switch", 9, "devs")
+        instance.defer_create(10, "Switch")
+        instance.defer_create_set(10, "up", True)
+        instance.defer_connect(7, "switch", 10, "devs")
+        instance.defer_connect(8, "switch", 10, "devs")
+        instance.defer_create(11, "Application")
+        instance.defer_create_set(11, "command", "ping -qc1 10.0.1.2")
+        instance.defer_create_set(11, "user", user)
+        instance.defer_add_trace(11, "stdout")
+        instance.defer_connect(11, "node", 2, "apps")
+
+        instance.defer_add_route(2, "10.0.1.0", 24, "10.0.0.2")
+        instance.defer_add_route(4, "10.0.0.0", 24, "10.0.1.1")
 
         instance.do_setup()
         instance.do_create()
-        instance.do_connect()
+        instance.do_connect_init()
+        instance.do_connect_compl()
+        instance.do_preconfigure()
         instance.do_configure()
+        instance.do_prestart()
         instance.start()
-        while instance.status(11) != STATUS_FINISHED:
+        while instance.status(11) != AS.STATUS_FINISHED:
             time.sleep(0.5)
         ping_result = instance.trace(11, "stdout")
         comp_result = """PING 10.0.1.2 (10.0.1.2) 56(84) bytes of data.
@@ -158,9 +165,53 @@ class NetnsExecuteTestCase(unittest.TestCase):
         self.assertTrue(ping_result.startswith(comp_result))
         instance.stop()
         instance.shutdown()
-        
+
+    @test_util.skipUnless(os.getuid() == 0, "Test requires root privileges")
+    def test_node_pcap_trace(self):
+        user = getpass.getuser()
+        testbed_version = "01"
+        instance = netns.TestbedController(testbed_version)
+        instance.defer_configure("homeDirectory", self.root_dir)
+        instance.defer_create(2, "Node")
+        instance.defer_add_trace(2, "pcap")
+        instance.defer_create(3, "Node")
+        instance.defer_create(4, "P2PNodeInterface")
+        instance.defer_create_set(4, "up", True)
+        instance.defer_connect(2, "devs", 4, "node")
+        instance.defer_add_address(4, "10.0.0.1", 24, None)
+        instance.defer_create(5, "P2PNodeInterface")
+        instance.defer_create_set(5, "up", True)
+        instance.defer_connect(3, "devs", 5, "node")
+        instance.defer_add_address(5, "10.0.0.2", 24, None)
+        instance.defer_connect(4, "p2p", 5, "p2p")
+        instance.defer_create(6, "Application")
+        instance.defer_add_trace(6, "stdout")
+        instance.defer_create_set(6, "command", "ping -qc5 10.0.0.2")
+        instance.defer_create_set(6, "user", user)
+        instance.defer_connect(6, "node", 2, "apps")
+
+        instance.do_setup()
+        instance.do_create()
+        instance.do_connect_init()
+        instance.do_connect_compl()
+        instance.do_preconfigure()
+        instance.do_configure()
+        instance.do_prestart()
+        instance.start()
+        while instance.status(6) != AS.STATUS_FINISHED:
+            time.sleep(0.5)
+        pcap_result = instance.trace(2, "pcap")
+        self.assertEquals(len(pcap_result), 1024)
+        instance.stop()
+        instance.shutdown()
+
     def tearDown(self):
-        shutil.rmtree(self._home_dir)
+        try:
+            shutil.rmtree(self.root_dir)
+        except:
+            # retry
+            time.sleep(0.1)
+            shutil.rmtree(self.root_dir)
 
 if __name__ == '__main__':
     unittest.main()