Ticket #1330: trac-admin-test-in-memory.patch

trac/scripts/admin.py

@@ -84 +84 @@
     ## Environment methods
     ##
 
-    def env_set(self, envname):
+    def env_set(self, envname, env=None):
         self.envname = envname
         self.prompt = "Trac [%s]> " % self.envname
+        if env is not None:
+            self.__env = env
 
     def env_check(self):
         try:

trac/scripts/tests/admin.py

@@ -86 +86 @@
     def setUp(self):
         EnvironmentTestBase.setUp(self)
         self._admin = admin.TracAdmin()
-        self._admin.env_set(self._get_envpath())
+        self._admin.env_set(self._get_envpath(), self.env)
 
     def _execute(self, cmd):
         try:

trac/tests/environment.py

@@ -1 @@
-from trac import db_default
+from trac import db_default, test
 from trac.Environment import Environment
 
 import os
@@ -6 +6 @@
 import tempfile
 import shutil
 
+"""
+A subclass of trac.Environment that keeps its' DB in memory.
+"""
+class InMemoryEnvironment(Environment):
+    def get_db_cnx(self):
+        if not hasattr(self, '_db'):
+            self._db = test.InMemoryDatabase()
+        return self._db
 
 class EnvironmentTestBase:
 
     def setUp(self):
-        self.env = Environment(self._get_envpath(), create=1)
+        self.env = InMemoryEnvironment(self._get_envpath(), create=1)
         self.env.insert_default_data()
         self.db = self.env.get_db_cnx()