test_client.py 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847
  1. """Tests for ChattoClient.
  2. Since the client now goes through generated ConnectRPC service stubs, tests
  3. mock at the service-client method level with ``AsyncMock`` rather than at the
  4. HTTP transport layer. This exercises the request-building and response-parsing
  5. paths without any real network traffic.
  6. """
  7. from __future__ import annotations
  8. from unittest.mock import AsyncMock
  9. import pytest
  10. from chattolib._pb.chatto.admin.v1 import (
  11. room_layout_pb2,
  12. )
  13. from chattolib._pb.chatto.admin.v1 import (
  14. server_pb2 as admin_server_pb2,
  15. )
  16. from chattolib._pb.chatto.api.v1 import (
  17. account_pb2,
  18. asset_uploads_pb2,
  19. attachments_pb2,
  20. messages_pb2,
  21. notification_preferences_pb2,
  22. notifications_pb2,
  23. presence_pb2,
  24. push_notifications_pb2,
  25. reactions_pb2,
  26. read_state_pb2,
  27. roles_pb2,
  28. room_directory_pb2,
  29. room_timeline_pb2,
  30. rooms_pb2,
  31. threads_pb2,
  32. viewer_pb2,
  33. voice_calls_pb2,
  34. )
  35. from chattolib._pb.chatto.discovery.v1 import server_pb2 as discovery_server_pb2
  36. from chattolib.client import ChattoClient
  37. from chattolib.types import (
  38. ImageFitMode,
  39. ImageTransformOptions,
  40. NotificationLevel,
  41. PresenceStatus,
  42. RoomKind,
  43. TimeFormat,
  44. )
  45. BASE = "https://chat.chatto.run"
  46. @pytest.fixture
  47. def client():
  48. return ChattoClient(token="test-token", base_url=BASE)
  49. def _mock_method(client: ChattoClient, service: str, method: str, response):
  50. """Replace a service-client method with an AsyncMock returning ``response``."""
  51. svc = getattr(client._svc, service)
  52. setattr(svc, method, AsyncMock(return_value=response))
  53. return getattr(svc, method)
  54. # --- Server discovery ---------------------------------------------------
  55. async def test_get_server(client):
  56. resp = discovery_server_pb2.GetServerResponse()
  57. resp.profile.name = "Chatto HQ"
  58. resp.profile.version = "0.4.2"
  59. resp.login.direct_registration_enabled = True
  60. resp.login.authorize_url = "/oauth"
  61. _mock_method(client, "server_discovery", "get_server", resp)
  62. async with client:
  63. profile, login = await client.get_server()
  64. assert profile.name == "Chatto HQ"
  65. assert profile.version == "0.4.2"
  66. assert login.direct_registration_enabled is True
  67. assert login.authorize_url == "/oauth"
  68. # --- Viewer -------------------------------------------------------------
  69. async def test_me(client):
  70. resp = viewer_pb2.GetViewerResponse()
  71. resp.user.profile.id = "u1"
  72. resp.user.profile.login = "alice"
  73. resp.user.profile.display_name = "Alice"
  74. resp.user.profile.presence_status = "PRESENCE_STATUS_ONLINE"
  75. resp.user.has_verified_email = True
  76. resp.user.settings.timezone = "UTC"
  77. resp.user.settings.time_format = "TIME_FORMAT_24_HOUR"
  78. resp.user.has_password = True
  79. _mock_method(client, "viewer", "get_viewer", resp)
  80. async with client:
  81. user = await client.me()
  82. assert user.id == "u1"
  83. assert user.login == "alice"
  84. assert user.display_name == "Alice"
  85. assert user.presence_status is PresenceStatus.ONLINE
  86. # --- Room directory ----------------------------------------------------
  87. async def test_list_rooms(client):
  88. resp = room_directory_pb2.ListRoomsResponse()
  89. entry = resp.rooms.add()
  90. entry.room.id = "r1"
  91. entry.room.kind = "ROOM_KIND_CHANNEL"
  92. entry.room.name = "general"
  93. entry.room.group_id = "g1"
  94. entry.viewer_state.is_member = True
  95. entry.viewer_state.has_unread = False
  96. _mock_method(client, "room_directory", "list_rooms", resp)
  97. async with client:
  98. rooms = await client.list_rooms()
  99. assert len(rooms) == 1
  100. assert rooms[0].room is not None
  101. assert rooms[0].room.name == "general"
  102. assert rooms[0].room.kind is RoomKind.CHANNEL
  103. assert rooms[0].viewer_state.is_member is True
  104. # --- Messages ----------------------------------------------------------
  105. async def test_post_message(client):
  106. resp = messages_pb2.CreateMessageResponse()
  107. resp.message.id = "e1"
  108. resp.message.room_id = "r1"
  109. resp.message.actor_id = "u1"
  110. resp.message.body = "Hello!"
  111. _mock_method(client, "messages", "create_message", resp)
  112. async with client:
  113. message = await client.post_message("r1", "Hello!")
  114. assert message.id == "e1"
  115. assert message.body == "Hello!"
  116. assert message.room_id == "r1"
  117. async def test_update_message(client):
  118. resp = messages_pb2.UpdateMessageResponse()
  119. resp.message.id = "e1"
  120. resp.message.room_id = "r1"
  121. resp.message.actor_id = "u1"
  122. resp.message.body = "edited"
  123. resp.message.updated_at.FromJsonString("2026-01-01T00:01:00Z")
  124. m = _mock_method(client, "messages", "update_message", resp)
  125. async with client:
  126. msg = await client.update_message("r1", "e1", body="edited")
  127. assert msg.body == "edited"
  128. assert msg.updated_at is not None
  129. # verify the request had the optional body field explicitly set
  130. call_args = m.call_args
  131. req = call_args.args[0]
  132. assert isinstance(req, messages_pb2.UpdateMessageRequest)
  133. assert req.HasField("body")
  134. assert req.body == "edited"
  135. async def test_delete_message(client):
  136. _mock_method(client, "messages", "delete_message", messages_pb2.DeleteMessageResponse())
  137. async with client:
  138. assert await client.delete_message("r1", "e1") is None
  139. async def test_add_reaction(client):
  140. resp = reactions_pb2.AddReactionResponse(added=True)
  141. _mock_method(client, "messages", "add_reaction", resp)
  142. async with client:
  143. assert await client.add_reaction("r1", "e1", "thumbsup") is True
  144. async def test_remove_reaction(client):
  145. resp = reactions_pb2.RemoveReactionResponse(removed=True)
  146. _mock_method(client, "messages", "remove_reaction", resp)
  147. async with client:
  148. assert await client.remove_reaction("r1", "e1", "thumbsup") is True
  149. # --- Room lifecycle ---------------------------------------------------
  150. async def test_create_room(client):
  151. resp = rooms_pb2.CreateRoomResponse()
  152. resp.room.id = "r1"
  153. resp.room.kind = "ROOM_KIND_CHANNEL"
  154. resp.room.name = "general"
  155. resp.room.group_id = "g1"
  156. _mock_method(client, "rooms", "create_room", resp)
  157. async with client:
  158. room = await client.create_room("general", "g1")
  159. assert room.id == "r1"
  160. assert room.name == "general"
  161. assert room.group_id == "g1"
  162. async def test_archive_room(client):
  163. resp = rooms_pb2.ArchiveRoomResponse()
  164. resp.room.id = "r1"
  165. resp.room.name = "general"
  166. resp.room.archived = True
  167. _mock_method(client, "rooms", "archive_room", resp)
  168. async with client:
  169. room = await client.archive_room("r1")
  170. assert room.archived is True
  171. async def test_ban_member(client):
  172. _mock_method(client, "rooms", "ban_member", rooms_pb2.BanMemberResponse())
  173. async with client:
  174. assert await client.ban_member("r1", "u2", "spam") is None
  175. async def test_start_dm(client):
  176. resp = rooms_pb2.StartDMResponse()
  177. resp.room.id = "r1"
  178. resp.room.kind = "ROOM_KIND_DM"
  179. _mock_method(client, "rooms", "start_dm", resp)
  180. async with client:
  181. room = await client.start_dm(["u1", "u2"])
  182. assert room.id == "r1"
  183. assert room.kind is RoomKind.DM
  184. # --- Room timeline ----------------------------------------------------
  185. async def test_get_room_events(client):
  186. resp = room_timeline_pb2.GetRoomEventsResponse()
  187. event = resp.page.events.add()
  188. event.id = "e1"
  189. event.actor_id = "u1"
  190. event.created_at.FromJsonString("2026-01-01T00:00:00Z")
  191. event.message_posted.message.id = "e1"
  192. event.message_posted.message.room_id = "r1"
  193. event.message_posted.message.actor_id = "u1"
  194. event.message_posted.message.body = "hi"
  195. event.message_posted.message.created_at.FromJsonString("2026-01-01T00:00:00Z")
  196. resp.page.start_cursor = "cA"
  197. resp.page.end_cursor = "cB"
  198. resp.page.includes.users["u1"].id = "u1"
  199. resp.page.includes.users["u1"].login = "alice"
  200. resp.page.includes.users["u1"].display_name = "Alice"
  201. _mock_method(client, "rooms", "get_room_events", resp)
  202. async with client:
  203. page = await client.get_room_events("r1", limit=50)
  204. assert len(page.events) == 1
  205. assert page.events[0].kind == "message_posted"
  206. assert page.events[0].message is not None
  207. assert page.events[0].message.body == "hi"
  208. assert page.users_by_id["u1"].login == "alice"
  209. async def test_mark_room_as_read(client):
  210. resp = read_state_pb2.MarkRoomAsReadResponse()
  211. resp.last_read_at.FromJsonString("2026-01-01T00:00:00Z")
  212. resp.previous_last_read_at.FromJsonString("2025-12-31T00:00:00Z")
  213. _mock_method(client, "rooms", "mark_room_as_read", resp)
  214. async with client:
  215. last, previous = await client.mark_room_as_read("r1", up_to_event_id="e1")
  216. assert last is not None
  217. assert previous is not None
  218. # --- Profile / account ------------------------------------------------
  219. async def test_update_profile(client):
  220. resp = account_pb2.UpdateProfileResponse()
  221. resp.user.id = "u1"
  222. resp.user.login = "newname"
  223. resp.user.display_name = "New Name"
  224. resp.user.presence_status = "PRESENCE_STATUS_ONLINE"
  225. _mock_method(client, "account", "update_profile", resp)
  226. async with client:
  227. user = await client.update_profile(login="newname", display_name="New Name")
  228. assert user.login == "newname"
  229. assert user.display_name == "New Name"
  230. async def test_set_presence(client):
  231. resp = presence_pb2.SetPresenceResponse()
  232. resp.status = "PRESENCE_STATUS_ONLINE"
  233. _mock_method(client, "account", "set_presence", resp)
  234. async with client:
  235. result = await client.set_presence(PresenceStatus.ONLINE)
  236. assert result is PresenceStatus.ONLINE
  237. async def test_set_presence_offline_rejected(client):
  238. async with client:
  239. with pytest.raises(ValueError):
  240. await client.set_presence(PresenceStatus.OFFLINE)
  241. async def test_get_presence_preference(client):
  242. resp = presence_pb2.GetPresencePreferenceResponse()
  243. resp.preference.status = "PRESENCE_STATUS_OFFLINE"
  244. resp.preference.revision = "rev-1"
  245. _mock_method(client, "account", "get_presence_preference", resp)
  246. async with client:
  247. pref = await client.get_presence_preference()
  248. assert pref is not None
  249. assert pref.status is PresenceStatus.OFFLINE
  250. assert pref.revision == "rev-1"
  251. async def test_get_presence_preference_absent(client):
  252. # No preference saved yet: the response carries no sub-message.
  253. _mock_method(
  254. client, "account", "get_presence_preference", presence_pb2.GetPresencePreferenceResponse()
  255. )
  256. async with client:
  257. assert await client.get_presence_preference() is None
  258. async def test_set_presence_preference(client):
  259. resp = presence_pb2.SetPresencePreferenceResponse()
  260. resp.preference.status = "PRESENCE_STATUS_DO_NOT_DISTURB"
  261. resp.preference.revision = "rev-2"
  262. m = _mock_method(client, "account", "set_presence_preference", resp)
  263. async with client:
  264. pref = await client.set_presence_preference(
  265. PresenceStatus.DO_NOT_DISTURB, expected_revision="rev-1"
  266. )
  267. assert pref.status is PresenceStatus.DO_NOT_DISTURB
  268. assert pref.revision == "rev-2"
  269. req = m.await_args.args[0]
  270. assert req.status == presence_pb2.PresenceStatus.Value("PRESENCE_STATUS_DO_NOT_DISTURB")
  271. assert req.expected_revision == "rev-1"
  272. async def test_set_presence_preference_offline_allowed(client):
  273. # Unlike set_presence, OFFLINE is a valid saved choice ("Appear Offline").
  274. resp = presence_pb2.SetPresencePreferenceResponse()
  275. resp.preference.status = "PRESENCE_STATUS_OFFLINE"
  276. _mock_method(client, "account", "set_presence_preference", resp)
  277. async with client:
  278. assert (await client.set_presence_preference(PresenceStatus.OFFLINE)).status is (
  279. PresenceStatus.OFFLINE
  280. )
  281. async def test_set_presence_preference_unspecified_rejected(client):
  282. async with client:
  283. with pytest.raises(ValueError):
  284. await client.set_presence_preference(PresenceStatus.UNSPECIFIED)
  285. async def test_refresh_presence(client):
  286. resp = presence_pb2.RefreshPresenceResponse()
  287. resp.preference.status = "PRESENCE_STATUS_ONLINE"
  288. resp.preference.revision = "rev-3"
  289. _mock_method(client, "account", "refresh_presence", resp)
  290. async with client:
  291. pref = await client.refresh_presence()
  292. assert pref is not None
  293. assert pref.status is PresenceStatus.ONLINE
  294. assert pref.revision == "rev-3"
  295. async def test_update_settings(client):
  296. resp = account_pb2.UpdateSettingsResponse()
  297. resp.settings.timezone = "Europe/Rome"
  298. resp.settings.time_format = "TIME_FORMAT_24_HOUR"
  299. _mock_method(client, "account", "update_settings", resp)
  300. async with client:
  301. settings = await client.update_settings(
  302. timezone="Europe/Rome", time_format=TimeFormat.HOUR_24
  303. )
  304. assert settings.timezone == "Europe/Rome"
  305. assert settings.time_format is TimeFormat.HOUR_24
  306. # --- Threads ----------------------------------------------------------
  307. async def test_list_followed_threads(client):
  308. resp = threads_pb2.ListFollowedThreadsResponse()
  309. ft = resp.threads.add()
  310. ft.room.id = "r1"
  311. ft.room.name = "general"
  312. ft.root_message.id = "e1"
  313. ft.root_message.room_id = "r1"
  314. ft.root_message.actor_id = "u1"
  315. ft.root_message.body = "root"
  316. ft.root_message.created_at.FromJsonString("2026-01-01T00:00:00Z")
  317. ft.thread.thread_root_event_id = "e1"
  318. ft.thread.reply_count = 3
  319. ft.thread.last_reply_at.FromJsonString("2026-01-02T00:00:00Z")
  320. ft.thread.viewer_state.is_following = True
  321. ft.thread.viewer_state.has_unread_replies = False
  322. resp.page.total_count = 1
  323. resp.page.has_more = False
  324. _mock_method(client, "threads", "list_followed_threads", resp)
  325. async with client:
  326. page = await client.list_followed_threads()
  327. assert page.page.total_count == 1
  328. assert page.threads[0].thread is not None
  329. assert page.threads[0].thread.reply_count == 3
  330. assert page.threads[0].thread.is_following is True
  331. async def test_follow_and_unfollow_thread(client):
  332. _mock_method(client, "threads", "follow_thread", threads_pb2.FollowThreadResponse())
  333. _mock_method(client, "threads", "unfollow_thread", threads_pb2.UnfollowThreadResponse())
  334. async with client:
  335. assert await client.follow_thread("r1", "e1") is None
  336. assert await client.unfollow_thread("r1", "e1") is None
  337. # --- Notifications ----------------------------------------------------
  338. async def test_list_notification_occurrences(client):
  339. resp = notifications_pb2.ListNotificationOccurrencesResponse()
  340. occ = resp.occurrences.add()
  341. occ.id = "n1"
  342. occ.created_at.FromJsonString("2026-01-01T00:00:00Z")
  343. occ.actor.id = "u1"
  344. occ.actor.login = "alice"
  345. occ.actor.display_name = "Alice"
  346. occ.unread = True
  347. sig = occ.signal.direct_mention_received
  348. sig.message.room.id = "r1"
  349. sig.message.room.name = "general"
  350. sig.message.event_id = "e1"
  351. resp.page.total_count = 1
  352. resp.unread_count = 1
  353. _mock_method(client, "notifications", "list_notification_occurrences", resp)
  354. async with client:
  355. page = await client.list_notification_occurrences()
  356. assert page.page.total_count == 1
  357. assert page.unread_count == 1
  358. assert page.occurrences[0].signal is not None
  359. assert page.occurrences[0].signal.kind == "direct_mention_received"
  360. assert page.occurrences[0].signal.message is not None
  361. assert page.occurrences[0].signal.message.event_id == "e1"
  362. assert page.occurrences[0].signal.message.room is not None
  363. assert page.occurrences[0].signal.message.room.name == "general"
  364. async def test_mark_and_delete_notification_occurrence(client):
  365. read_resp = notifications_pb2.MarkNotificationReadResponse()
  366. read_resp.occurrence.id = "n1"
  367. read_resp.occurrence.unread = False
  368. _mock_method(client, "notifications", "mark_notification_read", read_resp)
  369. del_resp = notifications_pb2.DeleteNotificationOccurrenceResponse(deleted=True)
  370. _mock_method(client, "notifications", "delete_notification_occurrence", del_resp)
  371. all_resp = notifications_pb2.DeleteAllNotificationOccurrencesResponse(deleted_count=3)
  372. _mock_method(client, "notifications", "delete_all_notification_occurrences", all_resp)
  373. async with client:
  374. occ = await client.mark_notification_read("n1")
  375. assert occ is not None and occ.unread is False
  376. assert await client.delete_notification_occurrence("n1") is True
  377. assert await client.delete_all_notification_occurrences() == 3
  378. async def test_notification_preference(client):
  379. resp = notification_preferences_pb2.UpdateNotificationPreferenceResponse()
  380. resp.preference.level = "NOTIFICATION_LEVEL_MUTED"
  381. resp.preference.effective_level = "NOTIFICATION_LEVEL_MUTED"
  382. _mock_method(
  383. client,
  384. "notification_prefs",
  385. "update_room_notification_preference",
  386. resp,
  387. )
  388. async with client:
  389. pref = await client.update_room_notification_preference("r1", NotificationLevel.MUTED)
  390. assert pref.level is NotificationLevel.MUTED
  391. assert pref.effective_level is NotificationLevel.MUTED
  392. # --- Push -------------------------------------------------------------
  393. async def test_subscribe_push(client):
  394. _mock_method(client, "push", "subscribe", push_notifications_pb2.SubscribeResponse())
  395. async with client:
  396. assert await client.subscribe_push("https://push", "k", "a") is True
  397. # --- Voice calls ------------------------------------------------------
  398. async def test_active_calls(client):
  399. resp = voice_calls_pb2.ListActiveCallsResponse()
  400. call = resp.calls.add()
  401. call.call_id = "c1"
  402. call.room.id = "r1"
  403. call.room.name = "voice"
  404. p = call.participants.add()
  405. p.user.id = "u1"
  406. p.user.login = "alice"
  407. p.user.display_name = "Alice"
  408. _mock_method(client, "voice_calls", "list_active_calls", resp)
  409. async with client:
  410. calls = await client.list_active_calls()
  411. assert len(calls) == 1
  412. assert calls[0].call_id == "c1"
  413. assert calls[0].room is not None
  414. assert calls[0].room.name == "voice"
  415. # --- Assets -----------------------------------------------------------
  416. async def test_get_asset_with_thumbnail(client):
  417. resp = attachments_pb2.GetAssetResponse()
  418. resp.asset.id = "a1"
  419. resp.asset.filename = "pic.png"
  420. resp.asset.content_type = "image/png"
  421. resp.asset.size = 1024
  422. resp.asset.asset_url.url = "https://cdn/pic.png"
  423. mock = _mock_method(client, "assets", "get_asset", resp)
  424. async with client:
  425. asset = await client.get_asset(
  426. "r1",
  427. "a1",
  428. thumbnail=ImageTransformOptions(128, 128, ImageFitMode.COVER),
  429. )
  430. assert asset is not None
  431. assert asset.filename == "pic.png"
  432. assert asset.asset_url is not None
  433. assert asset.asset_url.url == "https://cdn/pic.png"
  434. # verify the request encoded thumbnail options
  435. req = mock.call_args.args[0]
  436. assert req.thumbnail.width == 128
  437. assert req.thumbnail.height == 128
  438. # --- Roles ------------------------------------------------------------
  439. async def test_list_roles(client):
  440. resp = roles_pb2.ListRolesResponse()
  441. r1 = resp.roles.add()
  442. r1.name = "everyone"
  443. r1.display_name = "Everyone"
  444. r1.is_system = True
  445. r1.position = 0
  446. r2 = resp.roles.add()
  447. r2.name = "admin"
  448. r2.display_name = "Admin"
  449. r2.position = 10
  450. _mock_method(client, "roles", "list_roles", resp)
  451. async with client:
  452. roles = await client.list_roles()
  453. assert [r.name for r in roles] == ["everyone", "admin"]
  454. assert roles[0].is_system is True
  455. # --- Asset uploads ---------------------------------------------------
  456. async def test_asset_upload_end_to_end(client, tmp_path):
  457. import hashlib
  458. payload = b"hello world" * 100
  459. file = tmp_path / "greet.txt"
  460. file.write_bytes(payload)
  461. sha = hashlib.sha256(payload).hexdigest()
  462. create_resp = asset_uploads_pb2.CreateUploadResponse()
  463. create_resp.upload.upload_id = "up1"
  464. create_resp.upload.room_id = "r1"
  465. create_resp.upload.status = "ASSET_UPLOAD_STATUS_OPEN"
  466. create_resp.upload.committed_offset = 0
  467. create_resp.upload.size = len(payload)
  468. create_resp.upload.max_chunk_size = 4096
  469. create_resp.upload.sha256 = sha
  470. create_mock = _mock_method(client, "asset_uploads", "create_upload", create_resp)
  471. chunk_resp = asset_uploads_pb2.UploadChunkResponse()
  472. chunk_resp.upload.upload_id = "up1"
  473. chunk_resp.upload.room_id = "r1"
  474. chunk_resp.upload.status = "ASSET_UPLOAD_STATUS_OPEN"
  475. chunk_resp.upload.committed_offset = len(payload)
  476. chunk_resp.upload.size = len(payload)
  477. chunk_resp.upload.max_chunk_size = 4096
  478. chunk_resp.upload.sha256 = sha
  479. _mock_method(client, "asset_uploads", "upload_chunk", chunk_resp)
  480. complete_resp = asset_uploads_pb2.CompleteUploadResponse()
  481. complete_resp.upload.upload_id = "up1"
  482. complete_resp.upload.room_id = "r1"
  483. complete_resp.upload.status = "ASSET_UPLOAD_STATUS_COMPLETED"
  484. complete_resp.upload.committed_offset = len(payload)
  485. complete_resp.upload.size = len(payload)
  486. complete_resp.upload.sha256 = sha
  487. complete_resp.upload.asset_id = "a1"
  488. complete_resp.asset.id = "a1"
  489. complete_resp.asset.filename = "greet.txt"
  490. complete_resp.asset.content_type = "text/plain"
  491. complete_resp.asset.size = len(payload)
  492. _mock_method(client, "asset_uploads", "complete_upload", complete_resp)
  493. async with client:
  494. asset = await client.upload_attachment("r1", file, content_type="text/plain")
  495. assert asset.id == "a1"
  496. assert asset.filename == "greet.txt"
  497. create_req = create_mock.call_args.args[0]
  498. assert create_req.sha256 == sha
  499. assert create_req.size == len(payload)
  500. # --- External identities --------------------------------------------
  501. # --- Admin (structural smoke tests) ----------------------------------
  502. async def test_admin_list_room_groups(client):
  503. resp = room_layout_pb2.ListRoomGroupsResponse()
  504. g = resp.groups.add()
  505. g.id = "g1"
  506. g.name = "General"
  507. room_item = g.items.add()
  508. room_item.room.id = "r1"
  509. room_item.room.kind = "ROOM_KIND_CHANNEL"
  510. room_item.room.name = "chat"
  511. link_item = g.items.add()
  512. link_item.sidebar_link.id = "sl1"
  513. link_item.sidebar_link.label = "Docs"
  514. link_item.sidebar_link.url = "https://x"
  515. g.can_create_room = True
  516. _mock_method(client, "admin_room_layout", "list_room_groups", resp)
  517. async with client:
  518. groups = await client.admin_list_room_groups()
  519. assert len(groups) == 1
  520. assert groups[0].name == "General"
  521. assert groups[0].rooms[0].id == "r1"
  522. assert groups[0].sidebar_links[0].label == "Docs"
  523. assert groups[0].can_create_room is True
  524. async def test_admin_create_sidebar_link(client):
  525. resp = room_layout_pb2.CreateSidebarLinkResponse()
  526. resp.sidebar_link.id = "sl1"
  527. resp.sidebar_link.label = "Docs"
  528. resp.sidebar_link.url = "https://x"
  529. _mock_method(client, "admin_room_layout", "create_sidebar_link", resp)
  530. async with client:
  531. result = await client.admin_create_sidebar_link("g1", "Docs", "https://x")
  532. assert result == {"id": "sl1", "label": "Docs", "url": "https://x"}
  533. async def test_admin_move_room_group(client):
  534. resp = room_layout_pb2.MoveRoomGroupResponse()
  535. g = resp.groups.add()
  536. g.id = "g1"
  537. g.name = "General"
  538. mock = _mock_method(client, "admin_room_layout", "move_room_group", resp)
  539. async with client:
  540. groups = await client.admin_move_room_group("g1", before_group_id="g2")
  541. assert [g.id for g in groups] == ["g1"]
  542. sent = mock.call_args.args[0]
  543. assert sent.group_id == "g1"
  544. assert sent.before_group_id == "g2"
  545. async def test_admin_move_room_group_to_end(client):
  546. resp = room_layout_pb2.MoveRoomGroupResponse()
  547. mock = _mock_method(client, "admin_room_layout", "move_room_group", resp)
  548. async with client:
  549. await client.admin_move_room_group("g1")
  550. sent = mock.call_args.args[0]
  551. assert sent.group_id == "g1"
  552. assert not sent.HasField("before_group_id")
  553. async def test_admin_move_sidebar_item(client):
  554. resp = room_layout_pb2.MoveSidebarItemResponse()
  555. resp.group.id = "g1"
  556. resp.group.name = "General"
  557. mock = _mock_method(client, "admin_room_layout", "move_sidebar_item", resp)
  558. async with client:
  559. group = await client.admin_move_sidebar_item(
  560. ("sidebar_link", "sl1"),
  561. "g1",
  562. before=("room", "r1"),
  563. )
  564. assert group.id == "g1"
  565. sent = mock.call_args.args[0]
  566. assert sent.item.sidebar_link_id == "sl1"
  567. assert sent.before.room_id == "r1"
  568. assert sent.group_id == "g1"
  569. async def test_admin_update_server_config(client):
  570. resp = admin_server_pb2.UpdateServerConfigResponse()
  571. resp.config.server_name = "MyServer"
  572. resp.config.motd = "hi"
  573. resp.public_profile.name = "MyServer"
  574. resp.public_profile.version = "0.4.2"
  575. _mock_method(client, "admin_server", "update_server_config", resp)
  576. async with client:
  577. config, profile = await client.admin_update_server_config(server_name="MyServer", motd="hi")
  578. assert config.server_name == "MyServer"
  579. assert profile.version == "0.4.2"
  580. # --- Neighbors (0.5.0-alpha.3) -----------------------------------------
  581. async def test_list_neighbors_public(client):
  582. resp = discovery_server_pb2.ListNeighborsResponse()
  583. resp.origins.extend(["https://a.example", "https://b.example"])
  584. _mock_method(client, "server_discovery", "list_neighbors", resp)
  585. async with client:
  586. origins = await client.list_neighbors()
  587. assert origins == ["https://a.example", "https://b.example"]
  588. async def test_admin_neighbor_crud(client):
  589. list_resp = admin_server_pb2.ListNeighborsResponse()
  590. n = list_resp.neighbors.add()
  591. n.id = "n1"
  592. n.origin = "https://a.example"
  593. n.revision = "rev1"
  594. _mock_method(client, "admin_server", "list_neighbors", list_resp)
  595. get_resp = admin_server_pb2.GetNeighborResponse()
  596. get_resp.neighbor.id = "n1"
  597. get_resp.neighbor.origin = "https://a.example"
  598. get_resp.neighbor.revision = "rev1"
  599. _mock_method(client, "admin_server", "get_neighbor", get_resp)
  600. create_resp = admin_server_pb2.CreateNeighborResponse()
  601. create_resp.neighbor.id = "n2"
  602. create_resp.neighbor.origin = "https://b.example"
  603. create_resp.neighbor.revision = "rev2"
  604. _mock_method(client, "admin_server", "create_neighbor", create_resp)
  605. update_resp = admin_server_pb2.UpdateNeighborResponse()
  606. update_resp.neighbor.id = "n2"
  607. update_resp.neighbor.origin = "https://b2.example"
  608. update_resp.neighbor.revision = "rev3"
  609. _mock_method(client, "admin_server", "update_neighbor", update_resp)
  610. del_resp = admin_server_pb2.DeleteNeighborResponse()
  611. _mock_method(client, "admin_server", "delete_neighbor", del_resp)
  612. async with client:
  613. listed = await client.admin_list_neighbors()
  614. assert [x.origin for x in listed] == ["https://a.example"]
  615. got = await client.admin_get_neighbor("n1")
  616. assert got is not None and got.revision == "rev1"
  617. created = await client.admin_create_neighbor("https://b.example")
  618. assert created.id == "n2" and created.revision == "rev2"
  619. updated = await client.admin_update_neighbor("n2", "https://b2.example", "rev2")
  620. assert updated.origin == "https://b2.example" and updated.revision == "rev3"
  621. await client.admin_delete_neighbor("n2", "rev3")
  622. async def test_update_settings_share_timezone(client):
  623. resp = account_pb2.UpdateSettingsResponse()
  624. resp.settings.timezone = "Europe/Berlin"
  625. resp.settings.share_timezone = True
  626. _mock_method(client, "account", "update_settings", resp)
  627. async with client:
  628. settings = await client.update_settings(timezone="Europe/Berlin", share_timezone=True)
  629. assert settings.timezone == "Europe/Berlin"
  630. assert settings.share_timezone is True