def test_get_cpu_usage_should_return_the_cpu_usage_since_its_last_invocation(
self):
cgroup = CpuCgroup("test", "/sys/fs/cgroup/cpu/system.slice/test")
TestCpuCgroup.mock_read_file_map = {
"/proc/stat":
os.path.join(data_dir, "cgroups", "proc_stat_t0"),
os.path.join(cgroup.path, "cpuacct.stat"):
os.path.join(data_dir, "cgroups", "cpuacct.stat_t0")
}
cgroup.initialize_cpu_usage()
TestCpuCgroup.mock_read_file_map = {
"/proc/stat":
os.path.join(data_dir, "cgroups", "proc_stat_t1"),
os.path.join(cgroup.path, "cpuacct.stat"):
os.path.join(data_dir, "cgroups", "cpuacct.stat_t1")
}
cpu_usage = cgroup.get_cpu_usage()
self.assertEqual(cpu_usage, 0.031)
TestCpuCgroup.mock_read_file_map = {
"/proc/stat":
os.path.join(data_dir, "cgroups", "proc_stat_t2"),
os.path.join(cgroup.path, "cpuacct.stat"):
os.path.join(data_dir, "cgroups", "cpuacct.stat_t2")
}
cpu_usage = cgroup.get_cpu_usage()
self.assertEqual(cpu_usage, 0.045)
def test_get_cpu_usage(self, patch_get_proc_stat, *args):
patch_get_proc_stat.return_value = fileutil.read_file(os.path.join(data_dir, "cgroups", "dummy_proc_stat"))
test_cpu_cg = CpuCgroup("test_extension", os.path.join(data_dir, "cgroups", "cpu_mount"))
# Mocking CPU consumption
patch_get_proc_stat.return_value = fileutil.read_file(os.path.join(data_dir, "cgroups",
"dummy_proc_stat_updated"))
cpu_usage = test_cpu_cg.get_cpu_usage()
self.assertEqual(5.114, cpu_usage)
def test_get_cpu_usage_should_return_the_cpu_usage_since_its_last_invocation(
self):
osutil = get_osutil()
cgroup = CpuCgroup("test", "/sys/fs/cgroup/cpu/system.slice/test")
TestCpuCgroup.mock_read_file_map = {
"/proc/stat":
os.path.join(data_dir, "cgroups", "proc_stat_t0"),
os.path.join(cgroup.path, "cpuacct.stat"):
os.path.join(data_dir, "cgroups", "cpuacct.stat_t0")
}
cgroup.initialize_cpu_usage()
TestCpuCgroup.mock_read_file_map = {
"/proc/stat":
os.path.join(data_dir, "cgroups", "proc_stat_t1"),
os.path.join(cgroup.path, "cpuacct.stat"):
os.path.join(data_dir, "cgroups", "cpuacct.stat_t1")
}
cpu_usage = cgroup.get_cpu_usage()
self.assertEqual(
cpu_usage,
round(100.0 * 0.000307697876885 * osutil.get_processor_cores(), 3))
TestCpuCgroup.mock_read_file_map = {
"/proc/stat":
os.path.join(data_dir, "cgroups", "proc_stat_t2"),
os.path.join(cgroup.path, "cpuacct.stat"):
os.path.join(data_dir, "cgroups", "cpuacct.stat_t2")
}
cpu_usage = cgroup.get_cpu_usage()
self.assertEqual(
cpu_usage,
round(100.0 * 0.000445181085968 * osutil.get_processor_cores(), 3))
def test_get_cpu_usage_should_raise_an_exception_when_initialize_cpu_usage_has_not_been_invoked(
self):
cgroup = CpuCgroup("test", "/sys/fs/cgroup/cpu/system.slice/test")
with self.assertRaises(CGroupsException):
cpu_usage = cgroup.get_cpu_usage() # pylint: disable=unused-variable