[#31] Added more integration tests