Finally figured out how to do injection in test classes