#!/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()
