Added support for Memcached in Windows (#56)

celery · Oct 12, 2023 · cf50961 · cf50961
1 parent 9ec6ced
commit cf50961
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 4 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -50,7 +50,7 @@ version = "1.0.0a1"
 
 [tool.poetry.dependencies]
 python = ">= 3.8,<4.0"
-celery = { version = "^5.3", extras = ["redis", "memcache"] }
+celery = { version = "^5.3", extras = ["redis", "memcache", "pymemcache"] }
 retry = "^0.9.2"
 pytest-docker-tools = "^3.1.3"
 

diff --git a/src/pytest_celery/vendors/memcached/container.py b/src/pytest_celery/vendors/memcached/container.py
@@ -1,4 +1,9 @@
-import pylibmc as memcache
+import sys
+
+if sys.platform == "win32":
+    import memcache
+else:
+    import pylibmc as memcache
 
 from pytest_celery.api.container import CeleryTestContainer
 from pytest_celery.vendors.memcached.defaults import MEMCACHED_ENV

diff --git a/src/pytest_celery/vendors/worker/Dockerfile b/src/pytest_celery/vendors/worker/Dockerfile
@@ -20,7 +20,7 @@ ENV PYTHONDONTWRITEBYTECODE=1
 
 # Install Python dependencies
 RUN pip install --no-cache-dir --upgrade pip \
-    && pip install --no-cache-dir celery[redis,memcache]${CELERY_VERSION:+==$CELERY_VERSION}
+    && pip install --no-cache-dir celery[redis,memcache,pymemcache]${CELERY_VERSION:+==$CELERY_VERSION}
 
 # The workdir must be /app
 WORKDIR /app