Add dependency injection tests