1 # Copyright (C) 2013-2014 CEA/DEN, EDF R&D, OPEN CASCADE
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.
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.
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
17 # See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
25 from cStringIO import StringIO
28 logger = logging.getLogger("TestLauncherLogger")
29 logger.level = logging.DEBUG
30 logger.addHandler(logging.StreamHandler())
32 class TestSessionArgs(unittest.TestCase):
35 # Set some predefined command args and corresponding output messages
36 hello0 = ["hello.py", "args:outfile="+logFile]
38 hello1 = ["hello.py", "args:you,outfile="+logFile]
39 hello1Msg = "Hello to: you"
40 helloToAdd = ["hello.py", "args:add.py,1,2,3,outfile="+logFile]
41 helloToAddMsg = "Hello to: add.py, 1, 2, 3"
42 add0 = ["add.py", "args:outfile="+logFile]
44 add3 = ["add.py", "args:1,2,3,outfile="+logFile]
46 lines0 = ["lines.py", "args:outfile="+logFile]
47 lines0Msg = "No files given"
48 lines2 = ["lines.py", "args:hello.py,add.py,outfile="+logFile]
49 lines2Msg = "hello.py is 35 lines longadd.py is 37 lines long"
50 linesUnreadable = ["lines.py", "args:hello.py,add.py,1,2,outfile="+logFile]
51 linesUnreadableMsg = "hello.py is 35 lines longadd.py is 37 lines longFile '1' cannot be readFile '2' cannot be read"
54 # Initialize path to SALOME application
55 path_to_launcher = os.getenv("SALOME_LAUNCHER")
56 appli_dir = os.path.dirname(path_to_launcher)
57 envd_dir = os.path.join(appli_dir, "env.d")
59 # Configure session startup
60 self.SALOME = imp.load_source("SALOME", os.path.join(appli_dir,"salome"))
61 self.SALOME_args = ["shell", "--config="+envd_dir]
63 sys.stdout = StringIO()
69 def session(self, args=[]):
70 self.SALOME.main(self.SALOME_args + args)
72 def removeLogFile(self):
74 os.remove(self.logFile)
78 def assertLogFileContentsEqual(self, message):
79 with open(self.logFile, "r") as f:
80 contents = f.read().replace('\n', '')
82 self.assertTrue(contents==message, "Contents differ!\n\tGenerated contents: %s\n\tExpected contents: %s"%(contents, message))
85 self.session(self.hello0)
86 self.assertLogFileContentsEqual(self.hello0Msg)
88 def testPythonHello0(self):
89 self.session(["python"]+self.hello0)
90 self.assertLogFileContentsEqual(self.hello0Msg)
93 self.session(self.hello1)
94 self.assertLogFileContentsEqual(self.hello1Msg)
97 self.session(self.add0)
98 self.assertLogFileContentsEqual(self.add0Msg)
101 self.session(self.add3)
102 self.assertLogFileContentsEqual(self.add3Msg)
104 def testHello0Add3(self):
105 self.session(self.hello0+self.add3)
106 self.assertLogFileContentsEqual(self.hello0Msg+self.add3Msg)
108 def testHello1Add3(self):
109 self.session(self.hello1+self.add3)
110 self.assertLogFileContentsEqual(self.hello1Msg+self.add3Msg)
112 def testHelloToAdd(self):
113 self.session(self.helloToAdd)
114 self.assertLogFileContentsEqual(self.helloToAddMsg)
116 def testLines0(self):
117 self.session(self.lines0)
118 self.assertLogFileContentsEqual(self.lines0Msg)
120 def testLines2(self):
121 self.session(self.lines2)
122 self.assertLogFileContentsEqual(self.lines2Msg)
124 def testLines2Add3(self):
125 self.session(self.lines2+self.add3)
126 self.assertLogFileContentsEqual(self.lines2Msg+self.add3Msg)
128 def testLinesUnreadable(self):
129 self.session(self.linesUnreadable)
130 self.assertLogFileContentsEqual(self.linesUnreadableMsg)
132 def testAddAddHello(self):
133 self.session(self.add3+self.add3+self.hello1)
134 self.assertLogFileContentsEqual(self.add3Msg+self.add3Msg+self.hello1Msg)
136 def testHello0Add3Hello0Add3Hello0(self):
137 self.session(self.hello1+self.add3+self.hello0+self.add3+self.hello0)
138 self.assertLogFileContentsEqual(self.hello1Msg+self.add3Msg+self.hello0Msg+self.add3Msg+self.hello0Msg)
143 if __name__ == "__main__":
144 path_to_launcher = os.getenv("SALOME_LAUNCHER")
145 if not path_to_launcher:
146 msg = "Error: please set SALOME_LAUNCHER variable to the salome command of your application folder."