Python tests.sh函数代码示例

    def test_virtual_memory(self):
        out = sh('/usr/bin/svmon -O unit=KB')
        re_pattern = "memory\s*"
        for field in ("size inuse free pin virtual available mmode").split():
            re_pattern += "(?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()

        # MEMORY_TOLERANCE 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.
        MEMORY_TOLERANCE = 2 * KB * KB   # 2 MB
        self.assertEqual(psutil_result.total, total)
        self.assertAlmostEqual(
            psutil_result.used, used, delta=MEMORY_TOLERANCE)
        self.assertAlmostEqual(
            psutil_result.available, available, delta=MEMORY_TOLERANCE)
        self.assertAlmostEqual(
            psutil_result.free, free, delta=MEMORY_TOLERANCE)

 def test_get_kernel_version(self):
     if not POSIX:
         self.assertEqual(get_kernel_version(), tuple())
     else:
         kernel = get_kernel_version()
         assert kernel, kernel
         self.assertIn('.'.join(map(str, kernel)), sh("uname -a"))

 def test_process_create_time(self):
     output = sh("ps -o lstart -p %s" % self.pid)
     start_ps = output.replace('STARTED', '').strip()
     start_psutil = psutil.Process(self.pid).create_time()
     start_psutil = time.strftime("%a %b %e %H:%M:%S %Y",
                                  time.localtime(start_psutil))
     self.assertEqual(start_ps, start_psutil)

 def test_memory_maps(self):
     src = textwrap.dedent("""
         import time
         with open("%s", "w") as f:
             time.sleep(10)
         """ % TESTFN)
     sproc = pyrun(src)
     self.addCleanup(reap_children)
     call_until(lambda: os.listdir('.'), "'%s' not in ret" % TESTFN)
     p = psutil.Process(sproc.pid)
     time.sleep(.1)
     maps = p.memory_maps(grouped=False)
     pmap = sh('pmap -x %s' % p.pid).split('\n')
     # get rid of header
     del pmap[0]
     del pmap[0]
     while maps and pmap:
         this = maps.pop(0)
         other = pmap.pop(0)
         addr, _, rss, dirty, mode, path = other.split(None, 5)
         if not path.startswith('[') and not path.endswith(']'):
             self.assertEqual(path, os.path.basename(this.path))
         self.assertEqual(int(rss) * 1024, this.rss)
         # test only rwx chars, ignore 's' and 'p'
         self.assertEqual(mode[:3], this.perms[:3])

    def test_cpu_stats(self):
        out = sh('/usr/bin/mpstat -a')

        re_pattern = "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 += "(?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_swapmem_total(self):
     out = sh('sysctl vm.swapusage')
     out = out.replace('vm.swapusage: ', '')
     total, used, free = re.findall('\d+.\d+\w', out)
     psutil_smem = psutil.swap_memory()
     self.assertEqual(psutil_smem.total, human2bytes(total))
     self.assertEqual(psutil_smem.used, human2bytes(used))
     self.assertEqual(psutil_smem.free, human2bytes(free))

 def test_sensors_battery(self):
     out = sh("pmset -g batt")
     percent = re.search("(\d+)%", out).group(1)
     drawing_from = re.search("Now drawing from '([^']+)'", out).group(1)
     power_plugged = drawing_from == "AC Power"
     psutil_result = psutil.sensors_battery()
     self.assertEqual(psutil_result.power_plugged, power_plugged)
     self.assertEqual(psutil_result.percent, int(percent))

def free_swap():
    """Parse 'free' cmd and return swap memory's s total, used and free
    values.
    """
    lines = sh('free').split('\n')
    for line in lines:
        if line.startswith('Swap'):
            _, total, used, free = line.split()
    return (int(total) * 1024, int(used) * 1024, int(free) * 1024)

 def test_users(self):
     out = sh("who")
     lines = out.split('\n')
     users = [x.split()[0] for x in lines]
     terminals = [x.split()[1] for x in lines]
     self.assertEqual(len(users), len(psutil.users()))
     for u in psutil.users():
         self.assertIn(u.name, users)
         self.assertIn(u.terminal, terminals)

 def df(device):
     out = sh("df -k %s" % device).strip()
     line = out.split('\n')[1]
     fields = line.split()
     total = int(fields[1]) * 1024
     used = int(fields[2]) * 1024
     free = int(fields[3]) * 1024
     percent = float(fields[4].replace('%', ''))
     return (total, used, free, percent)

 def test_nic_names(self):
     out = sh('ipconfig /all')
     nics = psutil.net_io_counters(pernic=True).keys()
     for nic in nics:
         if "pseudo-interface" in nic.replace(' ', '-').lower():
             continue
         if nic not in out:
             self.fail(
                 "%r nic wasn't found in 'ipconfig /all' output" % nic)

def vm_stat(field):
    """Wrapper around 'vm_stat' cmdline utility."""
    out = sh('vm_stat')
    for line in out.split('\n'):
        if field in line:
            break
    else:
        raise ValueError("line not found")
    return int(re.search(r'\d+', line).group(0)) * PAGESIZE

 def test_users(self):
     out = sh("who")
     lines = out.split('\n')
     users = [x.split()[0] for x in lines]
     self.assertEqual(len(users), len(psutil.users()))
     terminals = [x.split()[1] for x in lines]
     for u in psutil.users():
         self.assertTrue(u.name in users, u.name)
         self.assertTrue(u.terminal in terminals, u.terminal)

def muse(field):
    """Thin wrapper around 'muse' cmdline utility."""
    out = sh('muse')
    for line in out.split('\n'):
        if line.startswith(field):
            break
    else:
        raise ValueError("line not found")
    return int(line.split()[1])

 def test_net_if_stats(self):
     for name, stats in psutil.net_if_stats().items():
         try:
             out = sh("ifconfig %s" % name)
         except RuntimeError:
             pass
         else:
             self.assertEqual(stats.isup, 'RUNNING' in out, msg=out)
             self.assertEqual(stats.mtu,
                              int(re.findall(r'mtu (\d+)', out)[0]))