paula.testing/src/paula/testing/mock.py

   1 # Copyright (c) 2008 by Florian Friesdorf
   2 #
   3 # GNU Affero General Public License (AGPL)
   4 #
   5 # This program is free software; you can redistribute it and/or modify
   6 # it under the terms of the GNU Affero General Public License as
   7 # published by the Free Software Foundation; either version 3 of the
   8 # License, or (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13 # GNU Affero General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU Affero General Public
  16 # License along with this program.  If not, see
  17 # <http://www.gnu.org/licenses/>.
  18 """
  19 """
  20 __author__ = "Florian Friesdorf <flo@chaoflow.net>"
  21 __docformat__ = "plaintext"
  22
  23 import types
  24
  25 from zope.interface import alsoProvides, implements, Interface
  26
  27 class Mock(object):
  28     """a mock object that carries desired interfaces
  29
  30         >>> class IA(Interface):
  31         ...     pass
  32         >>> class IB(Interface):
  33         ...     pass
  34         >>> class IC(Interface):
  35         ...     pass
  36
  37         >>> m = Mock( a = 1, f = lambda : 2, alsoProvides=(IA,IB))
  38         >>> m.a
  39         1
  40         >>> m.f()
  41         2
  42         >>> IA.providedBy(m)
  43         True
  44         >>> IB.providedBy(m)
  45         True
  46         >>> m = Mock( a = 1, f = lambda : 2, alsoProvides=IC)
  47         >>> IC.providedBy(m)
  48         True
  49     """
  50     implements(Interface)
  51
  52     def __init__(self, **kws):
  53         try:
  54             alsoprovides = kws['alsoProvides']
  55         except KeyError:
  56             pass
  57         else:
  58             if type(alsoprovides) is types.TupleType:
  59                 alsoProvides(self, *alsoprovides)
  60             else:
  61                 alsoProvides(self, alsoprovides)
  62             del kws['alsoProvides']
  63
  64         for k,v in kws.items():
  65             setattr(self, k, v)