test/testbeds/planetlab/execute.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3
   4 import getpass
   5 from nepi.util.constants import STATUS_FINISHED
   6 from nepi.testbeds import planetlab
   7 import os
   8 import shutil
   9 import tempfile
  10 import test_util
  11 import time
  12 import unittest
  13 import re
  14
  15 class NetnsExecuteTestCase(unittest.TestCase):
  16     def setUp(self):
  17         self.root_dir = tempfile.mkdtemp()
  18
  19     def tearDown(self):
  20         shutil.rmtree(self.root_dir)
  21
  22     def test_simple(self):
  23         testbed_version = "01"
  24         instance = planetlab.TestbedController(testbed_version)
  25
  26         instance.defer_configure("homeDirectory", self.root_dir)
  27         instance.defer_configure("slice", "inria_nepi12")
  28         instance.defer_configure("authUser", "claudio-daniel.freire@inria.fr")
  29         instance.defer_configure("authPass", getpass.getpass())
  30
  31         instance.defer_create(2, "Node")
  32         instance.defer_create_set(2, "hostname", "onelab11.pl.sophia.inria.fr")
  33         instance.defer_create(3, "Node")
  34         instance.defer_create_set(3, "hostname", "onelab10.pl.sophia.inria.fr")
  35         instance.defer_create(4, "NodeInterface")
  36         instance.defer_connect(2, "devs", 4, "node")
  37         instance.defer_create(5, "NodeInterface")
  38         instance.defer_connect(3, "devs", 5, "node")
  39         instance.defer_create(6, "Internet")
  40         instance.defer_connect(4, "inet", 6, "devs")
  41         instance.defer_connect(5, "inet", 6, "devs")
  42         instance.defer_create(7, "Application")
  43         instance.defer_create_set(7, "command", "ping -qc1 {#GUID-3.addr[0].[ip]#}")
  44         instance.defer_add_trace(7, "stdout")
  45         instance.defer_connect(7, "node", 2, "apps")
  46
  47         instance.do_setup()
  48         instance.do_create()
  49         instance.do_connect()
  50         instance.do_configure()
  51
  52         print instance.elements[4]
  53         print instance.elements[5]
  54
  55         instance.start()
  56         while instance.status(7) != STATUS_FINISHED:
  57             time.sleep(0.5)
  58         ping_result = instance.trace(7, "stdout")
  59         comp_result = r"""PING .* \(.*) \d*\(\d*\) bytes of data.
  60
  61 --- .* ping statistics ---
  62 1 packets transmitted, 1 received, 0% packet loss, time \d*ms.*
  63 """
  64         self.assertTrue(re.match(comp_result, ping_result, re.MULTILINE))
  65         instance.stop()
  66         instance.shutdown()
  67
  68
  69 if __name__ == '__main__':
  70     unittest.main()
  71