Breaking tests into unit and integration