Add some DI tests