LETTUCE_USE_TEST_DATABASE setting for always running on a test database

lettuce/django/management/commands/harvest.py

@@ -50,7 +50,8 @@ class Command(BaseCommand):
         make_option('-S', '--no-server', action='store_true', dest='no_server', default=False,
             help="will not run django's builtin HTTP server"),
 
-        make_option('-T', '--test-server', action='store_true', dest='test_database', default=False,
+        make_option('-T', '--test-server', action='store_true', dest='test_database',
+            default=getattr(settings, "LETTUCE_USE_TEST_DATABASE", False),
             help="will run django's builtin HTTP server using the test databases"),
 
         make_option('-P', '--port', type='int', dest='port',

tests/integration/django/dill/leaves/features/steps.py

@@ -44,3 +44,7 @@ def database_dump(step):
 @step(r'I have populated the database')
 def database_populated(step):
     pass
+
+@step(r'I count the harvesters')
+def count_harvesters(step):
+    print "Harvester count: %d" % Harvester.objects.count()

tests/integration/django/dill/leaves/features/testdb.feature

+Feature: Test running with the test database
+  Scenario: Test running with the test database
+    Given I have a harvester in the database:
+      | make  |
+      | Frank |
+    Then I count the harvesters

tests/integration/django/dill/testdbsettings.py

+from settings import *
+
+LETTUCE_USE_TEST_DATABASE = True

tests/integration/test_dill.py

@@ -76,3 +76,18 @@ def test_model_existence_check():
     assert "Expected 2 geese, found 1" in out
 
     FileSystem.popd()
+
+def test_use_test_database_setting():
+    'Test database is recreated each time if LETTUCE_USE_TEST_DATABASE is set'
+
+    FileSystem.pushd(current_directory, "django", "dill")
+
+    for i in range(1, 2):
+        status, out = commands.getstatusoutput(
+            "python manage.py harvest --settings=testdbsettings " +
+            "leaves/features/testdb.feature")
+
+        assert_equals(status, 0, out)
+        assert "Harvester count: 1" in out, out
+
+    FileSystem.popd()