test/_testTools/tools.py

   1 #!/usr/bin/env python
   2 #-*- coding:utf-8 -*-
   3 #  Copyright (C) 2010-2012  CEA/DEN
   4 #
   5 #  This library is free software; you can redistribute it and/or
   6 #  modify it under the terms of the GNU Lesser General Public
   7 #  License as published by the Free Software Foundation; either
   8 #  version 2.1 of the License.
   9 #
  10 #  This library 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 GNU
  13 #  Lesser General Public License for more details.
  14 #
  15 #  You should have received a copy of the GNU Lesser General Public
  16 #  License along with this library; if not, write to the Free Software
  17 #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  18
  19 import tempfile
  20 import sys
  21 import subprocess
  22
  23 class outRedirection():
  24     '''redirection of standart output
  25     useful for testing the terminal display
  26     '''
  27     def __init__(self):
  28         '''initialization
  29         '''
  30         self._fstream = tempfile.NamedTemporaryFile(mode='w')
  31         self.saveout = sys.stdout
  32         sys.stdout = self._fstream
  33
  34     def flush(self):
  35         self._fstream.flush()
  36
  37     def end_redirection(self):
  38         self._fstream.seek(0)
  39         ff = open(self._fstream.name, 'r')
  40         self.res = ff.read()
  41         self._fstream.close()
  42         sys.stdout = self.saveout
  43
  44     def read_results(self):
  45         try:
  46             return self.res
  47         except Exception as exc:
  48             print('Problem with redirection : %s' % exc)
  49             sys.exit(1)
  50
  51 def kill9(pid):
  52     subprocess.call("kill -9 " + pid, shell=True)
  53
  54 def check_proc_existence_and_kill(regex):
  55     cmd = 'ps aux | grep "' + regex + '"'
  56     psRes = subprocess.Popen(cmd, shell=True, stdout=subprocess.PIPE).communicate()[0]
  57     psRes = psRes.split('\n')
  58     for line in psRes:
  59         if 'grep' in line or len(line) == 0:
  60             continue
  61         line2 = [i for i in line.split(' ') if i != '']
  62         pid = line2[1]
  63         kill9(pid)
  64         return pid
  65     return 0