Add the unitest.py file

1 files changed, 238 insertions, 0 deletions

diff --git a/unitest.py b/unitest.py
new file mode 100755
index 00000000..4f319a26
--- /dev/null
+++ b/unitest.py
@@ -0,0 +1,238 @@
+#!/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()