1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/testing/mozbase/mozdevice/sut_tests/test_exec_env.py Wed Dec 31 06:09:35 2014 +0100
1.3 @@ -0,0 +1,31 @@
1.4 +# This Source Code Form is subject to the terms of the Mozilla Public
1.5 +# License, v. 2.0. If a copy of the MPL was not distributed with this
1.6 +# file, You can obtain one at http://mozilla.org/MPL/2.0/.
1.7 +
1.8 +import os
1.9 +import posixpath
1.10 +from StringIO import StringIO
1.11 +
1.12 +from dmunit import DeviceManagerTestCase
1.13 +
1.14 +class ExecEnvTestCase(DeviceManagerTestCase):
1.15 +
1.16 + def runTest(self):
1.17 + """Exec test with env vars."""
1.18 + # Push the file
1.19 + localfile = os.path.join('test-files', 'test_script.sh')
1.20 + remotefile = posixpath.join(self.dm.getDeviceRoot(), 'test_script.sh')
1.21 + self.dm.pushFile(localfile, remotefile)
1.22 +
1.23 + # Run the cmd
1.24 + out = StringIO()
1.25 + self.dm.shell(['sh', remotefile], out, env={'THE_ANSWER': 42})
1.26 +
1.27 + # Rewind the output file
1.28 + out.seek(0)
1.29 + # Make sure first line is 42
1.30 + line = out.readline()
1.31 + self.assertTrue(int(line) == 42)
1.32 +
1.33 + # Clean up
1.34 + self.dm.removeFile(remotefile)
