Adding unit test and a single integration test