import pytest

from sentry_sdk import Hub, start_span, start_transaction

from sentry_sdk.tracing import Span, Transaction

def test_span_trimming(sentry_init, capture_events):

sentry_init(traces_sample_rate=1.0, _experiments={"max_spans": 3})

events = capture_events()

with start_transaction(name="hi"):

for i in range(10):

with start_span(op="foo{}".format(i)):

pass

(event,) = events

# the transaction is its own first span (which counts for max_spans) but it

# doesn't show up in the span list in the event, so this is 1 less than our

# max_spans value

assert len(event["spans"]) == 2

span1, span2 = event["spans"]

assert span1["op"] == "foo0"

assert span2["op"] == "foo1"

def test_transaction_method_signature(sentry_init, capture_events):

sentry_init(traces_sample_rate=1.0)

events = capture_events()

with pytest.raises(TypeError):

start_span(name="foo")

assert len(events) == 0

with start_transaction() as transaction:

pass

assert transaction.name == "<unlabeled transaction>"

assert len(events) == 1

with start_transaction() as transaction:

transaction.name = "name-known-after-transaction-started"

assert len(events) == 2

with start_transaction(name="a"):

pass

assert len(events) == 3

with start_transaction(Transaction(name="c")):

pass

assert len(events) == 4

def test_finds_transaction_on_scope(sentry_init):

sentry_init(traces_sample_rate=1.0)

transaction = start_transaction(name="dogpark")

scope = Hub.current.scope

# See note in Scope class re: getters and setters of the `transaction`

# property. For the moment, assigning to scope.transaction merely sets the

# transaction name, rather than putting the transaction on the scope, so we

# have to assign to _span directly.

scope._span = transaction

# Reading scope.property, however, does what you'd expect, and returns the

# transaction on the scope.

assert scope.transaction is not None

assert isinstance(scope.transaction, Transaction)

assert scope.transaction.name == "dogpark"

# If the transaction is also set as the span on the scope, it can be found

# by accessing _span, too.

assert scope._span is not None

assert isinstance(scope._span, Transaction)

assert scope._span.name == "dogpark"

def test_finds_transaction_when_decedent_span_is_on_scope(

sentry_init,

):

sentry_init(traces_sample_rate=1.0)

transaction = start_transaction(name="dogpark")

child_span = transaction.start_child(op="sniffing")

scope = Hub.current.scope

scope._span = child_span

# this is the same whether it's the transaction itself or one of its

# decedents directly attached to the scope

assert scope.transaction is not None

assert isinstance(scope.transaction, Transaction)

assert scope.transaction.name == "dogpark"

# here we see that it is in fact the span on the scope, rather than the

# transaction itself

assert scope._span is not None

assert isinstance(scope._span, Span)

assert scope._span.op == "sniffing"

def test_finds_orphan_span_on_scope(sentry_init):

# this is deprecated behavior which may be removed at some point (along with

# the start_span function)

sentry_init(traces_sample_rate=1.0)

span = start_span(op="sniffing")

scope = Hub.current.scope

scope._span = span

assert scope._span is not None

assert isinstance(scope._span, Span)

assert scope._span.op == "sniffing"

def test_finds_non_orphan_span_on_scope(sentry_init):

sentry_init(traces_sample_rate=1.0)

transaction = start_transaction(name="dogpark")

child_span = transaction.start_child(op="sniffing")

scope = Hub.current.scope

scope._span = child_span

assert scope._span is not None

assert isinstance(scope._span, Span)

assert scope._span.op == "sniffing"