test/resources/linux/test_utils.py

   1 """
   2     NEPI, a framework to manage network experiments
   3     Copyright (C) 2013 INRIA
   4
   5     This program is free software: you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation, either version 3 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License
  16     along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17
  18 """
  19
  20 from nepi.resources.linux.node import LinuxNode
  21
  22 import os
  23
  24 class DummyEC(object):
  25     @property
  26     def exp_id(self):
  27         return "nepi-1"
  28
  29 def create_node(hostname, username):
  30     ec = DummyEC()
  31     node = LinuxNode(ec, 1)
  32     node.set("hostname", hostname)
  33     node.set("username", username)
  34
  35     # If we don't return the reference to the EC
  36     # it will be released by the garbage collector since
  37     # the resources only save a weak refernce to it.
  38     return node, ec
  39
  40 def skipIfNotAlive(func):
  41     name = func.__name__
  42     def wrapped(*args, **kwargs):
  43         node, ec = create_node(args[1], args[2])
  44
  45         if not node.is_alive():
  46             print "*** WARNING: Skipping test %s: Node %s is not alive\n" % (
  47                 name, node.get("hostname"))
  48             return
  49
  50         return func(*args, **kwargs)
  51
  52     return wrapped
  53
  54 def skipInteractive(func):
  55     name = func.__name__
  56     def wrapped(*args, **kwargs):
  57         mode = os.environ.get("NEPI_INTERACTIVE", False)
  58         mode = mode and  mode.lower() in ['true', 'yes']
  59         if not mode:
  60             print "*** WARNING: Skipping test %s: Interactive mode off \n" % name
  61             return
  62
  63         return func(*args, **kwargs)
  64
  65     return wrapped
  66
  67