test/core/execute.py

   1 #!/usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3
   4 from nepi.util import tags
   5 from nepi.util.constants import ApplicationStatus as AS
   6 import getpass
   7 import mock
   8 import mock.metadata_v01
   9 import sys
  10 import time
  11 import unittest
  12
  13 class ExecuteTestCase(unittest.TestCase):
  14     def setUp(self):
  15         sys.modules["nepi.testbeds.mock.metadata_v01"] = mock.metadata_v01
  16
  17     def test_execute(self):
  18         testbed_version = "01"
  19         testbed_id = "mock"
  20         instance = mock.TestbedController(testbed_version)
  21         instance.defer_configure("fake", True)
  22         instance.defer_create(2, "Node")
  23         instance.defer_create(3, "Node")
  24         instance.defer_create(4, "Node")
  25         instance.defer_create(5, "Interface")
  26         instance.defer_create_set(5, "fake", True)
  27         instance.defer_connect(2, "devs", 5, "node")
  28         instance.defer_create(6, "Interface")
  29         instance.defer_create_set(6, "fake", True)
  30         instance.defer_connect(3, "devs", 6, "node")
  31         instance.defer_connect(5, "iface", 6, "iface")
  32         instance.defer_create(7, "Application")
  33         instance.defer_add_trace(7, "fake")
  34         instance.defer_connect(7, "node", 2, "apps")
  35
  36         instance.do_setup()
  37         instance.do_create()
  38         instance.do_connect_init()
  39         instance.do_connect_compl()
  40         instance.do_configure()
  41         instance.start()
  42         attr_list = instance.get_attribute_list(5)
  43         self.assertEquals(attr_list, ["test", "fake", "cross", "maxAddresses", "label"])
  44         while instance.status(7) != AS.STATUS_FINISHED:
  45             time.sleep(0.5)
  46         app_result = instance.trace(7, "fake")
  47         comp_result = """PING 10.0.0.2 (10.0.0.2) 56(84) bytes of data.
  48
  49 --- 10.0.0.2 ping statistics ---
  50 1 packets transmitted, 1 received, 0% packet loss, time 0ms
  51 """
  52         self.assertTrue(app_result.startswith(comp_result))
  53
  54         traces_info = instance.traces_info()
  55         expected_traces_info = dict({
  56             7 : dict({
  57                 'fake': dict({
  58                     'host': 'localhost',
  59                     'user': getpass.getuser(),
  60                     'filepath': '<test>'
  61                     })
  62                 })
  63             })
  64         self.assertEquals(traces_info, expected_traces_info)
  65
  66         instance.stop()
  67         instance.shutdown()
  68
  69 if __name__ == '__main__':
  70     unittest.main()
  71