test/tools/Robot_Tool/libraries/ArpHandler.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 *
  11 from testmodule import TestModule
  12
  13 sys.path.remove('..')
  14
  15
  16 class ArpHandler(TestModule):
  17     """
  18     Test for the arp handler.
  19     Start 2-layer tree topology network. e.g., in Mininet, run  'sudo mn --controller=remote,ip=127.0.0.1 --mac --topo tree,2'
  20     """
  21
  22     def __init__(self, restSubContext='/controller/nb/v2/subnetservice', user=DEFAULT_USER, password=DEFAULT_PWD,
  23                  container=DEFAULT_CONTAINER, contentType='json', prefix=DEFAULT_PREFIX):
  24         super(self.__class__, self).__init__(restSubContext, user, password, container, contentType, prefix)
  25
  26     def get_subnets(self):
  27         """
  28         The name is suggested to match the NB API.
  29         list all subnets and their properties.
  30         """
  31         return super(self.__class__, self).get_entries('subnets')
  32
  33     def add_subnet_gateway(self, name, body):
  34         """
  35         Add a subnet gateway.
  36         """
  37         super(self.__class__, self).add_entry('subnet', name, body)
  38
  39     def remove_subnet_gateway(self, name):
  40         """
  41         Remove a subnet gateway.
  42         """
  43         super(self.__class__, self).remove_entry('subnet', name)
  44
  45     def test_subnet_operations(self, name, body):
  46         """
  47         Test subnet operations, like adding and removeing a subnet.
  48         >>> ArpHandler().test_subnet_operations('test',{'name':'test','subnet':'10.0.0.254/8'})
  49         True
  50         """
  51         return super(self.__class__, self).test_add_remove_operations('subnets', 'subnet', name, body, 'subnetConfig')
  52
  53
  54 if __name__ == '__main__':
  55     print 'arp handler'