ain't no redis in travis

Author: ewdurbin

tests/unit/search/test_tasks.py

@@ -12,6 +12,8 @@
 
 import os
 
+from contextlib import contextmanager
+
 import celery
 import elasticsearch
 import packaging.version
@@ -132,6 +134,11 @@ def __init__(self):
         self.indices = FakeESIndices()
 
 
+@contextmanager
+def _not_lock(*a, **kw):
+    yield True
+
+
 class TestReindex:
 
     def test_fails_when_raising(self, db_request, monkeypatch):
@@ -168,6 +175,10 @@ def parallel_bulk(client, iterable):
             assert iterable is docs
             raise TestException
 
+        monkeypatch.setattr(
+            redis.StrictRedis, "from_url",
+            lambda *a, **kw: pretend.stub(lock=_not_lock))
+
         monkeypatch.setattr(
             warehouse.search.tasks, "parallel_bulk", parallel_bulk)
 
@@ -233,6 +244,10 @@ def project_docs(db):
             "celery.scheduler_url": "redis://redis:6379/0",
         }
 
+        monkeypatch.setattr(
+            redis.StrictRedis, "from_url",
+            lambda *a, **kw: pretend.stub(lock=_not_lock))
+
         parallel_bulk = pretend.call_recorder(lambda client, iterable: [None])
         monkeypatch.setattr(
             warehouse.search.tasks, "parallel_bulk", parallel_bulk)
@@ -305,6 +320,10 @@ def project_docs(db):
             "celery.scheduler_url": "redis://redis:6379/0",
         }
 
+        monkeypatch.setattr(
+            redis.StrictRedis, "from_url",
+            lambda *a, **kw: pretend.stub(lock=_not_lock))
+
         parallel_bulk = pretend.call_recorder(lambda client, iterable: [None])
         monkeypatch.setattr(
             warehouse.search.tasks, "parallel_bulk", parallel_bulk)
@@ -385,6 +404,10 @@ def parallel_bulk(client, iterable):
             assert iterable is docs
             raise TestException
 
+        monkeypatch.setattr(
+            redis.StrictRedis, "from_url",
+            lambda *a, **kw: pretend.stub(lock=_not_lock))
+
         monkeypatch.setattr(
             warehouse.search.tasks, "parallel_bulk", parallel_bulk)
 
@@ -394,7 +417,7 @@ def parallel_bulk(client, iterable):
         assert es_client.indices.put_settings.calls == []
         assert es_client.indices.forcemerge.calls == []
 
-    def test_unindex_fails_when_raising(self, db_request):
+    def test_unindex_fails_when_raising(self, db_request, monkeypatch):
         class TestException(Exception):
             pass
 
@@ -413,6 +436,10 @@ class TestException(Exception):
             "celery.scheduler_url": "redis://redis:6379/0",
         }
 
+        monkeypatch.setattr(
+            redis.StrictRedis, "from_url",
+            lambda *a, **kw: pretend.stub(lock=_not_lock))
+
         with pytest.raises(TestException):
             unindex_project(task, db_request, 'foo')
 
@@ -462,7 +489,7 @@ def test_reindex_retry_on_lock(self, db_request, monkeypatch):
             pretend.call(countdown=60, exc=le)
         ]
 
-    def test_unindex_accepts_defeat(self, db_request):
+    def test_unindex_accepts_defeat(self, db_request, monkeypatch):
         task = pretend.stub()
         es_client = FakeESClient()
         es_client.delete = pretend.call_recorder(
@@ -479,6 +506,10 @@ def test_unindex_accepts_defeat(self, db_request):
             "celery.scheduler_url": "redis://redis:6379/0",
         }
 
+        monkeypatch.setattr(
+            redis.StrictRedis, "from_url",
+            lambda *a, **kw: pretend.stub(lock=_not_lock))
+
         unindex_project(task, db_request, 'foo')
 
         assert es_client.delete.calls == [
@@ -516,6 +547,10 @@ def project_docs(db, project_name=None):
             "celery.scheduler_url": "redis://redis:6379/0",
         }
 
+        monkeypatch.setattr(
+            redis.StrictRedis, "from_url",
+            lambda *a, **kw: pretend.stub(lock=_not_lock))
+
         parallel_bulk = pretend.call_recorder(lambda client, iterable: [None])
         monkeypatch.setattr(
             warehouse.search.tasks, "parallel_bulk", parallel_bulk)