Add a test for context propagation