Add support for context annotation in JUnit