Add more integration tests for cli