contrib/gnunet_pyexpect.py.in

   1 #!@PYTHON@\r
   2 #    This file is part of GNUnet.\r
   3 #    (C) 2010 Christian Grothoff (and other contributing authors)\r
   4 #\r
   5 #    GNUnet is free software; you can redistribute it and/or modify\r
   6 #    it under the terms of the GNU General Public License as published\r
   7 #    by the Free Software Foundation; either version 2, or (at your\r
   8 #    option) any later version.\r
   9 #\r
  10 #    GNUnet is distributed in the hope that it will be useful, but\r
  11 #    WITHOUT ANY WARRANTY; without even the implied warranty of\r
  12 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU\r
  13 #    General Public License for more details.\r
  14 #\r
  15 #    You should have received a copy of the GNU General Public License\r
  16 #    along with GNUnet; see the file COPYING.  If not, write to the\r
  17 #    Free Software Foundation, Inc., 59 Temple Place - Suite 330,\r
  18 #    Boston, MA 02111-1307, USA.\r
  19 #\r
  20 # Testcase for gnunet-peerinfo\r
  21 from __future__ import print_function\r
  22 import os\r
  23 import re\r
  24 import subprocess\r
  25 import sys\r
  26 import shutil\r
  27 import time\r
  28 \r
  29 class pexpect (object):\r
  30   def __init__ (self):\r
  31     super (pexpect, self).__init__ ()\r
  32 \r
  33   def spawn (self, stdin, arglist, *pargs, **kwargs):\r
  34     env = kwargs.pop ('env', None)\r
  35     if env is None:\r
  36       env = os.environ.copy ()\r
  37     # This messes up some testcases, disable log redirection\r
  38     env.pop ('GNUNET_FORCE_LOGFILE', None)\r
  39     self.proc = subprocess.Popen (arglist, *pargs, env=env, **kwargs)\r
  40     if self.proc is None:\r
  41       print ("Failed to spawn a process {0}".format (arglist))\r
  42       sys.exit (1)\r
  43     if stdin is not None:\r
  44       self.stdo, self.stde = self.proc.communicate (stdin)\r
  45     else:\r
  46       self.stdo, self.stde = self.proc.communicate ()\r
  47     return self.proc\r
  48 \r
  49   def expect (self, s, r, flags=0):\r
  50     stream = self.stdo if s == 'stdout' else self.stde\r
  51     if isinstance (r, str):\r
  52       if r == "EOF":\r
  53         if len (stream) == 0:\r
  54           return True\r
  55         else:\r
  56           print ("Failed to find `{1}' in {0}, which is `{2}' ({3})".format (s, r, stream, len (stream)))\r
  57           sys.exit (2)\r
  58       raise ValueError ("Argument `r' should be an instance of re.RegexObject or a special string, but is `{0}'".format (r))\r
  59     m = r.search (stream, flags)\r
  60     if not m:\r
  61       print ("Failed to find `{1}' in {0}, which is is `{2}'".format (s, r.pattern, stream))\r
  62       sys.exit (2)\r
  63     stream = stream[m.end ():]\r
  64     if s == 'stdout':\r
  65       self.stdo = stream\r
  66     else:\r
  67       self.stde = stream\r
  68     return m\r
  69 \r
  70   def read (self, s, size=-1):\r
  71     stream = self.stdo if s == 'stdout' else self.stde\r
  72     result = ""\r
  73     if size < 0:\r
  74       result = stream\r
  75       new_stream = ""\r
  76     else:\r
  77       result = stream[0:size]\r
  78       new_stream = stream[size:]\r
  79     if s == 'stdout':\r
  80       self.stdo = new_stream\r
  81     else:\r
  82       self.stde = new_stream\r
  83     return result\r