MINOR: Use mock in unit test and exclude container in unit tests