Adding test cases for using custom injection marks