Added mock integration tests