You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
smart-top/test/TestServerMonitor.py

63 lines
1.7 KiB
Python

from __future__ import division
import unittest
from monitor import Env, DetailProcess, ProcessCounter, Sandbox
import monitor
import os
import json
from utils.workers import Top, Learner
#from multiprocessing import Lock
from threading import Lock
path = os.environ['MONITOR_CONFIG_PATH']
f = open(path)
CONFIG = json.loads( f.read())
f.close()
class TestMonitorServer(unittest.TestCase):
def test_Environment(self):
"""
This test case is designed to test the existance of a resource set as an environment variable. This applies to files, folders (not values)
"""
p = Env()
p.init(['PATH','HOME','SHELL'])
r = p.composite()
value = r['value']
self.assertTrue(value > 0 and value >= (100*2/3),value)
self.assertTrue(p.evaluate('PATH') == 0)
def test_RunningProcess(self):
p = DetailProcess()
p.init(['kate','rabbitmq-server','python','apache2','firefox'])
r = p.composite()
for row in r:
print row['label'],row['status'],row['proc_count']
self.assertTrue(r)
def test_ProcessCount(self):
p= ProcessCounter()
p.init(['foo','apache2','VBoxClient','rabbitmq-server','python'])
r = p.composite()
self.assertTrue( sum(r.values()) > 0 )
self.assertTrue( r['foo'] == 0)
def test_VirtualEnv(self):
requirements_path = os.sep.join([os.environ['PYTHONPATH'],"..","requirements.txt"])
sandbox_path = os.sep.join([os.environ['PYTHONPATH'],"..",'sandbox'])
p = Sandbox()
p.init({"sandbox":sandbox_path,"requirements":requirements_path})
p.composite()
def test_StartTop(self):
lock = Lock()
p = Top(CONFIG,lock)
#p.start()
#p.join()
def test_StartLearner(self):
lock = Lock()
p = Learner(CONFIG,lock)
p.start()
if __name__ == '__main__' :
unittest.main()