diff -r 000000000000 -r 6474c204b198 testing/mozbase/mozdevice/sut_tests/test_exec_env.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/testing/mozbase/mozdevice/sut_tests/test_exec_env.py Wed Dec 31 06:09:35 2014 +0100 @@ -0,0 +1,31 @@ +# This Source Code Form is subject to the terms of the Mozilla Public +# License, v. 2.0. If a copy of the MPL was not distributed with this +# file, You can obtain one at http://mozilla.org/MPL/2.0/. + +import os +import posixpath +from StringIO import StringIO + +from dmunit import DeviceManagerTestCase + +class ExecEnvTestCase(DeviceManagerTestCase): + + def runTest(self): + """Exec test with env vars.""" + # Push the file + localfile = os.path.join('test-files', 'test_script.sh') + remotefile = posixpath.join(self.dm.getDeviceRoot(), 'test_script.sh') + self.dm.pushFile(localfile, remotefile) + + # Run the cmd + out = StringIO() + self.dm.shell(['sh', remotefile], out, env={'THE_ANSWER': 42}) + + # Rewind the output file + out.seek(0) + # Make sure first line is 42 + line = out.readline() + self.assertTrue(int(line) == 42) + + # Clean up + self.dm.removeFile(remotefile)