DI extension tests