from sentry_sdk.integrations.rq import RqIntegration

import os

import json

from fakeredis import FakeStrictRedis

import rq

from sentry_sdk import Hub

def crashing_job(foo):

1 / 0

def test_basic(sentry_init, capture_events):

sentry_init(integrations=[RqIntegration()])

events = capture_events()

queue = rq.Queue(connection=FakeStrictRedis())

worker = rq.SimpleWorker([queue], connection=queue.connection)

queue.enqueue(crashing_job, foo=42)

worker.work(burst=True)

event, = events

exception, = event["exception"]["values"]

assert exception["type"] == "ZeroDivisionError"

assert exception["mechanism"]["type"] == "rq"

assert exception["stacktrace"]["frames"][-1]["vars"]["foo"] == "42"

assert event["transaction"] == "tests.integrations.rq.test_rq.crashing_job"

assert event["extra"]["rq-job"] == {

"args": [],

"description": "tests.integrations.rq.test_rq.crashing_job(foo=42)",

"func": "tests.integrations.rq.test_rq.crashing_job",

"job_id": event["extra"]["rq-job"]["job_id"],

"kwargs": {"foo": 42},

}

def test_transport_shutdown(sentry_init):

sentry_init(integrations=[RqIntegration()])

events_r, events_w = os.pipe()

events_r = os.fdopen(events_r, "rb", 0)

events_w = os.fdopen(events_w, "wb", 0)

def capture_event(event):

events_w.write(json.dumps(event).encode("utf-8"))

events_w.write(b"\n")

def shutdown(timeout, callback=None):

events_w.write(b"shutdown\n")

Hub.current.client.transport.capture_event = capture_event

Hub.current.client.transport.shutdown = shutdown

queue = rq.Queue(connection=FakeStrictRedis())

worker = rq.Worker([queue], connection=queue.connection)

queue.enqueue(crashing_job, foo=42)

worker.work(burst=True)

event = events_r.readline()

shutdown = events_r.readline()

event = json.loads(event.decode("utf-8"))

assert shutdown == b"shutdown\n"

exception, = event["exception"]["values"]

assert exception["type"] == "ZeroDivisionError"