Add more integration tests for function calling