bin/appliskel/tests/launcher/TestLauncherSessionArgs.py

   1 import unittest
   2
   3 import os
   4 import sys
   5 import imp
   6 from cStringIO import StringIO
   7 import logging
   8
   9 logger = logging.getLogger("TestLauncherLogger")
  10 logger.level = logging.DEBUG
  11 logger.addHandler(logging.StreamHandler())
  12
  13 class TestSessionArgs(unittest.TestCase):
  14   @classmethod
  15   def setUpClass(cls):
  16     # Initialize path to SALOME application
  17     path_to_launcher = os.getenv("SALOME_LAUNCHER")
  18     if not path_to_launcher:
  19       msg = "Error: please set SALOME_LAUNCHER variable to the salome command of your application folder."
  20       self.fail(msg)
  21     #
  22     appli_dir = os.path.dirname(path_to_launcher)
  23     envd_dir = os.path.join(appli_dir, "env.d")
  24
  25     # Configure session startup
  26     cls.SALOME = imp.load_source("SALOME", os.path.join(appli_dir,"salome"))
  27     cls.SALOME_args = ["shell", "--config="+envd_dir]
  28
  29     cls.logFile = "log.txt"
  30     sys.stdout = StringIO()
  31
  32     # Set some predefined command args and corresponding output messages
  33     cls.hello0 = ["hello.py", "args:outfile="+cls.logFile]
  34     cls.hello0Msg = "Hello!"
  35     cls.hello1 = ["hello.py", "args:you,outfile="+cls.logFile]
  36     cls.hello1Msg = "Hello to: you"
  37     cls.helloToAdd = ["hello.py", "args:add.py,1,2,3,outfile="+cls.logFile]
  38     cls.helloToAddMsg = "Hello to: add.py, 1, 2, 3"
  39     cls.add0 = ["add.py", "args:outfile="+cls.logFile]
  40     cls.add0Msg = "No args!"
  41     cls.add3 = ["add.py", "args:1,2,3,outfile="+cls.logFile]
  42     cls.add3Msg = "1+2+3 = 6"
  43     cls.lines0 = ["lines.py", "args:outfile="+cls.logFile]
  44     cls.lines0Msg = "No files given"
  45     cls.lines2 = ["lines.py", "args:hello.py,add.py,outfile="+cls.logFile]
  46     cls.lines2Msg = "hello.py is 16 lines longadd.py is 18 lines long"
  47     cls.linesUnreadable = ["lines.py", "args:hello.py,add.py,1,2,outfile="+cls.logFile]
  48     cls.linesUnreadableMsg = "hello.py is 16 lines longadd.py is 18 lines longFile '1' cannot be readFile '2' cannot be read"
  49
  50   #
  51   @classmethod
  52   def tearDownClass(cls):
  53     pass
  54   #
  55   def setUp(self):
  56     self.removeLogFile()
  57   #
  58   def tearDown(self):
  59     self.removeLogFile()
  60   #
  61   def session(self, args=[]):
  62     self.SALOME.main(self.SALOME_args + args)
  63   #
  64   def removeLogFile(self):
  65     try:
  66       os.remove(self.logFile)
  67     except OSError:
  68       pass
  69   #
  70   def assertLogFileContentsEqual(self, message):
  71     with open(self.logFile, "r") as f:
  72       contents = f.read().replace('\n', '')
  73
  74     #sys.stderr.write("Generated contents :%s\n"%contents)
  75     #sys.stderr.write("Expected contents :%s\n"%message)
  76     self.assertTrue(contents==message)
  77   #
  78   def testHello0(self):
  79     self.session(self.hello0)
  80     self.assertLogFileContentsEqual(self.hello0Msg)
  81   #
  82   def testPythonHello0(self):
  83     self.session(["python"]+self.hello0)
  84     self.assertLogFileContentsEqual(self.hello0Msg)
  85   #
  86   def testHello1(self):
  87     self.session(self.hello1)
  88     self.assertLogFileContentsEqual(self.hello1Msg)
  89   #
  90   def testAdd0(self):
  91     self.session(self.add0)
  92     self.assertLogFileContentsEqual(self.add0Msg)
  93   #
  94   def testAdd3(self):
  95     self.session(self.add3)
  96     self.assertLogFileContentsEqual(self.add3Msg)
  97   #
  98   def testHello0Add3(self):
  99     self.session(self.hello0+self.add3)
 100     self.assertLogFileContentsEqual(self.hello0Msg+self.add3Msg)
 101   #
 102   def testHello1Add3(self):
 103     self.session(self.hello1+self.add3)
 104     self.assertLogFileContentsEqual(self.hello1Msg+self.add3Msg)
 105   #
 106   def testHelloToAdd(self):
 107     self.session(self.helloToAdd)
 108     self.assertLogFileContentsEqual(self.helloToAddMsg)
 109   #
 110   def testLines0(self):
 111     self.session(self.lines0)
 112     self.assertLogFileContentsEqual(self.lines0Msg)
 113   #
 114   def testLines2(self):
 115     self.session(self.lines2)
 116     self.assertLogFileContentsEqual(self.lines2Msg)
 117   #
 118   def testLines2Add3(self):
 119     self.session(self.lines2+self.add3)
 120     self.assertLogFileContentsEqual(self.lines2Msg+self.add3Msg)
 121   #
 122   def testLinesUnreadable(self):
 123     self.session(self.linesUnreadable)
 124     self.assertLogFileContentsEqual(self.linesUnreadableMsg)
 125   #
 126   def testAddAddHello(self):
 127     self.session(self.add3+self.add3+self.hello1)
 128     self.assertLogFileContentsEqual(self.add3Msg+self.add3Msg+self.hello1Msg)
 129   #
 130   def testHello0Add3Hello0Add3Hello0(self):
 131     self.session(self.hello1+self.add3+self.hello0+self.add3+self.hello0)
 132     self.assertLogFileContentsEqual(self.hello1Msg+self.add3Msg+self.hello0Msg+self.add3Msg+self.hello0Msg)
 133   #
 134 #
 135
 136
 137 if __name__ == "__main__":
 138   unittest.main()