Add injector mock for tests