aboutsummaryrefslogtreecommitdiff
path: root/crosperf/machine_manager_unittest.py
diff options
context:
space:
mode:
authorAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>2020-02-05 21:28:40 +0000
committerAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>2020-02-05 21:28:40 +0000
commitcddd960b0ba2eb62c372c0d3176c75f0bd05d5e8 (patch)
treef592e7ccc058bdd6c0beda99a104df4f73c5e1d3 /crosperf/machine_manager_unittest.py
parent81bca172159b02808081a57062a1c61ee9c54d76 (diff)
parent3c0fcaf157c060979b09c0d8e7a982ff5d1acdab (diff)
downloadtoolchain-utils-cddd960b0ba2eb62c372c0d3176c75f0bd05d5e8.tar.gz
Merging 8 commit(s) from Chromium's toolchain-utils am: 928fba9d74 am: cfb8763e2b am: 3c0fcaf157
Change-Id: Ie5c5dbcb6f4cfb497c32c3c73466e844ebf1dd6c
Diffstat (limited to 'crosperf/machine_manager_unittest.py')
-rwxr-xr-xcrosperf/machine_manager_unittest.py9
1 files changed, 4 insertions, 5 deletions
diff --git a/crosperf/machine_manager_unittest.py b/crosperf/machine_manager_unittest.py
index f6a77bb8..26eacbd7 100755
--- a/crosperf/machine_manager_unittest.py
+++ b/crosperf/machine_manager_unittest.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python2
+#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
@@ -13,8 +13,7 @@ import os.path
import time
import hashlib
import unittest
-
-import mock
+import unittest.mock as mock
import label
import machine_manager
@@ -778,7 +777,7 @@ class CrosMachineTest(unittest.TestCase):
self.assertEqual(cm.phys_kbytes, 4194304)
mock_run_cmd.return_value = [1, MEMINFO_STRING, '']
- self.assertRaises(cm._GetMemoryInfo)
+ self.assertRaises(Exception, cm._GetMemoryInfo)
@mock.patch.object(command_executer.CommandExecuter, 'CrosRunCommandWOutput')
@mock.patch.object(machine_manager.CrosMachine, 'SetUpChecksumInfo')
@@ -838,7 +837,7 @@ class CrosMachineTest(unittest.TestCase):
'44:6d:57:20:4a:c5 txqueuelen 1000 (Ethernet)')
mock_run_cmd.return_value = [0, 'invalid hardware config', '']
- self.assertRaises(cm._GetMachineID)
+ self.assertRaises(Exception, cm._GetMachineID)
def test_add_cooldown_waittime(self):
cm = machine_manager.CrosMachine('1.2.3.4.cros', '/usr/local/chromeos',