glances/unitest.py
2014-05-24 15:11:53 +02:00

239 lines
7.6 KiB
Python
Executable File

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#
# Glances - An eye on your system
#
# Copyright (C) 2014 Nicolargo <nicolas@nicolargo.com>
#
# Glances is free software; you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# Glances is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
Glances unitary tests suite...
"""
import os
import sys
import time
import gettext
import locale
import unittest
from glances.core.glances_globals import (
__appname__,
__version__,
is_bsd,
is_linux,
is_mac,
is_py3,
is_windows,
sys_prefix,
work_path
)
# Global variables
#=================
# Unitary test is only available from a GNU/Linus machine
if not is_linux:
print('ERROR: Unitaries tests should be ran on GNU/Linux operating system')
sys.exit(2)
else:
print('{} {} {}'.format('Unitary tests for', __appname__, __version__))
# Import local settings
from glances.core.glances_globals import gettext_domain, locale_dir
locale.setlocale(locale.LC_ALL, '')
gettext.install(gettext_domain, locale_dir)
# Init Glances core
from glances.core.glances_main import GlancesMain
core = GlancesMain()
if not core.is_standalone():
print('ERROR: Glances core should be ran in standalone mode')
sys.exit(1)
# Init Glances stats
from glances.core.glances_stats import GlancesStats
stats = GlancesStats()
# Unitest class
#==============
class testGlances(unittest.TestCase):
"""
Test glances class
"""
def setUp(self):
"""
This function is called *every time* before test_*
"""
print('\n' + '='*78)
def test_000_update(self):
"""
Update stats (mandatory step for all the stats)
The update is made twice (for rate computation)
"""
print('INFO: [TEST_000] Test the stats update function')
try:
stats.update()
except:
print('ERROR: Stats update failed')
self.assertTrue(False)
time.sleep(1)
try:
stats.update()
except:
print('ERROR: Stats update failed')
self.assertTrue(False)
self.assertTrue(True)
def test_001_plugins(self):
"""
Check mandatory plugins
"""
plug_to_check = [ 'system', 'cpu', 'load', 'mem', 'memswap', 'network', 'diskio', 'fs' ]
print('INFO: [TEST_001] Check the mandatory plugins list: %s' % ', '.join(plug_to_check))
plug_list = stats.getAllPlugins()
for p in plug_to_check:
self.assertTrue(p in plug_list)
def test_002_cpu(self):
"""
Check SYSTEM plugin
"""
stats_to_check = [ 'hostname', 'os_name' ]
print('INFO: [TEST_002] Check SYSTEM stats: %s' % ', '.join(stats_to_check))
stats_grab = stats.get_plugin('system').get_raw()
for s in stats_to_check:
# Check that the key exist
self.assertTrue(stats_grab.has_key(s), msg='Can not find key: %s' % s)
print('INFO: SYSTEM stats: %s' % stats_grab)
def test_003_cpu(self):
"""
Check CPU plugin
"""
stats_to_check = [ 'system', 'user', 'idle' ]
print('INFO: [TEST_003] Check mandatory CPU stats: %s' % ', '.join(stats_to_check))
stats_grab = stats.get_plugin('cpu').get_raw()
for s in stats_to_check:
# Check that the key exist
self.assertTrue(stats_grab.has_key(s), msg='Can not find key: %s' % s)
# Check that % is > 0 and < 100
self.assertGreaterEqual(stats_grab[s], 0)
self.assertLessEqual(stats_grab[s], 100)
print('INFO: CPU stats: %s' % stats_grab)
def test_004_load(self):
"""
Check LOAD plugin
"""
stats_to_check = [ 'cpucore', 'min1', 'min5', 'min15' ]
print('INFO: [TEST_004] Check LOAD stats: %s' % ', '.join(stats_to_check))
stats_grab = stats.get_plugin('load').get_raw()
for s in stats_to_check:
# Check that the key exist
self.assertTrue(stats_grab.has_key(s), msg='Can not find key: %s' % s)
# Check that % is > 0
self.assertGreaterEqual(stats_grab[s], 0)
print('INFO: LOAD stats: %s' % stats_grab)
def test_005_mem(self):
"""
Check MEM plugin
"""
stats_to_check = [ 'available', 'used', 'free', 'total' ]
print('INFO: [TEST_005] Check MEM stats: %s' % ', '.join(stats_to_check))
stats_grab = stats.get_plugin('mem').get_raw()
for s in stats_to_check:
# Check that the key exist
self.assertTrue(stats_grab.has_key(s), msg='Can not find key: %s' % s)
# Check that % is > 0
self.assertGreaterEqual(stats_grab[s], 0)
print('INFO: MEM stats: %s' % stats_grab)
def test_006_swap(self):
"""
Check MEMSWAP plugin
"""
stats_to_check = [ 'used', 'free', 'total' ]
print('INFO: [TEST_006] Check SWAP stats: %s' % ', '.join(stats_to_check))
stats_grab = stats.get_plugin('memswap').get_raw()
for s in stats_to_check:
# Check that the key exist
self.assertTrue(stats_grab.has_key(s), msg='Can not find key: %s' % s)
# Check that % is > 0
self.assertGreaterEqual(stats_grab[s], 0)
print('INFO: SWAP stats: %s' % stats_grab)
def test_007_network(self):
"""
Check NETWORK plugin
"""
print('INFO: [TEST_007] Check NETWORK stats')
stats_grab = stats.get_plugin('network').get_raw()
self.assertTrue(type(stats_grab) is list, msg='Network stats is not a list')
print('INFO: NETWORK stats: %s' % stats_grab)
def test_008_diskio(self):
"""
Check DISKIO plugin
"""
print('INFO: [TEST_008] Check DiskIO stats')
stats_grab = stats.get_plugin('diskio').get_raw()
self.assertTrue(type(stats_grab) is list, msg='DiskIO stats is not a list')
print('INFO: diskio stats: %s' % stats_grab)
def test_009_fs(self):
"""
Check FileSystem plugin
"""
stats_to_check = [ ]
print('INFO: [TEST_009] Check FS stats')
stats_grab = stats.get_plugin('fs').get_raw()
self.assertTrue(type(stats_grab) is list, msg='FileSystem stats is not a list')
print('INFO: FS stats: %s' % stats_grab)
def test_010_processes(self):
"""
Check Process plugin
"""
stats_to_check = [ ]
print('INFO: [TEST_010] Check PROCESS stats')
stats_grab = stats.get_plugin('processcount').get_raw()
total = stats_grab['total']
self.assertTrue(type(stats_grab) is dict, msg='Process count stats is not a dict')
print('INFO: PROCESS count stats: %s' % stats_grab)
stats_grab = stats.get_plugin('processlist').get_raw()
self.assertTrue(type(stats_grab) is list, msg='Process count stats is not a list')
print('INFO: PROCESS list stats: %s items in the list' % len(stats_grab))
# Check if number of processes in the list equal counter
# self.assertEqual(total, len(stats_grab))
if __name__ == '__main__':
unittest.main()