test_client.py 29 KB

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