bin/appliskel/tests/launcher/TestLauncherSessionArgs.py

   1 # Copyright (C) 2013-2014  CEA/DEN, EDF R&D, OPEN CASCADE
   2 #
   3 # This library is free software; you can redistribute it and/or
   4 # modify it under the terms of the GNU Lesser General Public
   5 # License as published by the Free Software Foundation; either
   6 # version 2.1 of the License, or (at your option) any later version.
   7 #
   8 # This library is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 # Lesser General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU Lesser General Public
  14 # License along with this library; if not, write to the Free Software
  15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 #
  17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 #
  19
  20 import unittest
  21
  22 import os
  23 import sys
  24 import imp
  25 from cStringIO import StringIO
  26 import logging
  27
  28 logger = logging.getLogger("TestLauncherLogger")
  29 logger.level = logging.DEBUG
  30 logger.addHandler(logging.StreamHandler())
  31
  32 class TestSessionArgs(unittest.TestCase):
  33   # setUpClass appears in Python 2.7
  34   @classmethod
  35   def setUpClass(cls):
  36     # Initialize path to SALOME application
  37     path_to_launcher = os.getenv("SALOME_LAUNCHER")
  38     appli_dir = os.path.dirname(path_to_launcher)
  39     envd_dir = os.path.join(appli_dir, "env.d")
  40
  41     # Configure session startup
  42     cls.SALOME = imp.load_source("SALOME", os.path.join(appli_dir,"salome"))
  43     cls.SALOME_args = ["shell", "--config="+envd_dir]
  44
  45     cls.logFile = "log.txt"
  46     sys.stdout = StringIO()
  47
  48     # Set some predefined command args and corresponding output messages
  49     cls.hello0 = ["hello.py", "args:outfile="+cls.logFile]
  50     cls.hello0Msg = "Hello!"
  51     cls.hello1 = ["hello.py", "args:you,outfile="+cls.logFile]
  52     cls.hello1Msg = "Hello to: you"
  53     cls.helloToAdd = ["hello.py", "args:add.py,1,2,3,outfile="+cls.logFile]
  54     cls.helloToAddMsg = "Hello to: add.py, 1, 2, 3"
  55     cls.add0 = ["add.py", "args:outfile="+cls.logFile]
  56     cls.add0Msg = "No args!"
  57     cls.add3 = ["add.py", "args:1,2,3,outfile="+cls.logFile]
  58     cls.add3Msg = "1+2+3 = 6"
  59     cls.lines0 = ["lines.py", "args:outfile="+cls.logFile]
  60     cls.lines0Msg = "No files given"
  61     cls.lines2 = ["lines.py", "args:hello.py,add.py,outfile="+cls.logFile]
  62     cls.lines2Msg = "hello.py is 35 lines longadd.py is 37 lines long"
  63     cls.linesUnreadable = ["lines.py", "args:hello.py,add.py,1,2,outfile="+cls.logFile]
  64     cls.linesUnreadableMsg = "hello.py is 35 lines longadd.py is 37 lines longFile '1' cannot be readFile '2' cannot be read"
  65   #
  66   @classmethod
  67   def tearDownClass(cls):
  68     pass
  69   #
  70   def setUp(self):
  71     self.removeLogFile()
  72   #
  73   def tearDown(self):
  74     self.removeLogFile()
  75   #
  76   def session(self, args=[]):
  77     self.SALOME.main(self.SALOME_args + args)
  78   #
  79   def removeLogFile(self):
  80     try:
  81       os.remove(self.logFile)
  82     except OSError:
  83       pass
  84   #
  85   def assertLogFileContentsEqual(self, message):
  86     with open(self.logFile, "r") as f:
  87       contents = f.read().replace('\n', '')
  88
  89     self.assertTrue(contents==message, "Contents differ!\n\tGenerated contents: %s\n\tExpected contents: %s"%(contents, message))
  90   #
  91   def testHello0(self):
  92     self.session(self.hello0)
  93     self.assertLogFileContentsEqual(self.hello0Msg)
  94   #
  95   def testPythonHello0(self):
  96     self.session(["python"]+self.hello0)
  97     self.assertLogFileContentsEqual(self.hello0Msg)
  98   #
  99   def testHello1(self):
 100     self.session(self.hello1)
 101     self.assertLogFileContentsEqual(self.hello1Msg)
 102   #
 103   def testAdd0(self):
 104     self.session(self.add0)
 105     self.assertLogFileContentsEqual(self.add0Msg)
 106   #
 107   def testAdd3(self):
 108     self.session(self.add3)
 109     self.assertLogFileContentsEqual(self.add3Msg)
 110   #
 111   def testHello0Add3(self):
 112     self.session(self.hello0+self.add3)
 113     self.assertLogFileContentsEqual(self.hello0Msg+self.add3Msg)
 114   #
 115   def testHello1Add3(self):
 116     self.session(self.hello1+self.add3)
 117     self.assertLogFileContentsEqual(self.hello1Msg+self.add3Msg)
 118   #
 119   def testHelloToAdd(self):
 120     self.session(self.helloToAdd)
 121     self.assertLogFileContentsEqual(self.helloToAddMsg)
 122   #
 123   def testLines0(self):
 124     self.session(self.lines0)
 125     self.assertLogFileContentsEqual(self.lines0Msg)
 126   #
 127   def testLines2(self):
 128     self.session(self.lines2)
 129     self.assertLogFileContentsEqual(self.lines2Msg)
 130   #
 131   def testLines2Add3(self):
 132     self.session(self.lines2+self.add3)
 133     self.assertLogFileContentsEqual(self.lines2Msg+self.add3Msg)
 134   #
 135   def testLinesUnreadable(self):
 136     self.session(self.linesUnreadable)
 137     self.assertLogFileContentsEqual(self.linesUnreadableMsg)
 138   #
 139   def testAddAddHello(self):
 140     self.session(self.add3+self.add3+self.hello1)
 141     self.assertLogFileContentsEqual(self.add3Msg+self.add3Msg+self.hello1Msg)
 142   #
 143   def testHello0Add3Hello0Add3Hello0(self):
 144     self.session(self.hello1+self.add3+self.hello0+self.add3+self.hello0)
 145     self.assertLogFileContentsEqual(self.hello1Msg+self.add3Msg+self.hello0Msg+self.add3Msg+self.hello0Msg)
 146   #
 147 #
 148
 149
 150 if __name__ == "__main__":
 151   path_to_launcher = os.getenv("SALOME_LAUNCHER")
 152   if not path_to_launcher:
 153     msg = "Error: please set SALOME_LAUNCHER variable to the salome command of your application folder."
 154     raise Exception(msg)
 155
 156   unittest.main()
 157 #