more correct stderr check

ajslater · Feb 5, 2022 · e61f0c5 · e61f0c5
1 parent df114f4
commit e61f0c5
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/tests/xapian_tests/tests/test_management_commands.py b/tests/xapian_tests/tests/test_management_commands.py
@@ -1,5 +1,6 @@
 import datetime
 import random
+import sys
 from decimal import Decimal
 from io import StringIO
 from unittest import TestCase
@@ -13,7 +14,7 @@
 
 class ManagementCommandTestCase(HaystackBackendTestCase, TestCase):
 
-    NUM_BLOG_ENTRIES = 20
+    NUM_BLOG_ENTRIES = 75
 
     def get_index(self):
         return BlogSearchIndex()
@@ -72,16 +73,16 @@ def test_multiprocessing(self):
         call_command("clear_index", interactive=False, verbosity=0)
         self.verify_indexed_document_count(0)
 
-        out = StringIO()
-        err = StringIO()
+        old_stderr = sys.stderr
+        sys.stderr = StringIO()
         call_command(
             "update_index",
             verbosity=2,
-            workers=2,
+            workers=10,
             batchsize=5,
-            stdout=out,
-            stderr=err,
         )
-        self.assertNotIn("xapian.DatabaseLockError", err.getvalue())
-        self.assertNotIn("xapian.DatabaseLockError", out.getvalue())
+        err = sys.stderr.getvalue()
+        sys.stderr = old_stderr
+        print(err)
+        self.assertNotIn("xapian.DatabaseLockError", err)
         self.verify_indexed_documents()