1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159
|
import unittest
from cpuinfo import *
import helpers
class MockDataSource:
bits = '64bit'
cpu_count = 1
is_windows = False
arch_string_raw = 'amd64'
uname_string_raw = 'x86_64'
can_cpuid = False
@staticmethod
def has_dmesg():
return True
@staticmethod
def has_var_run_dmesg_boot():
return True
@staticmethod
def dmesg_a():
retcode = 0
output = r'''Copyright (c) 1992-2014 The FreeBSD Project.
Copyright (c) 1979, 1980, 1983, 1986, 1988, 1989, 1991, 1992, 1993, 1994
The Regents of the University of California. All rights reserved.
FreeBSD is a registered trademark of The FreeBSD Foundation.
FreeBSD 10.0-RELEASE-p17 #0: Tue Sep 16 14:33:46 UTC 2014
root@amd64-builder.pcbsd.org:/usr/obj/usr/src/sys/GENERIC amd64
FreeBSD clang version 3.3 (tags/RELEASE_33/final 183502) 20130610
VT(vga): text 80x25
CPU: Intel(R) Pentium(R) CPU G640 @ 2.80GHz (2793.73-MHz K8-class CPU)
Origin="GenuineIntel" Id=0x206a7 Family=0x6 Model=02a Stepping=7
Features=0x1783fbff<FPU,VME,DE,PSE,TSC,MSR,PAE,MCE,CX8,APIC,SEP,MTRR,PGE,MCA,CMOV,PAT,PSE36,MMX,FXSR,SSE,SSE2,HTT>
Features2=0xc982203<SSE3,PCLMULQDQ,SSSE3,CX16,SSE4.1,SSE4.2,POPCNT,XSAVE,OSXSAVE>
AMD Features=0x28100800<SYSCALL,NX,RDTSCP,LM>
AMD Features2=0x1<LAHF>
TSC: P-state invariant
'''
return retcode, output
@staticmethod
def cat_var_run_dmesg_boot():
retcode = 0
output = r'''
VT(vga): text 80x25
CPU: Intel(R) Pentium(R) CPU G640 @ 2.80GHz (2793.73-MHz K8-class CPU)
Origin="GenuineIntel" Id=0x206a7 Family=0x6 Model=02a Stepping=7
Features=0x1783fbff<FPU,VME,DE,PSE,TSC,MSR,PAE,MCE,CX8,APIC,SEP,MTRR,PGE,MCA,CMOV,PAT,PSE36,MMX,FXSR,SSE,SSE2,HTT>
Features2=0xc982203<SSE3,PCLMULQDQ,SSSE3,CX16,SSE4.1,SSE4.2,POPCNT,XSAVE,OSXSAVE>
AMD Features=0x28100800<SYSCALL,NX,RDTSCP,LM>
AMD Features2=0x1<LAHF>
TSC: P-state invariant
'''
return retcode, output
class TestFreeBSD_11_X86_64(unittest.TestCase):
def setUp(self):
helpers.backup_data_source(cpuinfo)
helpers.monkey_patch_data_source(cpuinfo, MockDataSource)
def tearDown(self):
helpers.restore_data_source(cpuinfo)
'''
Make sure calls return the expected number of fields.
'''
def test_returns(self):
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_registry()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_cpufreq_info()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_lscpu()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_proc_cpuinfo()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_sysctl()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_kstat()))
self.assertEqual(10, len(cpuinfo._get_cpu_info_from_dmesg()))
self.assertEqual(10, len(cpuinfo._get_cpu_info_from_cat_var_run_dmesg_boot()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_ibm_pa_features()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_sysinfo()))
self.assertEqual(0, len(cpuinfo._get_cpu_info_from_cpuid()))
self.assertEqual(17, len(cpuinfo._get_cpu_info_internal()))
def test_get_cpu_info_from_dmesg(self):
info = cpuinfo._get_cpu_info_from_dmesg()
self.assertEqual('GenuineIntel', info['vendor_id_raw'])
self.assertEqual('Intel(R) Pentium(R) CPU G640 @ 2.80GHz', info['brand_raw'])
self.assertEqual('2.8000 GHz', info['hz_advertised_friendly'])
self.assertEqual('2.8000 GHz', info['hz_actual_friendly'])
self.assertEqual((2800000000, 0), info['hz_advertised'])
self.assertEqual((2800000000, 0), info['hz_actual'])
self.assertEqual(7, info['stepping'])
self.assertEqual(42, info['model'])
self.assertEqual(6, info['family'])
self.assertEqual(
['apic', 'cmov', 'cx16', 'cx8', 'de', 'fpu', 'fxsr', 'htt',
'lahf', 'lm', 'mca', 'mce', 'mmx', 'msr', 'mtrr', 'nx',
'osxsave', 'pae', 'pat', 'pclmulqdq', 'pge', 'popcnt', 'pse',
'pse36', 'rdtscp', 'sep', 'sse', 'sse2', 'sse3', 'sse4.1',
'sse4.2', 'ssse3', 'syscall', 'tsc', 'vme', 'xsave']
,
info['flags']
)
def test_get_cpu_info_from_cat_var_run_dmesg_boot(self):
info = cpuinfo._get_cpu_info_from_cat_var_run_dmesg_boot()
self.assertEqual('GenuineIntel', info['vendor_id_raw'])
self.assertEqual('Intel(R) Pentium(R) CPU G640 @ 2.80GHz', info['brand_raw'])
self.assertEqual('2.8000 GHz', info['hz_advertised_friendly'])
self.assertEqual('2.8000 GHz', info['hz_actual_friendly'])
self.assertEqual((2800000000, 0), info['hz_advertised'])
self.assertEqual((2800000000, 0), info['hz_actual'])
self.assertEqual(7, info['stepping'])
self.assertEqual(42, info['model'])
self.assertEqual(6, info['family'])
self.assertEqual(
['apic', 'cmov', 'cx16', 'cx8', 'de', 'fpu', 'fxsr', 'htt',
'lahf', 'lm', 'mca', 'mce', 'mmx', 'msr', 'mtrr', 'nx',
'osxsave', 'pae', 'pat', 'pclmulqdq', 'pge', 'popcnt', 'pse',
'pse36', 'rdtscp', 'sep', 'sse', 'sse2', 'sse3', 'sse4.1',
'sse4.2', 'ssse3', 'syscall', 'tsc', 'vme', 'xsave']
,
info['flags']
)
def test_all(self):
info = cpuinfo._get_cpu_info_internal()
self.assertEqual('GenuineIntel', info['vendor_id_raw'])
self.assertEqual('Intel(R) Pentium(R) CPU G640 @ 2.80GHz', info['brand_raw'])
self.assertEqual('2.8000 GHz', info['hz_advertised_friendly'])
self.assertEqual('2.8000 GHz', info['hz_actual_friendly'])
self.assertEqual((2800000000, 0), info['hz_advertised'])
self.assertEqual((2800000000, 0), info['hz_actual'])
self.assertEqual('X86_64', info['arch'])
self.assertEqual(64, info['bits'])
self.assertEqual(1, info['count'])
self.assertEqual('amd64', info['arch_string_raw'])
self.assertEqual(7, info['stepping'])
self.assertEqual(42, info['model'])
self.assertEqual(6, info['family'])
self.assertEqual(
['apic', 'cmov', 'cx16', 'cx8', 'de', 'fpu', 'fxsr', 'htt',
'lahf', 'lm', 'mca', 'mce', 'mmx', 'msr', 'mtrr', 'nx',
'osxsave', 'pae', 'pat', 'pclmulqdq', 'pge', 'popcnt', 'pse',
'pse36', 'rdtscp', 'sep', 'sse', 'sse2', 'sse3', 'sse4.1',
'sse4.2', 'ssse3', 'syscall', 'tsc', 'vme', 'xsave']
,
info['flags']
)
|