update import in tests

Author: alexander-alderman-webb

tests/integrations/django/test_db_transactions.py

@@ -14,7 +14,7 @@
 from werkzeug.test import Client
 
 from sentry_sdk import start_transaction
-from sentry_sdk.consts import SPANDATA, DBOPERATION
+from sentry_sdk.consts import SPANDATA, SPANNAME
 from sentry_sdk.integrations.django import DjangoIntegration
 
 from tests.integrations.django.utils import pytest_mark_django_db_decorator
@@ -97,7 +97,7 @@ def test_db_transaction_spans_disabled_no_autocommit(
     commit_spans = [
         span
         for span in itertools.chain(postgres_spans["spans"], sqlite_spans["spans"])
-        if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.COMMIT
+        if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.COMMIT
     ]
     assert len(commit_spans) == 0
 
@@ -168,7 +168,7 @@ def test_db_transaction_spans_disabled_atomic(sentry_init, client, capture_event
     commit_spans = [
         span
         for span in itertools.chain(postgres_spans["spans"], sqlite_spans["spans"])
-        if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.COMMIT
+        if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.COMMIT
     ]
     assert len(commit_spans) == 0
 
@@ -201,7 +201,7 @@ def test_db_no_autocommit_execute(sentry_init, client, capture_events):
     commit_spans = [
         span
         for span in event["spans"]
-        if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.COMMIT
+        if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.COMMIT
     ]
     assert len(commit_spans) == 1
     commit_span = commit_spans[0]
@@ -292,7 +292,7 @@ def test_db_no_autocommit_executemany(sentry_init, client, capture_events):
     commit_spans = [
         span
         for span in event["spans"]
-        if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.COMMIT
+        if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.COMMIT
     ]
     assert len(commit_spans) == 1
     commit_span = commit_spans[0]
@@ -343,7 +343,7 @@ def test_db_atomic_execute(sentry_init, client, capture_events):
     commit_spans = [
         span
         for span in event["spans"]
-        if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.COMMIT
+        if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.COMMIT
     ]
     assert len(commit_spans) == 1
     commit_span = commit_spans[0]
@@ -431,7 +431,7 @@ def test_db_atomic_executemany(sentry_init, client, capture_events):
     commit_spans = [
         span
         for span in event["spans"]
-        if span["data"].get(SPANDATA.DB_OPERATION) == DBOPERATION.COMMIT
+        if span["data"].get(SPANDATA.DB_OPERATION) == SPANNAME.COMMIT
     ]
     assert len(commit_spans) == 1
     commit_span = commit_spans[0]