81ada1d9b9
## Google File Picker The Google Picker lets users select or upload Google Drive files. Users can grant permission to your apps to access their Drive data, providing a secure and authorized way to interact with their files. The `generate_google_file_picker_url` returns a URL to a Google File Picker for the user.
45 lines
1.3 KiB
Python
45 lines
1.3 KiB
Python
import base64
|
|
import json
|
|
from urllib.parse import parse_qs, urlparse
|
|
|
|
import pytest
|
|
from arcade.core.schema import ToolMetadataItem
|
|
from arcade.sdk import ToolContext, ToolMetadataKey
|
|
|
|
from arcade_google.tools import generate_google_file_picker_url
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_context():
|
|
context = ToolContext(
|
|
metadata=[
|
|
ToolMetadataItem(key=ToolMetadataKey.CLIENT_ID, value="1234-3444534323"),
|
|
ToolMetadataItem(
|
|
key=ToolMetadataKey.COORDINATOR_URL, value="https://mock_coordinator_url"
|
|
),
|
|
],
|
|
)
|
|
return context
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_generate_google_file_picker_url(mock_context):
|
|
expected_decoded_config = {
|
|
"auth": {
|
|
"client_id": "1234-3444534323",
|
|
"app_id": "1234",
|
|
},
|
|
}
|
|
|
|
result = generate_google_file_picker_url(mock_context)
|
|
|
|
assert result["url"].startswith("https://mock_coordinator_url/google/drive_picker?config=")
|
|
|
|
# Decode the config from the URL
|
|
parsed_url = urlparse(result["url"])
|
|
query_params = parse_qs(parsed_url.query)
|
|
encoded_config = query_params.get("config", [None])[0]
|
|
assert encoded_config is not None
|
|
|
|
decoded_config = json.loads(base64.urlsafe_b64decode(encoded_config).decode("utf-8"))
|
|
assert decoded_config == expected_decoded_config
|