# HG changeset patch
# User Sverre Rabbelier <srabbelier@gmail.com>
# Date 1252946572 -7200
# Node ID 4f821debda338e07c301c59cb77d4875beca0491
# Parent  764b8812d3e812975b12a39a054f39b0e1a32825
Add a start and end method to MockRequest

diff -r 764b8812d3e8 -r 4f821debda33 tests/test_utils.py
--- a/tests/test_utils.py	Sun Sep 13 22:48:10 2009 +0200
+++ b/tests/test_utils.py	Mon Sep 14 18:42:52 2009 +0200
@@ -19,12 +19,42 @@
 
 __authors__ = [
   '"Augie Fackler" <durin42@gmail.com>',
+  '"Sverre Rabbelier" <sverre@rabbelier.nl>',
   ]
 
 
+from soc.modules import callback
+
+
 class MockRequest(object):
   """Shared dummy request object to mock common aspects of a request.
+
+  Before using the object, start should be called, when done (and
+  before calling start on a new request), end should be called.
   """
+
   def __init__(self, path=None):
-    self.REQUEST = self.GET = self.POST = {}
+    """Creates a new empty request object.
+
+    self.REQUEST, self.GET and self.POST are set to an empty
+    dictionary, and path to the value specified.
+    """
+
+    self.REQUEST = {}
+    self.GET = {}
+    self.POST = {}
     self.path = path
+
+  def start(self):
+    """Readies the core for a new request.
+    """
+
+    core = callback.getCore()
+    core.startNewRequest(self)
+
+  def end(self):
+    """Finishes up the current request.
+    """
+
+    core = callback.getCore()
+    core.endRequest(self, False)