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 STATUS_FINISHED
   6 import mock
   7 import mock.metadata_v01
   8 import sys
   9 import time
  10 import unittest
  11
  12 class ExecuteTestCase(unittest.TestCase):
  13     def setUp(self):
  14         sys.modules["nepi.testbeds.mock.metadata_v01"] = mock.metadata_v01
  15
  16     def test_execute(self):
  17         testbed_version = "01"
  18         testbed_id = "mock"
  19         instance = mock.TestbedController(testbed_version)
  20         instance.defer_configure("fake", True)
  21         instance.defer_create(2, "Node")
  22         instance.defer_create(3, "Node")
  23         instance.defer_create(4, "Node")
  24         instance.defer_create(5, "Interface")
  25         instance.defer_create_set(5, "fake", True)
  26         instance.defer_connect(2, "devs", 5, "node")
  27         instance.defer_create(6, "Interface")
  28         instance.defer_create_set(6, "fake", True)
  29         instance.defer_connect(3, "devs", 6, "node")
  30         instance.defer_connect(5, "iface", 6, "iface")
  31         instance.defer_create(7, "Application")
  32         instance.defer_add_trace(7, "fake")
  33         instance.defer_connect(7, "node", 2, "apps")
  34
  35         instance.do_setup()
  36         instance.do_create()
  37         instance.do_connect_init()
  38         instance.do_connect_compl()
  39         instance.do_configure()
  40         instance.start()
  41         attr_list = instance.get_attribute_list(5)
  42         self.assertEquals(attr_list, ["test", "fake", "cross", "label"])
  43         while instance.status(7) != STATUS_FINISHED:
  44             time.sleep(0.5)
  45         app_result = instance.trace(7, "fake")
  46         comp_result = """PING 10.0.0.2 (10.0.0.2) 56(84) bytes of data.
  47
  48 --- 10.0.0.2 ping statistics ---
  49 1 packets transmitted, 1 received, 0% packet loss, time 0ms
  50 """
  51
  52         self.assertTrue(app_result.startswith(comp_result))
  53         self.assertEquals(instance.get_tags(4), [tags.MOBILE])
  54
  55         instance.stop()
  56         instance.shutdown()
  57
  58 if __name__ == '__main__':
  59     unittest.main()
  60