project commit and dir tree change

1 files changed, 153 insertions, 0 deletions

diff --git a/Monitoring/MonitoringService/Driver/test.py b/Monitoring/MonitoringService/Driver/test.py
new file mode 100644
index 0000000..4395750
--- /dev/null
+++ b/Monitoring/MonitoringService/Driver/test.py
@@ -0,0 +1,153 @@
+'''
+Created on Aug 10, 2011
+
+@author: steger
+'''
+import unittest
+from random import randint
+from Example.Resources import PLnodes
+from threading import Thread
+from Example.credentials import noviCredential, sonomaCredential
+from SshExec import SshExec
+from SOAPClient import SOAPClient
+from LocalExec import LocalExec
+from logging import FileHandler
+import logging
+from time import sleep
+from SshTunnel import SshExecTunnel
+
+class Test(unittest.TestCase):
+    testnodes = map(lambda x: x.get_ipaddress("eth0")[0], PLnodes)
+    cred_novi = noviCredential
+    url_sonoma = "http://complex.elte.hu/~steger/sonoma/user.wsdl"
+    cred_sonoma = sonomaCredential
+    
+
+    def setUp(self):
+        pass
+
+    def tearDown(self):
+        pass
+    
+    def gettestnode(self):
+        '''
+        @summary: Return a test node IP address chosen random
+        @return: ip address
+        @rtype: string 
+        '''
+        return self.testnodes[randint(1, len(self.testnodes))-1]
+
+    def test_helloworld(self):
+        '''
+        @summary: Execute local command
+        '''
+        proc = LocalExec()
+        result = proc.execute()
+        self.assertTrue(result.startswith("helloworld"), "Local command output differs from expected")
+
+    def test_helloworldWithMaster(self):
+        '''
+        @summary: Execute remote command in the name of the testuser authenticated with the master key
+        '''
+        proc = SshExec(host = self.gettestnode(), credential = self.cred_novi)
+        result = proc.execute().read()
+        self.assertTrue(result.startswith("helloworld @ "), "Remote command output differs from expected")
+
+    def echoWithMaster(self, address):
+        '''
+        @summary: Execute remote echo command in the name of the testuser authenticated with the master key
+        @param address: ip address of the remote machine
+        @type address: string  
+        '''
+        try:
+            n = randint(0, 10000)
+            command = "echo %d" % n
+            proc = SshExec(host = address, credential = self.cred_novi, command = command)
+            result = proc.execute().read()
+            self.assertTrue(result.strip() == str(n), "Remote command @%s output differs from expected: (%s != %d)" % (address, result, n))
+        except Exception, e:
+            self.assertFalse(True, "Got an error %s" % e)
+
+    def test_echoWithMaster(self):
+        '''
+        @summary: Execute remote echo command in the name of the testuser authenticated with the master key
+        '''
+        self.echoWithMaster(self.gettestnode())
+        
+    def test_distributedEcho(self):
+        '''
+        @summary: Execute parallel remote echo commands in a distributed fashion
+        '''
+        threads = []
+        for n in self.testnodes:
+            t = Thread(target = self.echoWithMaster, args = (n,))
+            t.daemon = True
+            t.start()
+            threads.append(t)
+        while len(threads):
+            t = threads.pop()
+            t.join(5)
+
+    def test_parallelEcho(self):
+        '''
+        @summary: Execute parallel remote echo commands in a test node
+        '''
+        N = 20
+        n = self.gettestnode()
+        threads = []
+        while N:
+            N -= 1
+            t = Thread(target = self.echoWithMaster, args = (n,))
+            t.daemon = True
+            t.start()
+            threads.append(t)
+        while len(threads):
+            t = threads.pop()
+            t.join(5)
+
+    def test_stress(self):
+        '''
+        @summary: Consecutively execute parallel remote echo commands in a distributed fashion
+        '''
+        threads = []
+        for n in self.testnodes:
+            N = randint(5, 20)
+            while N:
+                N -= 1
+                t = Thread(target = self.echoWithMaster, args = (n,))
+                t.daemon = True
+                t.start()
+                threads.append(t)
+        while len(threads):
+            t = threads.pop()
+            t.join(5)
+    
+    def test_soap(self):
+        '''
+        @summary: Run SONoMA getNodeList
+        '''
+        client = SOAPClient(self.url_sonoma)
+        resources = client.service.getNodeList(filter = "AVAILABLE")
+        self.assertGreater(len(resources), 0, "sonoma reports no nodes") 
+        
+    def test_tunnel(self):
+        T = SshExecTunnel(host = 'novilab.elte.hu', credential = noviCredential, localport = 4000, port = 22,
+                           remoteserver = 'smilax1.man.poznan.pl', remoteport = 22)
+        response = T.execute().read().strip()
+        expected = "helloworld @ smilax1.man.poznan.pl"
+        self.assertEqual(response, expected, "Remote command output differs from expected %s != %s"  % (expected, response))
+
+if __name__ == "__main__":
+    #import sys;sys.argv = ['', 'Test.test_helloworldWithMaster']
+    fn = "/tmp/Driver_test.log"
+    hdlr = FileHandler(filename = fn, mode = 'w')
+    l = logging.getLogger("NOVI.DRIVER")
+    l.setLevel(level = logging.DEBUG)
+#    l.setLevel(level = logging.INFO)
+    l.addHandler(hdlr = hdlr)
+    l.info("START TEST")
+    try:
+        unittest.main()
+    finally:
+        l.info("FINISHED TEST")
+        hdlr.close()