10
10
11
11
# Maintenance: This should move to Functional tests and use Fake over mocks.
12
12
13
- __MODULE_PREFIX__ = "unit.test_tracing"
13
+ MODULE_PREFIX = "unit.test_tracing"
14
14
15
15
16
16
@pytest .fixture
@@ -128,10 +128,10 @@ def greeting(name, message):
128
128
# and use service name as a metadata namespace
129
129
assert in_subsegment_mock .in_subsegment .call_count == 1
130
130
assert in_subsegment_mock .in_subsegment .call_args == mocker .call (
131
- name = f"## { __MODULE_PREFIX__ } .test_tracer_method.<locals>.greeting"
131
+ name = f"## { MODULE_PREFIX } .test_tracer_method.<locals>.greeting"
132
132
)
133
133
assert in_subsegment_mock .put_metadata .call_args == mocker .call (
134
- key = f"{ __MODULE_PREFIX__ } .test_tracer_method.<locals>.greeting response" ,
134
+ key = f"{ MODULE_PREFIX } .test_tracer_method.<locals>.greeting response" ,
135
135
value = dummy_response ,
136
136
namespace = "booking" ,
137
137
)
@@ -261,7 +261,7 @@ def greeting(name, message):
261
261
put_metadata_mock_args = in_subsegment_mock .put_metadata .call_args [1 ]
262
262
assert (
263
263
put_metadata_mock_args ["key" ]
264
- == f"{ __MODULE_PREFIX__ } .test_tracer_method_exception_metadata.<locals>.greeting error"
264
+ == f"{ MODULE_PREFIX } .test_tracer_method_exception_metadata.<locals>.greeting error"
265
265
)
266
266
assert put_metadata_mock_args ["namespace" ] == "booking"
267
267
@@ -315,20 +315,20 @@ async def greeting(name, message):
315
315
# THEN we should add metadata for each response like we would for a sync decorated method
316
316
assert in_subsegment_mock .in_subsegment .call_count == 2
317
317
assert in_subsegment_greeting_call_args == mocker .call (
318
- name = f"## { __MODULE_PREFIX__ } .test_tracer_method_nested_async.<locals>.greeting"
318
+ name = f"## { MODULE_PREFIX } .test_tracer_method_nested_async.<locals>.greeting"
319
319
)
320
320
assert in_subsegment_greeting2_call_args == mocker .call (
321
- name = f"## { __MODULE_PREFIX__ } .test_tracer_method_nested_async.<locals>.greeting_2"
321
+ name = f"## { MODULE_PREFIX } .test_tracer_method_nested_async.<locals>.greeting_2"
322
322
)
323
323
324
324
assert in_subsegment_mock .put_metadata .call_count == 2
325
325
assert put_metadata_greeting2_call_args == mocker .call (
326
- key = f"{ __MODULE_PREFIX__ } .test_tracer_method_nested_async.<locals>.greeting_2 response" ,
326
+ key = f"{ MODULE_PREFIX } .test_tracer_method_nested_async.<locals>.greeting_2 response" ,
327
327
value = dummy_response ,
328
328
namespace = "booking" ,
329
329
)
330
330
assert put_metadata_greeting_call_args == mocker .call (
331
- key = f"{ __MODULE_PREFIX__ } .test_tracer_method_nested_async.<locals>.greeting response" ,
331
+ key = f"{ MODULE_PREFIX } .test_tracer_method_nested_async.<locals>.greeting response" ,
332
332
value = dummy_response ,
333
333
namespace = "booking" ,
334
334
)
@@ -374,7 +374,7 @@ async def greeting(name, message):
374
374
put_metadata_mock_args = in_subsegment_mock .put_metadata .call_args [1 ]
375
375
assert (
376
376
put_metadata_mock_args ["key" ]
377
- == f"{ __MODULE_PREFIX__ } .test_tracer_method_exception_metadata_async.<locals>.greeting error"
377
+ == f"{ MODULE_PREFIX } .test_tracer_method_exception_metadata_async.<locals>.greeting error"
378
378
)
379
379
assert put_metadata_mock_args ["namespace" ] == "booking"
380
380
@@ -408,7 +408,7 @@ def handler(event, context):
408
408
assert in_subsegment_mock .in_subsegment .call_count == 2
409
409
assert handler_trace == mocker .call (name = "## handler" )
410
410
assert yield_function_trace == mocker .call (
411
- name = f"## { __MODULE_PREFIX__ } .test_tracer_yield_from_context_manager.<locals>.yield_with_capture"
411
+ name = f"## { MODULE_PREFIX } .test_tracer_yield_from_context_manager.<locals>.yield_with_capture"
412
412
)
413
413
assert "test result" in result
414
414
@@ -435,7 +435,7 @@ def yield_with_capture():
435
435
put_metadata_mock_args = in_subsegment_mock .put_metadata .call_args [1 ]
436
436
assert (
437
437
put_metadata_mock_args ["key" ]
438
- == f"{ __MODULE_PREFIX__ } .test_tracer_yield_from_context_manager_exception_metadata.<locals>.yield_with_capture error" # noqa E501
438
+ == f"{ MODULE_PREFIX } .test_tracer_yield_from_context_manager_exception_metadata.<locals>.yield_with_capture error" # noqa E501
439
439
)
440
440
assert isinstance (put_metadata_mock_args ["value" ], ValueError )
441
441
assert put_metadata_mock_args ["namespace" ] == "booking"
@@ -479,7 +479,7 @@ def handler(event, context):
479
479
assert in_subsegment_mock .in_subsegment .call_count == 2
480
480
assert handler_trace == mocker .call (name = "## handler" )
481
481
assert yield_function_trace == mocker .call (
482
- name = f"## { __MODULE_PREFIX__ } .test_tracer_yield_from_nested_context_manager.<locals>.yield_with_capture"
482
+ name = f"## { MODULE_PREFIX } .test_tracer_yield_from_nested_context_manager.<locals>.yield_with_capture"
483
483
)
484
484
assert "test result" in result
485
485
@@ -511,7 +511,7 @@ def handler(event, context):
511
511
assert in_subsegment_mock .in_subsegment .call_count == 2
512
512
assert handler_trace == mocker .call (name = "## handler" )
513
513
assert generator_fn_trace == mocker .call (
514
- name = f"## { __MODULE_PREFIX__ } .test_tracer_yield_from_generator.<locals>.generator_fn"
514
+ name = f"## { MODULE_PREFIX } .test_tracer_yield_from_generator.<locals>.generator_fn"
515
515
)
516
516
assert "test result" in result
517
517
@@ -537,7 +537,7 @@ def generator_fn():
537
537
put_metadata_mock_args = in_subsegment_mock .put_metadata .call_args [1 ]
538
538
assert (
539
539
put_metadata_mock_args ["key" ]
540
- == f"{ __MODULE_PREFIX__ } .test_tracer_yield_from_generator_exception_metadata.<locals>.generator_fn error"
540
+ == f"{ MODULE_PREFIX } .test_tracer_yield_from_generator_exception_metadata.<locals>.generator_fn error"
541
541
)
542
542
assert put_metadata_mock_args ["namespace" ] == "booking"
543
543
assert isinstance (put_metadata_mock_args ["value" ], ValueError )
0 commit comments