from nepi.util.constants import ApplicationStatus as AS
from nepi.testbeds import planetlab
import os
+import re
import shutil
+import sys
import tempfile
+import test_util
import time
import unittest
-import re
-import test_util
-import sys
class PlanetLabExecuteTestCase(unittest.TestCase):
testbed_id = "planetlab"
os.path.dirname(planetlab.__file__),
'scripts',
'plr50.c')
+ TOS_PY = os.path.join(
+ os.path.dirname(planetlab.__file__),
+ 'scripts',
+ 'tosqueue.py')
+ CLS_PY = os.path.join(
+ os.path.dirname(planetlab.__file__),
+ 'scripts',
+ 'classqueue.py')
def setUp(self):
self.root_dir = tempfile.mkdtemp()
pl_ssh_key = os.environ.get(
"PL_SSH_KEY",
"%s/.ssh/id_rsa_planetlab" % (os.environ['HOME'],) )
+ slicename = os.environ.get(
+ "PL_SLICE",
+ slicename)
pl_user, pl_pwd = test_util.pl_auth()
instance.defer_configure("homeDirectory", self.root_dir)
instance.defer_configure("plcHost", plchost)
instance.defer_configure("tapPortBase", self.port_base)
instance.defer_configure("p2pDeployment", False) # it's interactive, we don't want it in tests
+ instance.defer_configure("cleanProc", True)
+
+ # Hack, but we need vsys_vnet
+ instance.do_setup()
+ vnet = instance.vsys_vnet
+ self.net_prefix = vnet.rsplit('.',1)[0]
return instance
ping_result = instance.trace(7, "stdout") or ""
instance.stop()
finally:
- instance.shutdown()
+ try:
+ instance.shutdown()
+ except:
+ pass
# asserts at the end, to make sure there's proper cleanup
self.assertTrue(re.match(comp_result, ping_result, re.MULTILINE),
comp_result = r".*GNU Fortran \(GCC\).*"
instance.stop()
finally:
- instance.shutdown()
+ try:
+ instance.shutdown()
+ except:
+ pass
# asserts at the end, to make sure there's proper cleanup
self.assertTrue(re.match(comp_result, ping_result, re.MULTILINE),
ping_result = instance.trace(10, "stdout") or ""
instance.stop()
finally:
- instance.shutdown()
+ try:
+ instance.shutdown()
+ except:
+ pass
# asserts at the end, to make sure there's proper cleanup
self.assertTrue(re.match(comp_result, ping_result, re.MULTILINE),
instance.defer_create(4, "Internet")
instance.defer_connect(3, "inet", 4, "devs")
instance.defer_create(5, "TunInterface")
- instance.defer_add_address(5, "192.168.2.2", 24, False)
+ instance.defer_add_address(5, self.net_prefix+".2", 24, False)
instance.defer_connect(2, "devs", 5, "node")
instance.defer_create(6, "Application")
instance.defer_create_set(6, "command", """
comp_result = "OKIDOKI"
instance.stop()
finally:
- instance.shutdown()
+ try:
+ instance.shutdown()
+ except:
+ pass
# asserts at the end, to make sure there's proper cleanup
self.assertEqual(comp_result, test_result)
instance.stop()
finally:
- instance.shutdown()
+ try:
+ instance.shutdown()
+ except:
+ pass
# asserts at the end, to make sure there's proper cleanup
match = re.match(comp_result, test_result, re.MULTILINE)
self.assertTrue(netpipe_stats, "Unavailable netpipe stats")
@test_util.skipUnless(test_util.pl_auth() is not None, "Test requires PlanetLab authentication info (PL_USER and PL_PASS environment variables)")
- def _pingtest(self, TunClass, ConnectionProto, Cipher, Filter1=None, Filter2=None):
+ def _pingtest(self, TunClass, ConnectionProto, Cipher, Filter1=None, Filter2=None, Filter1args=None, Filter2args=None, PLREX=None, flood=False):
instance = self.make_instance()
instance.defer_create(2, "Node")
instance.defer_create(7, TunClass)
instance.defer_create_set(7, "tun_cipher", Cipher)
instance.defer_add_trace(7, "packets")
- instance.defer_add_address(7, "192.168.2.2", 24, False)
+ instance.defer_add_address(7, self.net_prefix+".2", 24, False)
+ if flood:
+ instance.defer_create_set(7, "bwlimit", 128)
instance.defer_connect(2, "devs", 7, "node")
instance.defer_create(8, TunClass)
instance.defer_create_set(8, "tun_cipher", Cipher)
instance.defer_add_trace(8, "packets")
- instance.defer_add_address(8, "192.168.2.3", 24, False)
+ instance.defer_add_address(8, self.net_prefix+".3", 24, False)
instance.defer_connect(3, "devs", 8, "node")
instance.defer_create(9, "Application")
- instance.defer_create_set(9, "command", "ping -qc10 {#[GUID-8].addr[0].[Address]#}")
+ if flood:
+ instance.defer_create_set(9, "command", "sudo -S ping -s 1000 -l 1000 -qfc1000 {#[GUID-8].addr[0].[Address]#} ; sleep 20 ; ping -qc10 {#[GUID-8].addr[0].[Address]#}")
+ else:
+ instance.defer_create_set(9, "command", "ping -qc10 {#[GUID-8].addr[0].[Address]#}")
instance.defer_add_trace(9, "stdout")
instance.defer_add_trace(9, "stderr")
instance.defer_connect(9, "node", 2, "apps")
if Filter1:
instance.defer_create(10, "TunFilter")
instance.defer_create_set(10, "module", Filter1)
+ if Filter1args:
+ instance.defer_create_set(10, "args", Filter1args)
instance.defer_connect(7, "fd->", 10, "->fd")
if Filter2:
instance.defer_create(11, "TunFilter")
instance.defer_create_set(11, "module", Filter2)
+ if Filter2args:
+ instance.defer_create_set(11, "args", Filter2args)
instance.defer_connect(8, "fd->", 11, "->fd")
- if Filter1 and Filter2:
- plr = "[5-9][0-9]"
- elif Filter1 or Filter2:
- plr = "[3-9][0-9]"
+ if PLREX is None:
+ if Filter1 and Filter2:
+ plr = "[5-9][0-9]"
+ elif Filter1 or Filter2:
+ plr = "[3-9][0-9]"
+ else:
+ plr = "0"
else:
- plr = "0"
+ plr = PLREX
instance.defer_connect(
(10 if Filter1 else 7), ConnectionProto,
comp_result = r"""PING .* \(.*\) \d*\(\d*\) bytes of data.
--- .* ping statistics ---
-10 packets transmitted, [0-9]+ received, %s%% packet loss, time \d*ms.*
+10 packets transmitted, [0-9]+ received,.* %s%% packet loss, time \d*ms.*
""" % (plr,)
+ if flood:
+ comp_result = ".*" + comp_result
try:
instance.do_setup()
packets2 = instance.trace(8, "packets") or ""
instance.stop()
finally:
- instance.shutdown()
+ try:
+ instance.shutdown()
+ except:
+ pass
# asserts at the end, to make sure there's proper cleanup
- self.assertTrue(re.match(comp_result, ping_result, re.MULTILINE),
+ self.assertTrue(re.match(comp_result, ping_result, re.MULTILINE|re.DOTALL),
"Unexpected trace:\n%s\nPackets @ source:\n%s\nPackets @ target:\n%s" % (
ping_result,
packets1,
def test_tun_ping_gre(self):
self._pingtest("TunInterface", "gre", "PLAIN")
+ @test_util.skipUnless(test_util.pl_auth() is not None, "Test requires PlanetLab authentication info (PL_USER and PL_PASS environment variables)")
+ def test_tun_ping_flood(self):
+ self._pingtest("TunInterface", "tcp", "PLAIN", flood = True)
+
+ @test_util.skipUnless(test_util.pl_auth() is not None, "Test requires PlanetLab authentication info (PL_USER and PL_PASS environment variables)")
+ def test_tun_ping_flood_udp(self):
+ self._pingtest("TunInterface", "udp", "PLAIN", flood = True)
+
@test_util.skipUnless(test_util.pl_auth() is not None, "Test requires PlanetLab authentication info (PL_USER and PL_PASS environment variables)")
def test_tap_ping(self):
self._pingtest("TapInterface", "tcp", "AES")
@test_util.skipUnless(test_util.pl_auth() is not None, "Test requires PlanetLab authentication info (PL_USER and PL_PASS environment variables)")
def test_tap_ping_udp_loss1_py(self):
- self._pingtest("TapInterface", "udp", "AES", self.PLR50_PY)
+ self._pingtest("TapInterface", "udp", "AES", self.PLR50_PY, None, "plr=50")
@test_util.skipUnless(test_util.pl_auth() is not None, "Test requires PlanetLab authentication info (PL_USER and PL_PASS environment variables)")
def test_tap_ping_udp_loss2_py(self):
- self._pingtest("TapInterface", "udp", "AES", self.PLR50_PY, self.PLR50_PY)
+ self._pingtest("TapInterface", "udp", "AES", self.PLR50_PY, self.PLR50_PY, "plr=40", "plr=40")
@test_util.skipUnless(test_util.pl_auth() is not None, "Test requires PlanetLab authentication info (PL_USER and PL_PASS environment variables)")
def test_tap_ping_udp_loss1_c(self):
- self._pingtest("TapInterface", "udp", "AES", self.PLR50_C)
+ self._pingtest("TapInterface", "udp", "AES", self.PLR50_C, None, "plr=50")
@test_util.skipUnless(test_util.pl_auth() is not None, "Test requires PlanetLab authentication info (PL_USER and PL_PASS environment variables)")
def test_tap_ping_udp_loss2_c(self):
- self._pingtest("TapInterface", "udp", "AES", self.PLR50_C, self.PLR50_C)
+ self._pingtest("TapInterface", "udp", "AES", self.PLR50_C, self.PLR50_C, "plr=40", "plr=40")
+
+ @test_util.skipUnless(test_util.pl_auth() is not None, "Test requires PlanetLab authentication info (PL_USER and PL_PASS environment variables)")
+ def test_tun_ping_udp_tos(self):
+ self._pingtest("TunInterface", "udp", "AES", self.TOS_PY, self.TOS_PY, "size=1000", "size=1000", "0")
+
+ @test_util.skipUnless(test_util.pl_auth() is not None, "Test requires PlanetLab authentication info (PL_USER and PL_PASS environment variables)")
+ def test_tun_ping_udp_class(self):
+ self._pingtest("TunInterface", "udp", "AES", self.CLS_PY, self.CLS_PY, "size=10", "size=10", "0")
@test_util.skipUnless(test_util.pl_auth() is not None, "Test requires PlanetLab authentication info (PL_USER and PL_PASS environment variables)")
def test_nepi_depends(self):
ping_result = (instance.trace(12, "stderr") or "").strip()
instance.stop()
finally:
- instance.shutdown()
+ try:
+ instance.shutdown()
+ except:
+ pass
# asserts at the end, to make sure there's proper cleanup
self.assertEqual(ping_result, "")
ping_result = (instance.trace(12, "stderr") or "").strip()
instance.stop()
finally:
- instance.shutdown()
+ try:
+ instance.shutdown()
+ except:
+ pass
# asserts at the end, to make sure there's proper cleanup
self.assertEqual(ping_result, "")
ping_result = instance.trace(7, "stdout") or ""
instance.stop()
finally:
- instance.shutdown()
+ try:
+ instance.shutdown()
+ except:
+ pass
# asserts at the end, to make sure there's proper cleanup
self.assertTrue(re.match(comp_result, ping_result, re.MULTILINE),