# HG changeset patch
# User Augie Fackler <durin42@gmail.com>
# Date 1231428701 0
# Node ID 35a2d07e04e804df75cab8158f29574124c8285a
# Parent  042aafcd7dcedd66f1860fd33ac8f9b4850f92ca
Fix the setup of the test environment, including clearing the datastore
between tests.

diff -r 042aafcd7dce -r 35a2d07e04e8 tests/run.py
--- a/tests/run.py	Thu Jan 08 15:31:33 2009 +0000
+++ b/tests/run.py	Thu Jan 08 15:31:41 2009 +0000
@@ -15,6 +15,24 @@
               ]
 
 import nose
+from nose import plugins
+
+class AppEngineDatastoreClearPlugin(plugins.Plugin):
+  """Nose plugin to clear the AppEngine datastore between tests.
+  """
+  name = 'AppEngineDatastoreClearPlugin'
+  enabled = True
+  def options(self, parser, env):
+    return plugins.Plugin.options(self, parser, env)
+
+  def configure(self, parser, env):
+    plugins.Plugin.configure(self, parser, env)
+    self.enabled = True
+
+  def afterTest(self, test):
+    from google.appengine.api import apiproxy_stub_map
+    datastore = apiproxy_stub_map.apiproxy.GetStub('datastore')
+    datastore.Clear()
 
 
 def main():
@@ -34,11 +52,12 @@
                                           urlfetch_stub.URLFetchServiceStub())
   apiproxy_stub_map.apiproxy.RegisterStub('user',
                                           user_service_stub.UserServiceStub())
-  apiproxy_stub_map.apiproxy.RegisterStub('datastore_v3',
-    datastore_file_stub.DatastoreFileStub('your_app_id', '/dev/null',
-                                          '/dev/null'))
+  apiproxy_stub_map.apiproxy.RegisterStub('datastore',
+    datastore_file_stub.DatastoreFileStub('your_app_id', None, None))
   apiproxy_stub_map.apiproxy.RegisterStub('mail', mail_stub.MailServiceStub())
-  nose.main()
+  import django.test.utils
+  django.test.utils.setup_test_environment()
+  nose.main(plugins=[AppEngineDatastoreClearPlugin(), ])
 
 
 if __name__ == '__main__':