Merge remote branch 'autotest-upstream/master' into try-box1

client/common_lib/hosts/base_classes_unittest.py

Index: client/common_lib/hosts/base_classes_unittest.py
diff --git a/client/common_lib/hosts/base_classes_unittest.py b/client/common_lib/hosts/base_classes_unittest.py
index f0325f4c9b8b123badf7c58864a9d73d6e7d378d..bb8b85feb9d5fbe034e3482899cd5af733d7329b 100755
--- a/client/common_lib/hosts/base_classes_unittest.py
+++ b/client/common_lib/hosts/base_classes_unittest.py
@@ -3,6 +3,7 @@
 import unittest
 import common
 
+from autotest_lib.client.common_lib import error, utils
 from autotest_lib.client.common_lib.test_utils import mock
 from autotest_lib.client.common_lib.hosts import base_classes
 
@@ -21,5 +22,22 @@ class test_host_class(unittest.TestCase):
         self.assertRaises(NotImplementedError, host.run_output, "fake command")
 
 
+    def test_check_diskspace(self):
+        self.god.stub_function(base_classes.Host, 'run')
+        host = base_classes.Host()
+        host.hostname = 'unittest-host'
+        test_df_tail = ('/dev/sda1                    1061       939'
+                        '       123      89% /')
+        fake_cmd_status = utils.CmdResult(exit_status=0, stdout=test_df_tail)
+        host.run.expect_call('df -PB 1000000 /foo | tail -1').and_return(
+                fake_cmd_status)
+        self.assertRaises(error.AutoservDiskFullHostError,
+                          host.check_diskspace, '/foo', 0.2)
+        host.run.expect_call('df -PB 1000000 /foo | tail -1').and_return(
+                fake_cmd_status)
+        host.check_diskspace('/foo', 0.1)
+        self.god.check_playback()
+
+
 if __name__ == "__main__":
     unittest.main()