File tree 1 file changed +5
-2
lines changed 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -990,13 +990,13 @@ def __init__(
990
990
tracer = None ,
991
991
_config = None ,
992
992
):
993
- # type: (...) -> None
994
993
super (TestPin , self ).__init__ (service = service , tags = tags , _config = _config )
995
994
if tracer is not None :
996
995
self .tracer = tracer
997
996
998
997
def __setattr__ (self , name , value ):
999
998
if name == "tracer" :
999
+ # TestPin supports overriding the global tracer but ddtrace.trace.Pin does not.
1000
1000
super (Pin , self ).__setattr__ (name , value )
1001
1001
else :
1002
1002
super (TestPin , self ).__setattr__ (name , value )
@@ -1018,7 +1018,10 @@ def override(
1018
1018
elif isinstance (pin , TestPin ):
1019
1019
pin .clone (service = service , tags = tags , tracer = tracer ).onto (obj )
1020
1020
else :
1021
- pin .clone (service = service , tags = tags ).onto (obj )
1021
+ # replace ddtrace.trace.Pin with tests.utils.TestPin. This will
1022
+ # allow us to override the global tracer in tests.
1023
+ pin .remove_from (obj )
1024
+ TestPin (service = service , tags = tags , tracer = tracer ).onto (obj )
1022
1025
1023
1026
def clone (
1024
1027
self ,
You can’t perform that action at this time.
0 commit comments