Add tracing capability in test utilities