mirror of
https://github.com/Dasemu/x-space-downloader.git
synced 2026-10-12 00:18:35 +00:00
90 lines
2.9 KiB
Python
90 lines
2.9 KiB
Python
import json
|
|
|
|
from src.x_api import build_timeline_from_history, parse_audio_space_metadata
|
|
|
|
|
|
def test_parse_audio_space_metadata_extracts_title_creator_and_speakers():
|
|
payload = {
|
|
"data": {
|
|
"audioSpace": {
|
|
"metadata": {
|
|
"rest_id": "1ZkKzZWnVRRKv",
|
|
"title": "Example Space",
|
|
"media_key": "28_123",
|
|
"creator_results": {
|
|
"result": {
|
|
"rest_id": "10",
|
|
"legacy": {
|
|
"name": "Host User",
|
|
"screen_name": "host",
|
|
"profile_image_url_https": "https://example.com/host.jpg",
|
|
},
|
|
}
|
|
},
|
|
},
|
|
"participants": {
|
|
"speakers": [
|
|
{
|
|
"display_name": "Speaker User",
|
|
"twitter_screen_name": "speaker",
|
|
"avatar_url": "https://example.com/speaker.jpg",
|
|
"periscope_user_id": "20",
|
|
}
|
|
]
|
|
},
|
|
}
|
|
}
|
|
}
|
|
|
|
metadata = parse_audio_space_metadata(payload, "1ZkKzZWnVRRKv")
|
|
|
|
assert metadata.title == "Example Space"
|
|
assert metadata.media_key == "28_123"
|
|
assert [participant.username for participant in metadata.participants] == ["host", "speaker"]
|
|
assert metadata.participants[0].role == "host"
|
|
assert metadata.participants[1].role == "speaker"
|
|
|
|
|
|
def test_build_timeline_uses_unmute_mute_events_not_captions():
|
|
start_ms = 1_000_000
|
|
messages = [
|
|
_history_message("alice", 17, start_ms + 5_000),
|
|
_caption_message("bob", start_ms + 6_000),
|
|
_history_message("alice", 16, start_ms + 12_000),
|
|
]
|
|
|
|
intervals, first_seen, participants = build_timeline_from_history(messages, start_ms)
|
|
|
|
assert len(intervals) == 1
|
|
assert intervals[0].speaker == "alice"
|
|
assert intervals[0].start == 5
|
|
assert intervals[0].end == 12
|
|
assert first_seen == {"alice": 5}
|
|
assert {participant.username for participant in participants} == {"alice", "bob"}
|
|
|
|
|
|
def _history_message(username, event, timestamp_ms):
|
|
payload = {
|
|
"body": json.dumps({
|
|
"guestBroadcastingEvent": event,
|
|
"guestRemoteID": username,
|
|
"guestUsername": username,
|
|
"type": 40,
|
|
}),
|
|
"timestamp": timestamp_ms * 1_000_000,
|
|
}
|
|
return {"payload": json.dumps(payload)}
|
|
|
|
|
|
def _caption_message(username, timestamp_ms):
|
|
payload = {
|
|
"body": json.dumps({
|
|
"type": 45,
|
|
"username": username,
|
|
"displayName": username.title(),
|
|
"timestamp": timestamp_ms,
|
|
}),
|
|
"timestamp": timestamp_ms * 1_000_000,
|
|
}
|
|
return {"payload": json.dumps(payload)}
|