2014-05-05 08:49:29 +00:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
|
from __future__ import (absolute_import, division, print_function,
|
|
|
|
|
unicode_literals)
|
|
|
|
|
|
|
|
|
|
from rq import Connection, Queue
|
|
|
|
|
|
|
|
|
|
from tests import find_empty_redis_database, RQTestCase
|
New connection management.
Connections can now be set explicitly on Queues, Workers, and Jobs.
Jobs that are implicitly created by Queue or Worker API calls now
inherit the connection of their creator's.
For all RQ object instances that are created now holds that the
"current" connection is used if none is passed in explicitly. The
"current" connection is thus hold on to at creation time and won't be
changed for the lifetime of the object.
Effectively, this means that, given a default Redis connection, say you
create a queue Q1, then push another Redis connection onto the
connection stack, then create Q2. In that case, Q1 means a queue on the
first connection and Q2 on the second connection.
This is way more clear than it used to be.
Also, I've removed the `use_redis()` call, which was named ugly.
Instead, some new alternatives for connection management now exist.
You can push/pop connections now:
>>> my_conn = Redis()
>>> push_connection(my_conn)
>>> q = Queue()
>>> q.connection == my_conn
True
>>> pop_connection() == my_conn
Also, you can stack them syntactically:
>>> conn1 = Redis()
>>> conn2 = Redis('example.org', 1234)
>>> with Connection(conn1):
... q = Queue()
... with Connection(conn2):
... q2 = Queue()
... q3 = Queue()
>>> q.connection == conn1
True
>>> q2.connection == conn2
True
>>> q3.connection == conn1
True
Or, if you only require a single connection to Redis (for most uses):
>>> use_connection(Redis())
2012-03-23 13:33:49 +00:00
|
|
|
from tests.fixtures import do_nothing
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def new_connection():
|
|
|
|
|
return find_empty_redis_database()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class TestConnectionInheritance(RQTestCase):
|
|
|
|
|
def test_connection_detection(self):
|
|
|
|
|
"""Automatic detection of the connection."""
|
|
|
|
|
q = Queue()
|
|
|
|
|
self.assertEquals(q.connection, self.testconn)
|
|
|
|
|
|
|
|
|
|
def test_connection_stacking(self):
|
|
|
|
|
"""Connection stacking."""
|
|
|
|
|
conn1 = new_connection()
|
|
|
|
|
conn2 = new_connection()
|
|
|
|
|
|
|
|
|
|
with Connection(conn1):
|
|
|
|
|
q1 = Queue()
|
|
|
|
|
with Connection(conn2):
|
|
|
|
|
q2 = Queue()
|
|
|
|
|
self.assertNotEquals(q1.connection, q2.connection)
|
|
|
|
|
|
|
|
|
|
def test_connection_pass_thru(self):
|
|
|
|
|
"""Connection passed through from queues to jobs."""
|
|
|
|
|
q1 = Queue()
|
|
|
|
|
with Connection(new_connection()):
|
|
|
|
|
q2 = Queue()
|
|
|
|
|
job1 = q1.enqueue(do_nothing)
|
|
|
|
|
job2 = q2.enqueue(do_nothing)
|
|
|
|
|
self.assertEquals(q1.connection, job1.connection)
|
|
|
|
|
self.assertEquals(q2.connection, job2.connection)
|