mumo3/mumo_manager_test.py

#!/usr/bin/env python2
# -*- coding: utf-8

# Copyright (C) 2010 Stefan Hacker <dd0t@users.sourceforge.net>
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:

# - Redistributions of source code must retain the above copyright notice,
#   this list of conditions and the following disclaimer.
# - Redistributions in binary form must reproduce the above copyright notice,
#   this list of conditions and the following disclaimer in the documentation
#   and/or other materials provided with the distribution.
# - Neither the name of the Mumble Developers nor the names of its
#   contributors may be used to endorse or promote products derived from this
#   software without specific prior written permission.

# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
# `AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
# A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

import unittest
import Queue
from mumo_manager import MumoManager, MumoManagerRemote
from mumo_module import MumoModule
from logging import basicConfig, ERROR, getLogger
import logging
from threading import Event


class MumoManagerTest(unittest.TestCase):
    def setUp(self):
        l = getLogger("MumoManager")
        l.disabled = True
        
        class MyModule(MumoModule):
            def __init__(self, name, manager, configuration = None):
                MumoModule.__init__(self, name, manager, configuration)
                
                self.estarted = Event()
                self.estopped = Event()
                self.econnected = Event()
                self.edisconnected = Event()
                
                self.emeta = Event()
                self.econtext = Event()
                self.eserver = Event()
            
            def onStart(self):
                self.estarted.set()
            
            def onStop(self):
                self.estopped.set()
            
            def connected(self):
                man = self.manager()
                man.subscribeMetaCallbacks(self)
                man.subscribeServerCallbacks(self)
                man.subscribeContextCallbacks(self)
                self.econnected.set()
            
            def disconnected(self):
                self.edisconnected.set()
                
            def metaCallMe(self, arg1, arg2):
                if arg1 == "arg1" and arg2 == "arg2":
                    self.emeta.set()
                    
            def contextCallMe(self, server, arg1, arg2):
                if arg1 == "arg1" and arg2 == "arg2":
                    self.econtext.set()
                    
            def serverCallMe(self, server, arg1, arg2):
                if arg1 == "arg1" and arg2 == "arg2":
                    self.eserver.set()
        
        self.mymod = MyModule

        class conf(object):
            pass # Dummy class
        
        self.cfg = conf()
        self.cfg.test = 10

    #
    #--- Helpers for independent test env creation
    #
    def up(self):
        man = MumoManager(None)
        man.start()

        mod = man.loadModuleCls("MyModule", self.mymod, self.cfg)
        man.startModules()
        
        return (man, mod)
    
    def down(self, man, mod):
        man.stopModules()
        man.stop()
        man.join(timeout=1)
    
    #
    #--- Tests
    #
    def testModuleStarted(self):
        man, mod = self.up()
        
        mod.estarted.wait(timeout=1)
        assert(mod.estarted.is_set())
        
        self.down(man, mod)
    
    def testModuleStopStart(self):
        man ,mod = self.up()
        
        tos = ["MyModule"]
        self.assertEquals(list(man.stopModules(tos).iterkeys()), tos)
        mod.estopped.wait(timeout=1)
        assert(mod.estopped.is_set())
        
        self.down(man, mod)

    def testModuleConnectAndDisconnect(self):
        man, mod = self.up()
        
        man.announceConnected()
        mod.econnected.wait(timeout=1)
        assert(mod.econnected.is_set())
        man.announceDisconnected()
        mod.edisconnected.wait(timeout=1)
        assert(mod.edisconnected.is_set())
        
        self.down(man, mod)
        
    def testMetaCallback(self):
        man, mod = self.up()
        man.announceConnected()
        mod.econnected.wait(timeout=1)
        assert(mod.econnected.is_set())
        man.announceMeta(man.MAGIC_ALL, "metaCallMe", "arg1", arg2 = "arg2")
        mod.emeta.wait(timeout=1)
        assert(mod.emeta.is_set())
        man.announceDisconnected()
        self.down(man, mod)
    
    def testContextCallback(self):
        man, mod = self.up()
        man.announceConnected()
        mod.econnected.wait(timeout=1)
        assert(mod.econnected.is_set())
        man.announceContext(man.MAGIC_ALL, "contextCallMe", "server", "arg1", arg2 = "arg2")
        mod.econtext.wait(timeout=1)
        assert(mod.econtext.is_set())
        man.announceDisconnected()
        self.down(man, mod)
    
    def testServerCallback(self):
        man, mod = self.up()
        man.announceConnected()
        mod.econnected.wait(timeout=1)
        assert(mod.econnected.is_set())
        man.announceServer(man.MAGIC_ALL, "serverCallMe", "server", "arg1", arg2 = "arg2")
        mod.eserver.wait(timeout=1)
        assert(mod.eserver.is_set())
        man.announceDisconnected()
        self.down(man, mod)
        
    def tearDown(self):
        pass
        

if __name__ == "__main__":
    #import sys;sys.argv = ['', 'Test.testName']
    unittest.main()