diff options
| author | mbligh <mbligh@592f7852-d20e-0410-864c-8624ca9c26a4> | 2007-11-29 15:57:02 +0000 |
|---|---|---|
| committer | mbligh <mbligh@592f7852-d20e-0410-864c-8624ca9c26a4> | 2007-11-29 15:57:02 +0000 |
| commit | fb5705cc4daa30a05b537a6086132bcc4233790e (patch) | |
| tree | da850e1438553598de7ef898cd965209b76fd1b1 /client/bin/autotest_utils.py | |
| parent | 00f06ea8fbf40e1d4690ad4edf84531caf366d8b (diff) | |
Add miscellaneous helpers to autotest_utils.py
From: Jiqing Tang <jiqingtang@google.com>
git-svn-id: svn://test.kernel.org/autotest/trunk@998 592f7852-d20e-0410-864c-8624ca9c26a4
Diffstat (limited to 'client/bin/autotest_utils.py')
| -rwxr-xr-x | client/bin/autotest_utils.py | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/client/bin/autotest_utils.py b/client/bin/autotest_utils.py index 04a74eb6..0f0be2e6 100755 --- a/client/bin/autotest_utils.py +++ b/client/bin/autotest_utils.py @@ -576,11 +576,54 @@ def get_cpu_family(): else: raise TestError('Could not get valid cpu family data') + def get_disks(): df_output = system_output('df') disk_re = re.compile(r'^(/dev/hd[a-z]+)3', re.M) return disk_re.findall(df_output) + +def load_module(module_name): + # Checks if a module has already been loaded + if module_is_loaded(module_name): + return False + + system('/sbin/modprobe ' + module_name) + return True + + +def unload_module(module_name): + system('/sbin/rmmod ' + module_name) + + +def module_is_loaded(module_name): + modules = system_output('/sbin/lsmod').splitlines() + for module in modules: + if module.startswith(module_name) and module[len(module_name)] == ' ': + return True + return False + + +def get_huge_page_size(): + output = system_output('grep Hugepagesize /proc/meminfo') + return int(output.split()[1]) # Assumes units always in kB. :( + + +def get_num_huge_pages(): + raw_hugepages = system_output('/sbin/sysctl vm.nr_hugepages') + return int(raw_hugepages.split()[2]) + + +def set_num_huge_pages(num): + system('/sbin/sysctl vm.nr_hugepages=%d' % num) + + +def get_system_nodes(): + nodes = os.listdir('/sys/devices/system/node') + nodes.sort() + return nodes + + try: from site_utils import * except ImportError: |