@@ -806,22 +806,33 @@ def dispatch_request(self):
806806 assert event ["transaction" ] == "hello_class"
807807
808808
809- def test_sentry_trace_context (sentry_init , app , capture_events ):
810- sentry_init (integrations = [flask_sentry .FlaskIntegration ()])
809+ @pytest .mark .parametrize (
810+ "template_string" , ["{{ sentry_trace }}" , "{{ sentry_trace_meta }}" ]
811+ )
812+ def test_sentry_trace_context (sentry_init , app , capture_events , template_string ):
813+ sentry_init (integrations = [flask_sentry .FlaskIntegration ()], traces_sample_rate = 1.0 )
811814 events = capture_events ()
812815
813816 @app .route ("/" )
814817 def index ():
815- sentry_span = Hub .current . scope . span
816- capture_message (sentry_span . to_traceparent ())
817- return render_template_string ("{{ sentry_trace }}" )
818+ hub = Hub .current
819+ capture_message (hub . get_traceparent () + " \n " + hub . get_baggage ())
820+ return render_template_string (template_string )
818821
819822 with app .test_client () as client :
820823 response = client .get ("/" )
821824 assert response .status_code == 200
822- assert response .data .decode (
823- "utf-8"
824- ) == '<meta name="sentry-trace" content="%s" />' % (events [0 ]["message" ],)
825+
826+ rendered_meta = response .data .decode ("utf-8" )
827+ traceparent , baggage = events [0 ]["message" ].split ("\n " )
828+ expected_meta = (
829+ '<meta name="sentry-trace" content="%s"><meta name="baggage" content="%s">'
830+ % (
831+ traceparent ,
832+ baggage ,
833+ )
834+ )
835+ assert rendered_meta == expected_meta
825836
826837
827838def test_dont_override_sentry_trace_context (sentry_init , app ):
0 commit comments