Restore sys.stdout for db delete prompt

jazzband · Oct 7, 2015 · b8568ed · b8568ed
1 parent a4034f8
commit b8568ed
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/django_nose/plugin.py b/django_nose/plugin.py
@@ -363,7 +363,13 @@ def __init__(self, tests, runner):
 
     def setup(self):
         """Setup database."""
-        self.old_names = self.runner.setup_databases()
+        # temporarily restore sys.stdout in case of propmt to delete database
+        test_stdout = sys.stdout
+        sys.stdout = sys.__stdout__
+        try:
+            self.old_names = self.runner.setup_databases()
+        finally:
+            sys.stdout = test_stdout
 
     def teardown(self):
         """Tear down database."""