def test_can_make_handler(db):
"""Test whether a handler can be added to a task queue."""
q = pgtq.PgTq('q', db.url())
@q.handler()
# pylint: disable=unused-variable
def test_handler():
"""Compute the meaning of life, the universe, etc."""
return 42
def test_task_name_overide(db):
"""Test whether we can override the default name of a task."""
q = pgtq.PgTq('q', db.url())
@q.handler(name='more_awesome_name')
def test_handler(a, b):
"""Sum numbers"""
return a + b
test_handler.push(2, 3)
task = q.pop()
assert task.name == 'more_awesome_name'
def test_failing_task_marked_interupted(db):
"""Test whether a failing task will be recorded as having failed."""
q = pgtq.PgTq('q', db.url())
@q.handler(max_retries=1)
def test_handler():
"""Raise ValueError"""
raise ValueError
test_handler.push()
conn = psycopg2.connect(db.url())
sql = "SELECT count(*) FROM pgtq_q_runnable;"
with conn:
with conn.cursor() as cur:
cur.execute(sql)
assert cur.fetchone()[0] == 1
task = q.pop()
assert task.max_retries == 1
assert task.attempts == 0
assert task.name == "test_handler"
try:
task.execute()
assert False
except ValueError:
pass
sql = "SELECT count(*) FROM pgtq_q_runnable;"
with conn:
with conn.cursor() as cur:
cur.execute(sql)
assert cur.fetchone()[0] == 0
sql = "SELECT count(*) FROM pgtq_q_scheduled;"
with conn:
with conn.cursor() as cur:
cur.execute(sql)
assert cur.fetchone()[0] == 1
sql = "SELECT task FROM pgtq_q_scheduled;"
with conn:
with conn.cursor() as cur:
cur.execute(sql)
assert cur.fetchone()[0]['name'] == 'test_handler'
no_task = q.pop()
assert no_task is None
assert q.run_scheduled() is None
def test_conflicts_detected(db):
"""Test that duplicate handlers are reported."""
q = pgtq.PgTq('q', db.url())
@q.handler()
# pylint: disable=unused-variable
def test_handler(a, b):
"""Sum numbers"""
return a + b
try:
@q.handler(name="test_handler")
# pylint: disable=unused-variable
def test_handler_b(a, b):
"""Multiply numbers"""
return a * b
assert False
except RuntimeError as e:
assert "already exists" in str(e)
def test_can_call_handler_directly(db):
"""Test whether handlers can bypass the task queue."""
q = pgtq.PgTq('q', db.url())
@q.handler()
def test_handler(a, b):
"""Sum numbers."""
return a + b
assert test_handler(2, 3) == 5
@q.handler()
def failing_handler():
"""Raise RuntimeError."""
raise RuntimeError("42")
try:
failing_handler()
assert False
except RuntimeError as e:
assert "42" in str(e)
def test_most_things(db):
"""Test putting a task onto the queue and getting it out again."""
q = pgtq.PgTq('q', db.url())
@q.handler()
def test_handler(a, b):
"""Sum numbers"""
return a + b
test_handler.push(2, 3)
conn = psycopg2.connect(db.url())
sql = "SELECT count(*) FROM pgtq_q_runnable;"
with conn:
with conn.cursor() as cur:
cur.execute(sql)
assert cur.fetchone()[0] == 1
task = q.pop()
assert task.key == 1
assert task.name == 'test_handler'
assert task.attempts == 0
assert task.execute() == 5
sql = "SELECT count(*) FROM pgtq_q_runnable;"
with conn:
with conn.cursor() as cur:
cur.execute(sql)
assert cur.fetchone()[0] == 0
sql = "SELECT count(*) FROM pgtq_q_complete;"
with conn:
with conn.cursor() as cur:
cur.execute(sql)
assert cur.fetchone()[0] == 1
no_task = q.pop()
assert no_task is None
def main():
"""Parse command line arguments and start scheduling."""
args = parser.parse_args()
q = pgtq.PgTq(args.queue, args.connection)
schedule(q)
def test_can_make_queue(db):
"""Test whether a task queue can be created."""
pgtq.PgTq('q', db.url())