glances/unitest-restful.py

219 lines
6.5 KiB
Python
Raw Normal View History

2014-07-09 15:50:25 +04:00
#!/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 for the RESTFul API."""
import sys
import time
import unittest
import shlex
import subprocess
import requests
import json
import types
from glances.core.glances_globals import (
appname,
is_linux,
version
)
SERVER_PORT = 61234
URL = "http://localhost:%s/api/2" % SERVER_PORT
pid = None
# Global variables
# =================
# Unitary test is only available from a GNU/Linus machine
if not is_linux:
print(
'ERROR: RESTFul API unitaries tests should be ran on GNU/Linux operating system')
2014-07-09 15:50:25 +04:00
sys.exit(2)
else:
print('Unitary tests for {0} {1}'.format(appname, version))
# 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):
"""The function is called *every time* before test_*."""
print('\n' + '=' * 78)
def test_000_start_server(self):
"""Start the Glances Web Server"""
print('INFO: [TEST_000] Start the Glances Web Server')
2014-07-09 15:50:25 +04:00
global pid
cmdline = "/usr/bin/env python -m glances -w -p %s" % SERVER_PORT
2014-07-09 15:50:25 +04:00
print("Run the Glances Web Server on port %s" % SERVER_PORT)
args = shlex.split(cmdline)
pid = subprocess.Popen(args)
print("Please wait...")
time.sleep(3)
2014-07-09 15:50:25 +04:00
self.assertTrue(pid is not None)
def test_001_all(self):
"""All"""
method = "all"
print('INFO: [TEST_001] Get all stats')
2014-07-09 15:50:25 +04:00
print("HTTP RESTFul request: %s/%s" % (URL, method))
req = requests.get("%s/%s" % (URL, method))
2014-07-09 15:50:25 +04:00
self.assertTrue(req.ok)
def test_002_pluginslist(self):
"""Plugins list"""
method = "pluginslist"
print('INFO: [TEST_002] Plugins list')
print("HTTP RESTFul request: %s/%s" % (URL, method))
req = requests.get("%s/%s" % (URL, method))
self.assertTrue(req.ok)
2014-07-12 20:02:31 +04:00
self.assertIsInstance(req.json(), types.ListType)
self.assertIn('cpu', req.json())
2014-07-09 15:50:25 +04:00
def test_003_plugins(self):
"""Plugins"""
method = "pluginslist"
2014-07-09 15:50:25 +04:00
print('INFO: [TEST_003] Plugins')
plist = requests.get("%s/%s" % (URL, method))
2014-07-12 20:02:31 +04:00
for p in plist.json():
2014-07-09 15:50:25 +04:00
print("HTTP RESTFul request: %s/%s" % (URL, p))
req = requests.get("%s/%s" % (URL, p))
self.assertTrue(req.ok)
if p in ('uptime', 'now'):
2014-07-12 20:02:31 +04:00
self.assertIsInstance(req.json(), types.UnicodeType)
2014-07-09 15:50:25 +04:00
elif p in ('fs', 'monitor', 'percpu', 'sensors', 'alert', 'processlist', 'diskio', 'hddtemp', 'batpercent', 'network'):
2014-07-12 20:02:31 +04:00
self.assertIsInstance(req.json(), types.ListType)
2014-07-09 15:50:25 +04:00
elif p in ('psutilversion', 'help'):
pass
else:
2014-07-12 20:02:31 +04:00
self.assertIsInstance(req.json(), types.DictType)
2014-07-09 15:50:25 +04:00
def test_004_items(self):
"""Items"""
method = "cpu"
print('INFO: [TEST_004] Items for the CPU method')
ilist = requests.get("%s/%s" % (URL, method))
2014-07-12 20:02:31 +04:00
for i in ilist.json():
print("HTTP RESTFul request: %s/%s/%s" % (URL, method, i))
2014-07-09 15:50:25 +04:00
req = requests.get("%s/%s/%s" % (URL, method, i))
self.assertTrue(req.ok)
2014-07-12 20:02:31 +04:00
self.assertIsInstance(req.json(), types.DictType)
self.assertIsInstance(req.json()[i], types.FloatType)
2014-07-09 15:50:25 +04:00
def test_005_values(self):
"""Valuess"""
method = "processlist"
print('INFO: [TEST_005] Item=Value for the PROCESSLIST method')
print("%s/%s/pid/0" % (URL, method))
req = requests.get("%s/%s/pid/0" % (URL, method))
self.assertTrue(req.ok)
2014-07-12 20:02:31 +04:00
self.assertIsInstance(req.json(), types.DictType)
2014-07-09 15:50:25 +04:00
def test_006_all_limits(self):
"""All"""
method = "all/limits"
print('INFO: [TEST_006] Get all limits')
print("HTTP RESTFul request: %s/%s" % (URL, method))
req = requests.get("%s/%s" % (URL, method))
self.assertTrue(req.ok)
self.assertIsInstance(req.json(), types.DictType)
def test_007_all_views(self):
"""All"""
method = "all/views"
print('INFO: [TEST_007] Get all views')
print("HTTP RESTFul request: %s/%s" % (URL, method))
req = requests.get("%s/%s" % (URL, method))
self.assertTrue(req.ok)
self.assertIsInstance(req.json(), types.DictType)
def test_008_plugins_limits(self):
"""Plugins limits"""
method = "pluginslist"
print('INFO: [TEST_008] Plugins limits')
plist = requests.get("%s/%s" % (URL, method))
for p in plist.json():
print("HTTP RESTFul request: %s/%s/limits" % (URL, p))
req = requests.get("%s/%s/limits" % (URL, p))
self.assertTrue(req.ok)
self.assertIsInstance(req.json(), types.DictType)
def test_009_plugins_views(self):
"""Plugins views"""
method = "pluginslist"
print('INFO: [TEST_009] Plugins views')
plist = requests.get("%s/%s" % (URL, method))
for p in plist.json():
print("HTTP RESTFul request: %s/%s/views" % (URL, p))
req = requests.get("%s/%s/views" % (URL, p))
self.assertTrue(req.ok)
self.assertIsInstance(req.json(), types.DictType)
2014-07-09 15:50:25 +04:00
def test_999_stop_server(self):
"""Stop the Glances Web Server"""
print('INFO: [TEST_999] Stop the Glances Web Server')
print("Stop the Glances Web Server")
pid.terminate()
print("Please wait...")
time.sleep(1)
self.assertTrue(True)
if __name__ == '__main__':
unittest.main()