unittest_aspects.py :  » Aspect-Oriented » logilab-aspects » aspects-0.1.4 » tests » Python Open Source

Home
Python Open Source
1.3.1.2 Python
2.Ajax
3.Aspect Oriented
4.Blog
5.Build
6.Business Application
7.Chart Report
8.Content Management Systems
9.Cryptographic
10.Database
11.Development
12.Editor
13.Email
14.ERP
15.Game 2D 3D
16.GIS
17.GUI
18.IDE
19.Installer
20.IRC
21.Issue Tracker
22.Language Interface
23.Log
24.Math
25.Media Sound Audio
26.Mobile
27.Network
28.Parser
29.PDF
30.Project Management
31.RSS
32.Search
33.Security
34.Template Engines
35.Test
36.UML
37.USB Serial
38.Web Frameworks
39.Web Server
40.Web Services
41.Web Unit
42.Wiki
43.Windows
44.XML
Python Open Source » Aspect Oriented » logilab aspects 
logilab aspects » aspects 0.1.4 » tests » unittest_aspects.py
# -*- coding: ISO-8859-1 -*-
# pylint: disable-msg=C0103,W0613,W0142,W0232,R0801
# Copyright (c) 2003 LOGILAB S.A. (Paris, FRANCE).
# http://www.logilab.fr/ -- mailto:contact@logilab.fr
#
# This program is free software; you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free Software
# Foundation; either version 2 of the License, or (at your option) any later
# version.
#
# This program 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License along with
# this program; if not, write to the Free Software Foundation, Inc.,
# 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

"""
Unit tests for aspects
"""

__revision__ = '$Id: unittest_aspects.py,v 1.23 2005/05/31 10:14:21 adim Exp $'


import unittest
import sys

from logilab.aspects.weaver import weaver

from logilab.aspects.core import AbstractAspect


class EmptyAspect(AbstractAspect):
    """An empty aspect class to test if method wrapping
    is well done.
    """

    before_flag = 1
    around_flag = 2
    after_flag = 4
    proceed_flag = 8
    
    def __init__(self, pointcut, arg1, arg2):
        """arg1 and arg2 are here to test that the aspect can
        be be built with additional args
        """
        AbstractAspect.__init__(self, pointcut)
        self.arg1 = arg1
        self.arg2 = arg2
        # A flag to test if wrapping methods are called
        self.trace_flag = 0
        

    def before(self, wobj, context, *args, **kwargs):
        """Before method
        """
        self.trace_flag |= EmptyAspect.before_flag


    def after(self, wobj, context, *args, **kwargs):
        """After method
        """
        self.trace_flag |= EmptyAspect.after_flag


    def around(self, wobj, context, *args, **kwargs):
        """Around method
        """
        self.trace_flag |= EmptyAspect.around_flag
        method_name = context['method_name']
        wclass = context['__class__']
        self._proceed(wobj, wclass, method_name, *args, **kwargs)
        
        

class Foo:
    """A simple class to tests aspects
    """

    def silly_method(self):
        """Just for tests, does nothing
        """
        pass


    def raise_exception(self):
        """Just for tests, does nothing
        """
        raise ValueError('Yop')



class AspectTC(unittest.TestCase):
    """TestCase for aspects
    """

    def setUp(self):
        """setUp method for tests
        """
        self.foo = Foo()
        weaver.weave_methods(self.foo, EmptyAspect, 10, 20)
        self.aspect = weaver.get_aspects(self.foo, 'silly_method')[0]
        

    def test_aspect_init(self):
        """Tests if aspect's init is well done with additional args
        """
        self.assert_(self.aspect.arg1 == 10)
        self.assert_(self.aspect.arg2 == 20)
        

    def test_method_wrapping(self):
        """Tests if methods are well wrapped
        """
        ok_val = EmptyAspect.before_flag | EmptyAspect.after_flag | \
                 EmptyAspect.around_flag
        self.foo.silly_method()
        self.assert_(self.aspect.trace_flag == ok_val)
        

    def test_exception_management(self):
        """Tests if exception are re-raised
        """
        self.assertRaises(ValueError, self.foo.raise_exception)
        


def suite():
    """return the unitest suite"""
    loader = unittest.TestLoader()
    testsuite = loader.loadTestsFromModule(sys.modules[__name__])
    return testsuite
    
    
def Run(runner=None):
    """run tests"""
    testsuite = suite()
    if runner is None:
        runner = unittest.TextTestRunner()
    return runner.run(testsuite)
    
if __name__ == '__main__':
    Run()

www.java2java.com | Contact Us
Copyright 2009 - 12 Demo Source and Support. All rights reserved.
All other trademarks are property of their respective owners.