| """Tests for machine_resource_utils.""" |
| |
| from __future__ import absolute_import |
| import os |
| import sys |
| import unittest |
| |
| # Allow imports from cq/scripts folder |
| sys.path.insert( |
| 0, |
| os.path.realpath(os.path.join(os.path.dirname(__file__), os.pardir))) |
| from helpers import machine_resource_utils |
| |
| |
| class MachineResourceUtilsTest(unittest.TestCase): |
| |
| def testGetFreeDiskFraction(self): |
| """Tests getting the free disk fraction.""" |
| self.assertLessEqual(0.0, machine_resource_utils.get_free_disk_fraction()) |
| self.assertLessEqual(machine_resource_utils.get_free_disk_fraction(), 1.0) |
| |
| def testGetNetworkIf(self): |
| """Tests getting network interface list.""" |
| if_list = machine_resource_utils.get_network_if() |
| self.assertTrue(if_list) |
| self.assertGreaterEqual(len(if_list), 1) |
| |
| def testGetNetworkStat(self): |
| if_list = machine_resource_utils.get_network_if() |
| self.assertGreater( |
| machine_resource_utils.get_current_directional_traffic( |
| if_list=if_list, |
| direction=machine_resource_utils.NET_DIRECTION_RX), |
| 0) |
| self.assertGreater( |
| machine_resource_utils.get_current_directional_traffic( |
| if_list=if_list, |
| direction=machine_resource_utils.NET_DIRECTION_TX), |
| 0) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
