src/emuvim/test/base.py

   1 """
   2 Helper module that implements helpers for test implementations.
   3 """
   4
   5 import unittest
   6 import os
   7 import subprocess
   8 import docker
   9 from emuvim.dcemulator.net import DCNetwork
  10 from mininet.clean import cleanup
  11 from mininet.node import Controller
  12
  13 class SimpleTestTopology(unittest.TestCase):
  14     """
  15         Helper class to do basic test setups.
  16         s1 -- s2 -- s3 -- ... -- sN
  17     """
  18
  19     def __init__(self, *args, **kwargs):
  20         self.net = None
  21         self.s = []   # list of switches
  22         self.h = []   # list of hosts
  23         self.d = []   # list of docker containers
  24         self.dc = []  # list of data centers
  25         self.docker_cli = None
  26         super(SimpleTestTopology, self).__init__(*args, **kwargs)
  27
  28     def createNet(
  29             self,
  30             nswitches=0, ndatacenter=0, nhosts=0, ndockers=0,
  31             autolinkswitches=False):
  32         """
  33         Creates a Mininet instance and automatically adds some
  34         nodes to it.
  35         """
  36         self.net = DCNetwork()
  37
  38         # add some switches
  39         for i in range(0, nswitches):
  40             self.s.append(self.net.addSwitch('s%d' % i))
  41         # if specified, chain all switches
  42         if autolinkswitches:
  43             for i in range(0, len(self.s) - 1):
  44                 self.net.addLink(self.s[i], self.s[i + 1])
  45         # add some data centers
  46         for i in range(0, ndatacenter):
  47             self.dc.append(
  48                 self.net.addDatacenter(
  49                     'datacenter%d' % i,
  50                     metadata={"unittest_dc": i}))
  51         # add some hosts
  52         for i in range(0, nhosts):
  53             self.h.append(self.net.addHost('h%d' % i))
  54         # add some dockers
  55         for i in range(0, ndockers):
  56             self.d.append(self.net.addDocker('d%d' % i, dimage="ubuntu"))
  57
  58     def startNet(self):
  59         self.net.start()
  60
  61     def stopNet(self):
  62         self.net.stop()
  63
  64     def getDockerCli(self):
  65         """
  66         Helper to interact with local docker instance.
  67         """
  68         if self.docker_cli is None:
  69             self.docker_cli = docker.Client(
  70                 base_url='unix://var/run/docker.sock')
  71         return self.docker_cli
  72
  73     def getDockernetContainers(self):
  74         """
  75         List the containers managed by dockernet
  76         """
  77         return self.getDockerCli().containers(filters={"label": "com.dockernet"})
  78
  79     @staticmethod
  80     def setUp():
  81         pass
  82
  83     @staticmethod
  84     def tearDown():
  85         cleanup()
  86         # make sure that all pending docker containers are killed
  87         with open(os.devnull, 'w') as devnull:
  88             subprocess.call(
  89                 "sudo docker rm -f $(sudo docker ps --filter 'label=com.dockernet' -a -q)",
  90                 stdout=devnull,
  91                 stderr=devnull,
  92                 shell=True)