doctest para EmbrulhoRestritor

ramalho · ramalho · commit 51476cb293b1 · 2015-08-15T12:34:38.000-03:00
diff --git a/Martelli/embrulho.py b/Martelli/embrulho.py
@@ -1,3 +1,23 @@
+"""
+A classe EmbrulhoRestritor demonstra o uso de composição para limitar
+o acesso a atributos de instâncias de outra classe qualquer.
+
+    >>> rex = Cao()
+    >>> rex.latir()
+    Au!
+    >>> rex.morder()
+    Nhac!
+    >>> rex2 = EmbrulhoRestritor(rex, ['morder'])
+    >>> rex2.latir()
+    Au!
+    >>> rex2.morder()
+    Traceback (most recent call last):
+        ...
+    AttributeError: atributo bloqueado 'morder'
+
+"""
+
+
 class EmbrulhoRestritor:
 
     def __init__(self, embrulhado, bloqueios):
@@ -6,7 +26,7 @@ def __init__(self, embrulhado, bloqueios):
 
     def __getattr__(self, nome):
         if nome in self._bloqueios:
-            msg = 'atributo bloqueado: {!r}'.format(nome)
+            msg = 'atributo bloqueado {!r}'.format(nome)
             raise AttributeError(msg)
         else:
             return getattr(self._embrulhado, nome)
