Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(594)

Unified Diff: lib/cros_build_lib_unittest.py

Issue 6579048: Reintroduce RunCommand cleanup and unit tests. Fixed previous issue, and (Closed) Base URL: http://git.chromium.org/git/crosutils.git@master
Patch Set: Fixup RunCommand DocString Created 9 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « lib/cros_build_lib.py ('k') | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: lib/cros_build_lib_unittest.py
diff --git a/lib/cros_build_lib_unittest.py b/lib/cros_build_lib_unittest.py
new file mode 100755
index 0000000000000000000000000000000000000000..bb3f9d6e4e2d3ba734ad800e6b8c30e4a74b0237
--- /dev/null
+++ b/lib/cros_build_lib_unittest.py
@@ -0,0 +1,92 @@
+#!/usr/bin/python
+#
+# Copyright (c) 2011 The Chromium OS Authors. All rights reserved.
+# Use of this source code is governed by a BSD-style license that can be
+# found in the LICENSE file.
+
+"""Unit tests for cros_build_lib."""
+
+import unittest
+
+import cros_build_lib
+
+class CrosBuildLibTest(unittest.TestCase):
+ """Test class for cros_build_lib."""
+
+ def testRunCommandSimple(self):
+ """Test that RunCommand can run a simple successful command."""
+ result = cros_build_lib.RunCommand(['ls'],
+ # Keep the test quiet options
+ print_cmd=False,
+ redirect_stdout=True,
+ redirect_stderr=True,
+ # Test specific options
+ exit_code=True)
+ self.assertEqual(result, 0)
+
+ def testRunCommandError(self):
+ """Test that RunCommand can return an error code for a failed command."""
+ result = cros_build_lib.RunCommand(['ls', '/nosuchdir'],
+ # Keep the test quiet options
+ print_cmd=False,
+ redirect_stdout=True,
+ redirect_stderr=True,
+ # Test specific options
+ exit_code=True)
+ self.assertNotEqual(result, 0)
+ self.assertEquals(type(result), int)
+
+ def testRunCommandErrorRetries(self):
+ """Test that RunCommand can retry a failed command that always fails."""
+
+ # We don't actually check that it's retrying, just exercise the code path.
+ result = cros_build_lib.RunCommand(['ls', '/nosuchdir'],
+ # Keep the test quiet options
+ print_cmd=False,
+ redirect_stdout=True,
+ redirect_stderr=True,
+ # Test specific options
+ num_retries=2,
+ error_ok=True,
+ exit_code=True)
+ self.assertNotEqual(result, 0)
+ self.assertEquals(type(result), int)
+
+ def testRunCommandErrorException(self):
+ """Test that RunCommand can throw an exception when a command fails."""
+
+ function = lambda : cros_build_lib.RunCommand(['ls', '/nosuchdir'],
+ # Keep the test quiet options
+ print_cmd=False,
+ redirect_stdout=True,
+ redirect_stderr=True)
+ self.assertRaises(cros_build_lib.RunCommandException, function)
+
+ def testRunCommandErrorCodeNoException(self):
+ """Test that RunCommand doesn't throw an exception with exit_code."""
+
+ result = cros_build_lib.RunCommand(['ls', '/nosuchdir'],
+ # Keep the test quiet options
+ print_cmd=False,
+ redirect_stdout=True,
+ redirect_stderr=True,
+ # Test specific options
+ exit_code=True)
+ # We are really testing that it doesn't throw an exception if exit_code
+ # if true.
+ self.assertNotEqual(result, 0)
+ self.assertEquals(type(result), int)
+
+ def testRunCommandCaptureOutput(self):
+ """Test that RunCommand can capture stdout if a command succeeds."""
+
+ result = cros_build_lib.RunCommand(['echo', '-n', 'Hi'],
+ # Keep the test quiet options
+ print_cmd=False,
+ redirect_stdout=True,
+ redirect_stderr=True)
+ self.assertEqual(result, 'Hi')
+
+
+if __name__ == '__main__':
+ unittest.main()
« no previous file with comments | « lib/cros_build_lib.py ('k') | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698