add more integration tests