Skip to content

Commit 205cf24

Browse files
alexander-alderman-webbalexander-alderman-webb
committed
update import in tests
2 parents d333393 + 28d7050 commit 205cf24

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

‎tests/integrations/django/test_db_transactions.py‎

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
from werkzeug.test import Client
1515

1616
from sentry_sdk import start_transaction
17-
from sentry_sdk.consts import SPANDATA, DBOPERATION
17+
from sentry_sdk.consts import SPANDATA, SPANNAME
1818
from sentry_sdk.integrations.django import DjangoIntegration
1919

2020
from tests.integrations.django.utils import pytest_mark_django_db_decorator
@@ -145,8 +145,8 @@ def test_db_transaction_spans_disabled_no_autocommit(
145145
sqlite_rollback["spans"],
146146
sqlite_commit["spans"],
147147
)
148-
if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.COMMIT
149-
or span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.ROLLBACK
148+
if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.COMMIT
149+
or span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.ROLLBACK
150150
]
151151
assert len(commit_spans) == 0
152152

@@ -263,8 +263,8 @@ def test_db_transaction_spans_disabled_atomic(sentry_init, client, capture_event
263263
sqlite_rollback["spans"],
264264
sqlite_commit["spans"],
265265
)
266-
if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.COMMIT
267-
or span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.ROLLBACK
266+
if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.COMMIT
267+
or span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.ROLLBACK
268268
]
269269
assert len(commit_spans) == 0
270270

@@ -297,7 +297,7 @@ def test_db_no_autocommit_execute(sentry_init, client, capture_events):
297297
commit_spans = [
298298
span
299299
for span in event["spans"]
300-
if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.COMMIT
300+
if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.COMMIT
301301
]
302302
assert len(commit_spans) == 1
303303
commit_span = commit_spans[0]
@@ -388,7 +388,7 @@ def test_db_no_autocommit_executemany(sentry_init, client, capture_events):
388388
commit_spans = [
389389
span
390390
for span in event["spans"]
391-
if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.COMMIT
391+
if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.COMMIT
392392
]
393393
assert len(commit_spans) == 1
394394
commit_span = commit_spans[0]
@@ -439,7 +439,7 @@ def test_db_no_autocommit_rollback_execute(sentry_init, client, capture_events):
439439
rollback_spans = [
440440
span
441441
for span in event["spans"]
442-
if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.ROLLBACK
442+
if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.ROLLBACK
443443
]
444444
assert len(rollback_spans) == 1
445445
rollback_span = rollback_spans[0]
@@ -530,7 +530,7 @@ def test_db_no_autocommit_rollback_executemany(sentry_init, client, capture_even
530530
rollback_spans = [
531531
span
532532
for span in event["spans"]
533-
if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.ROLLBACK
533+
if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.ROLLBACK
534534
]
535535
assert len(rollback_spans) == 1
536536
rollback_span = rollback_spans[0]
@@ -581,7 +581,7 @@ def test_db_atomic_execute(sentry_init, client, capture_events):
581581
commit_spans = [
582582
span
583583
for span in event["spans"]
584-
if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.COMMIT
584+
if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.COMMIT
585585
]
586586
assert len(commit_spans) == 1
587587
commit_span = commit_spans[0]
@@ -669,7 +669,7 @@ def test_db_atomic_executemany(sentry_init, client, capture_events):
669669
commit_spans = [
670670
span
671671
for span in event["spans"]
672-
if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.COMMIT
672+
if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.COMMIT
673673
]
674674
assert len(commit_spans) == 1
675675
commit_span = commit_spans[0]
@@ -721,7 +721,7 @@ def test_db_atomic_rollback_execute(sentry_init, client, capture_events):
721721
rollback_spans = [
722722
span
723723
for span in event["spans"]
724-
if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.ROLLBACK
724+
if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.ROLLBACK
725725
]
726726
assert len(rollback_spans) == 1
727727
rollback_span = rollback_spans[0]
@@ -810,7 +810,7 @@ def test_db_atomic_rollback_executemany(sentry_init, client, capture_events):
810810
rollback_spans = [
811811
span
812812
for span in event["spans"]
813-
if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.ROLLBACK
813+
if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.ROLLBACK
814814
]
815815
assert len(rollback_spans) == 1
816816
rollback_span = rollback_spans[0]
@@ -862,7 +862,7 @@ def test_db_atomic_execute_exception(sentry_init, client, capture_events):
862862
rollback_spans = [
863863
span
864864
for span in event["spans"]
865-
if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.ROLLBACK
865+
if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.ROLLBACK
866866
]
867867
assert len(rollback_spans) == 1
868868
rollback_span = rollback_spans[0]
@@ -954,7 +954,7 @@ def test_db_atomic_executemany_exception(sentry_init, client, capture_events):
954954
rollback_spans = [
955955
span
956956
for span in event["spans"]
957-
if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.ROLLBACK
957+
if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.ROLLBACK
958958
]
959959
assert len(rollback_spans) == 1
960960
rollback_span = rollback_spans[0]

0 commit comments

Comments
 (0)