1.1 --- a/tests/test_handle.py Sat Jul 25 19:29:44 2015 +0200
1.2 +++ b/tests/test_handle.py Sat Jul 25 23:18:58 2015 +0200
1.3 @@ -19,26 +19,19 @@
1.4 this program. If not, see <http://www.gnu.org/licenses/>.
1.5 """
1.6
1.7 -from imiptools.client import Client
1.8 +from imiptools.client import ClientForObject
1.9 from imiptools.data import Object, get_address
1.10 -from imiptools.handlers import Handler
1.11 from imiptools.mail import Messenger
1.12 import imip_store
1.13 import sys
1.14
1.15 -class TestHandler(Handler):
1.16 +class TestClient(ClientForObject):
1.17
1.18 """
1.19 A content handler for use in testing, as opposed to operating within the
1.20 mail processing pipeline.
1.21 """
1.22
1.23 - def __init__(self, obj, user, messenger):
1.24 - Handler.__init__(self, messenger=messenger)
1.25 - Client.__init__(self, user) # this redefines the Handler initialisation
1.26 -
1.27 - self.set_object(obj)
1.28 -
1.29 # Action methods.
1.30
1.31 def handle_request(self, accept):
1.32 @@ -55,7 +48,7 @@
1.33 if not attendee_attr:
1.34 return None
1.35
1.36 - # NOTE: This is a simpler form of the code in imipweb.handler.
1.37 + # NOTE: This is a simpler form of the code in imipweb.client.
1.38
1.39 organiser = get_address(self.obj.get_value("ORGANIZER"))
1.40
1.41 @@ -91,7 +84,7 @@
1.42 sys.exit(1)
1.43
1.44 obj = Object(fragment)
1.45 - handler = TestHandler(obj, user, Messenger())
1.46 + handler = TestClient(obj, user, Messenger())
1.47 response = handler.handle_request(accept == "accept")
1.48
1.49 if response: