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.
63 lines
1.7 KiB
Python
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()
|