diff options
| author | Selwin Ong <selwin.ong@gmail.com> | 2017-11-18 08:13:36 +0700 |
|---|---|---|
| committer | Selwin Ong <selwin.ong@gmail.com> | 2017-11-18 08:13:36 +0700 |
| commit | e25c5dbc16333c2c976539bad48a583cd007bfb7 (patch) | |
| tree | 70fe8cf97cd3f77219af61c7eac603e683f6d458 /tests/test_worker.py | |
| parent | 8cca220cce8739bf1c2b1852830874e35bf1ebaf (diff) | |
| download | rq-worker-compression.tar.gz | |
job.data is now stored in compressed format.worker-compression
Diffstat (limited to 'tests/test_worker.py')
| -rw-r--r-- | tests/test_worker.py | 20 |
1 files changed, 12 insertions, 8 deletions
diff --git a/tests/test_worker.py b/tests/test_worker.py index 4563e30..8e44d29 100644 --- a/tests/test_worker.py +++ b/tests/test_worker.py @@ -4,13 +4,16 @@ from __future__ import (absolute_import, division, print_function, import os import shutil -from datetime import datetime, timedelta -from time import sleep import signal -import time -from multiprocessing import Process import subprocess import sys +import time +import zlib + +from datetime import datetime, timedelta +from multiprocessing import Process +from time import sleep + from unittest import skipIf import pytest @@ -180,10 +183,11 @@ class TestWorker(RQTestCase): # importable from the worker process. job = Job.create(func=div_by_zero, args=(3,)) job.save() - data = self.testconn.hget(job.key, 'data') - invalid_data = data.replace(b'div_by_zero', b'nonexisting') - assert data != invalid_data - self.testconn.hset(job.key, 'data', invalid_data) + + job_data = job.data + invalid_data = job_data.replace(b'div_by_zero', b'nonexisting') + assert job_data != invalid_data + self.testconn.hset(job.key, 'data', zlib.compress(invalid_data)) # We use the low-level internal function to enqueue any data (bypassing # validity checks) |
