test/util.py

   1 import socket
   2 import sys
   3 from cStringIO import StringIO
   4
   5
   6 def ppp(headline, packet):
   7     """ Return string containing the output of scapy packet.show() call. """
   8     o = StringIO()
   9     old_stdout = sys.stdout
  10     sys.stdout = o
  11     print(headline)
  12     packet.show()
  13     sys.stdout = old_stdout
  14     return o.getvalue()
  15
  16
  17 class Host(object):
  18     """ Generic test host "connected" to VPPs interface. """
  19
  20     @property
  21     def mac(self):
  22         """ MAC address """
  23         return self._mac
  24
  25     @property
  26     def ip4(self):
  27         """ IPv4 address """
  28         return self._ip4
  29
  30     @property
  31     def ip4n(self):
  32         """ IPv4 address """
  33         return socket.inet_pton(socket.AF_INET, self._ip4)
  34
  35     @property
  36     def ip6(self):
  37         """ IPv6 address """
  38         return self._ip6
  39
  40     def __init__(self, mac=None, ip4=None, ip6=None):
  41         self._mac = mac
  42         self._ip4 = ip4
  43         self._ip6 = ip6