Added unittest to show the effect of DI