138 lines
4.5 KiB
Python
138 lines
4.5 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
# Copyright (c) 2009, Giampaolo Rodola'
|
||
|
|
# Copyright (c) 2017, Arnon Yaari
|
||
|
|
# All rights reserved.
|
||
|
|
# Use of this source code is governed by a BSD-style license that can be
|
||
|
|
# found in the LICENSE file.
|
||
|
|
|
||
|
|
"""AIX specific tests."""
|
||
|
|
|
||
|
|
import re
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
import psutil
|
||
|
|
from psutil import AIX
|
||
|
|
from psutil.tests import PsutilTestCase
|
||
|
|
from psutil.tests import sh
|
||
|
|
|
||
|
|
|
||
|
|
@unittest.skipIf(not AIX, "AIX only")
|
||
|
|
class AIXSpecificTestCase(PsutilTestCase):
|
||
|
|
def test_virtual_memory(self):
|
||
|
|
out = sh('/usr/bin/svmon -O unit=KB')
|
||
|
|
re_pattern = r"memory\s*"
|
||
|
|
for field in ("size inuse free pin virtual available mmode").split():
|
||
|
|
re_pattern += r"(?P<%s>\S+)\s+" % (field,)
|
||
|
|
matchobj = re.search(re_pattern, out)
|
||
|
|
|
||
|
|
self.assertIsNotNone(
|
||
|
|
matchobj, "svmon command returned unexpected output"
|
||
|
|
)
|
||
|
|
|
||
|
|
KB = 1024
|
||
|
|
total = int(matchobj.group("size")) * KB
|
||
|
|
available = int(matchobj.group("available")) * KB
|
||
|
|
used = int(matchobj.group("inuse")) * KB
|
||
|
|
free = int(matchobj.group("free")) * KB
|
||
|
|
|
||
|
|
psutil_result = psutil.virtual_memory()
|
||
|
|
|
||
|
|
# TOLERANCE_SYS_MEM from psutil.tests is not enough. For some reason
|
||
|
|
# we're seeing differences of ~1.2 MB. 2 MB is still a good tolerance
|
||
|
|
# when compared to GBs.
|
||
|
|
TOLERANCE_SYS_MEM = 2 * KB * KB # 2 MB
|
||
|
|
self.assertEqual(psutil_result.total, total)
|
||
|
|
self.assertAlmostEqual(
|
||
|
|
psutil_result.used, used, delta=TOLERANCE_SYS_MEM
|
||
|
|
)
|
||
|
|
self.assertAlmostEqual(
|
||
|
|
psutil_result.available, available, delta=TOLERANCE_SYS_MEM
|
||
|
|
)
|
||
|
|
self.assertAlmostEqual(
|
||
|
|
psutil_result.free, free, delta=TOLERANCE_SYS_MEM
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_swap_memory(self):
|
||
|
|
out = sh('/usr/sbin/lsps -a')
|
||
|
|
# From the man page, "The size is given in megabytes" so we assume
|
||
|
|
# we'll always have 'MB' in the result
|
||
|
|
# TODO maybe try to use "swap -l" to check "used" too, but its units
|
||
|
|
# are not guaranteed to be "MB" so parsing may not be consistent
|
||
|
|
matchobj = re.search(
|
||
|
|
r"(?P<space>\S+)\s+"
|
||
|
|
r"(?P<vol>\S+)\s+"
|
||
|
|
r"(?P<vg>\S+)\s+"
|
||
|
|
r"(?P<size>\d+)MB",
|
||
|
|
out,
|
||
|
|
)
|
||
|
|
|
||
|
|
self.assertIsNotNone(
|
||
|
|
matchobj, "lsps command returned unexpected output"
|
||
|
|
)
|
||
|
|
|
||
|
|
total_mb = int(matchobj.group("size"))
|
||
|
|
MB = 1024**2
|
||
|
|
psutil_result = psutil.swap_memory()
|
||
|
|
# we divide our result by MB instead of multiplying the lsps value by
|
||
|
|
# MB because lsps may round down, so we round down too
|
||
|
|
self.assertEqual(int(psutil_result.total / MB), total_mb)
|
||
|
|
|
||
|
|
def test_cpu_stats(self):
|
||
|
|
out = sh('/usr/bin/mpstat -a')
|
||
|
|
|
||
|
|
re_pattern = r"ALL\s*"
|
||
|
|
for field in (
|
||
|
|
"min maj mpcs mpcr dev soft dec ph cs ics bound rq "
|
||
|
|
"push S3pull S3grd S0rd S1rd S2rd S3rd S4rd S5rd "
|
||
|
|
"sysc"
|
||
|
|
).split():
|
||
|
|
re_pattern += r"(?P<%s>\S+)\s+" % (field,)
|
||
|
|
matchobj = re.search(re_pattern, out)
|
||
|
|
|
||
|
|
self.assertIsNotNone(
|
||
|
|
matchobj, "mpstat command returned unexpected output"
|
||
|
|
)
|
||
|
|
|
||
|
|
# numbers are usually in the millions so 1000 is ok for tolerance
|
||
|
|
CPU_STATS_TOLERANCE = 1000
|
||
|
|
psutil_result = psutil.cpu_stats()
|
||
|
|
self.assertAlmostEqual(
|
||
|
|
psutil_result.ctx_switches,
|
||
|
|
int(matchobj.group("cs")),
|
||
|
|
delta=CPU_STATS_TOLERANCE,
|
||
|
|
)
|
||
|
|
self.assertAlmostEqual(
|
||
|
|
psutil_result.syscalls,
|
||
|
|
int(matchobj.group("sysc")),
|
||
|
|
delta=CPU_STATS_TOLERANCE,
|
||
|
|
)
|
||
|
|
self.assertAlmostEqual(
|
||
|
|
psutil_result.interrupts,
|
||
|
|
int(matchobj.group("dev")),
|
||
|
|
delta=CPU_STATS_TOLERANCE,
|
||
|
|
)
|
||
|
|
self.assertAlmostEqual(
|
||
|
|
psutil_result.soft_interrupts,
|
||
|
|
int(matchobj.group("soft")),
|
||
|
|
delta=CPU_STATS_TOLERANCE,
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_cpu_count_logical(self):
|
||
|
|
out = sh('/usr/bin/mpstat -a')
|
||
|
|
mpstat_lcpu = int(re.search(r"lcpu=(\d+)", out).group(1))
|
||
|
|
psutil_lcpu = psutil.cpu_count(logical=True)
|
||
|
|
self.assertEqual(mpstat_lcpu, psutil_lcpu)
|
||
|
|
|
||
|
|
def test_net_if_addrs_names(self):
|
||
|
|
out = sh('/etc/ifconfig -l')
|
||
|
|
ifconfig_names = set(out.split())
|
||
|
|
psutil_names = set(psutil.net_if_addrs().keys())
|
||
|
|
self.assertSetEqual(ifconfig_names, psutil_names)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
from psutil.tests.runner import run_from_name
|
||
|
|
|
||
|
|
run_from_name(__file__)
|