tools/CSIT_Test/base/modules/container_manager.py

   1 """
   2 CSIT test tools.
   3 Authors: Baohua Yang@IBM, Denghui Huang@IBM
   4 Updated: 2013-11-01
   5 """
   6
   7 import sys
   8
   9 sys.path.append('..')
  10 from restlib import *  # noqa
  11 from testmodule import TestModule
  12
  13 sys.path.remove('..')
  14
  15
  16 class ContainerManager(TestModule):
  17     """
  18     Test for the container manager.
  19     Start 2-layer tree topology network. e.g., in Mininet, run
  20         'sudo mn --controller=remote,ip=127.0.0.1 --mac --topo tree,2'
  21     """
  22
  23     def __init__(self, restSubContext='/controller/nb/v2/containermanager', user=DEFAULT_USER, password=DEFAULT_PWD,
  24                  container=None, contentType='json', prefix=DEFAULT_PREFIX):
  25         super(self.__class__, self).__init__(restSubContext, user, password, container, contentType, prefix)
  26
  27     def get_containers(self):
  28         """
  29         The name is suggested to match the NB API.
  30         Show the containers
  31         """
  32         return super(self.__class__, self).get_entries('containers')
  33
  34     def add_container(self, name, body):
  35         """
  36         Add a container
  37         """
  38         super(self.__class__, self).add_entry('container', name, body)
  39
  40     def remove_container(self, name):
  41         """
  42         Remove a container
  43         """
  44         super(self.__class__, self).remove_entry('container', name)
  45
  46     def test_container_operations(self, name, body):
  47         """
  48         Test subnet operations, like adding and removeing a subnet.
  49         >>> ContainerManager().test_container_operations('cont1',{'container':'cont1','flowSpecs': [],'staticVlan':'10',
  50                 'nodeConnectors':["OF|1@OF|00:00:00:00:00:00:00:01","OF|23@OF|00:00:00:00:00:00:20:21"]})
  51         True
  52         """
  53         return super(self.__class__, self).test_add_remove_operations('containers', 'container', name, body,
  54                                                                       'containerConfig')