Add DI tests