test_client.py 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779
  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. member_directory_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. resp = messages_pb2.DeleteMessageResponse(deleted=True)
  138. _mock_method(client, "messages", "delete_message", resp)
  139. async with client:
  140. assert await client.delete_message("r1", "e1") is True
  141. async def test_add_reaction(client):
  142. resp = reactions_pb2.AddReactionResponse(added=True)
  143. _mock_method(client, "messages", "add_reaction", resp)
  144. async with client:
  145. assert await client.add_reaction("r1", "e1", "thumbsup") is True
  146. async def test_remove_reaction(client):
  147. resp = reactions_pb2.RemoveReactionResponse(removed=True)
  148. _mock_method(client, "messages", "remove_reaction", resp)
  149. async with client:
  150. assert await client.remove_reaction("r1", "e1", "thumbsup") is True
  151. # --- Room lifecycle ---------------------------------------------------
  152. async def test_create_room(client):
  153. resp = rooms_pb2.CreateRoomResponse()
  154. resp.room.id = "r1"
  155. resp.room.kind = "ROOM_KIND_CHANNEL"
  156. resp.room.name = "general"
  157. resp.room.group_id = "g1"
  158. _mock_method(client, "rooms", "create_room", resp)
  159. async with client:
  160. room = await client.create_room("general", "g1")
  161. assert room.id == "r1"
  162. assert room.name == "general"
  163. assert room.group_id == "g1"
  164. async def test_archive_room(client):
  165. resp = rooms_pb2.ArchiveRoomResponse()
  166. resp.room.id = "r1"
  167. resp.room.name = "general"
  168. resp.room.archived = True
  169. _mock_method(client, "rooms", "archive_room", resp)
  170. async with client:
  171. room = await client.archive_room("r1")
  172. assert room.archived is True
  173. async def test_ban_member(client):
  174. resp = rooms_pb2.BanMemberResponse(banned=True)
  175. _mock_method(client, "rooms", "ban_member", resp)
  176. async with client:
  177. assert await client.ban_member("r1", "u2", "spam") is True
  178. async def test_start_dm(client):
  179. resp = rooms_pb2.StartDMResponse()
  180. resp.room.id = "r1"
  181. resp.room.kind = "ROOM_KIND_DM"
  182. _mock_method(client, "rooms", "start_dm", resp)
  183. async with client:
  184. room = await client.start_dm(["u1", "u2"])
  185. assert room.id == "r1"
  186. assert room.kind is RoomKind.DM
  187. # --- Room timeline ----------------------------------------------------
  188. async def test_get_room_events(client):
  189. resp = room_timeline_pb2.GetRoomEventsResponse()
  190. event = resp.page.events.add()
  191. event.id = "e1"
  192. event.actor_id = "u1"
  193. event.created_at.FromJsonString("2026-01-01T00:00:00Z")
  194. event.message_posted.message.id = "e1"
  195. event.message_posted.message.room_id = "r1"
  196. event.message_posted.message.actor_id = "u1"
  197. event.message_posted.message.body = "hi"
  198. event.message_posted.message.created_at.FromJsonString("2026-01-01T00:00:00Z")
  199. resp.page.start_cursor = "cA"
  200. resp.page.end_cursor = "cB"
  201. resp.page.includes.users["u1"].id = "u1"
  202. resp.page.includes.users["u1"].login = "alice"
  203. resp.page.includes.users["u1"].display_name = "Alice"
  204. _mock_method(client, "rooms", "get_room_events", resp)
  205. async with client:
  206. page = await client.get_room_events("r1", limit=50)
  207. assert len(page.events) == 1
  208. assert page.events[0].kind == "message_posted"
  209. assert page.events[0].message is not None
  210. assert page.events[0].message.body == "hi"
  211. assert page.users_by_id["u1"].login == "alice"
  212. async def test_mark_room_as_read(client):
  213. resp = read_state_pb2.MarkRoomAsReadResponse()
  214. resp.last_read_at.FromJsonString("2026-01-01T00:00:00Z")
  215. resp.previous_last_read_at.FromJsonString("2025-12-31T00:00:00Z")
  216. _mock_method(client, "rooms", "mark_room_as_read", resp)
  217. async with client:
  218. last, previous = await client.mark_room_as_read("r1", up_to_event_id="e1")
  219. assert last is not None
  220. assert previous is not None
  221. # --- Profile / account ------------------------------------------------
  222. async def test_update_profile(client):
  223. resp = account_pb2.UpdateProfileResponse()
  224. resp.user.id = "u1"
  225. resp.user.login = "newname"
  226. resp.user.display_name = "New Name"
  227. resp.user.presence_status = "PRESENCE_STATUS_ONLINE"
  228. _mock_method(client, "account", "update_profile", resp)
  229. async with client:
  230. user = await client.update_profile(login="newname", display_name="New Name")
  231. assert user.login == "newname"
  232. assert user.display_name == "New Name"
  233. async def test_upload_avatar(client, tmp_path):
  234. resp = account_pb2.UploadAvatarResponse()
  235. resp.user.id = "u1"
  236. resp.user.login = "alice"
  237. resp.user.display_name = "Alice"
  238. resp.user.avatar_url = "https://example.com/avatar.jpg"
  239. mock = _mock_method(client, "account", "upload_avatar", resp)
  240. avatar = tmp_path / "avatar.png"
  241. avatar.write_bytes(b"\x89PNGfake")
  242. async with client:
  243. user = await client.upload_avatar(str(avatar), content_type="image/png")
  244. assert user.avatar_url == "https://example.com/avatar.jpg"
  245. req = mock.call_args.args[0]
  246. assert req.image.image == b"\x89PNGfake"
  247. assert req.image.filename == "avatar.png"
  248. assert req.image.content_type == "image/png"
  249. async def test_update_presence(client):
  250. resp = presence_pb2.UpdatePresenceResponse()
  251. resp.status = "PRESENCE_STATUS_ONLINE"
  252. _mock_method(client, "account", "update_presence", resp)
  253. async with client:
  254. result = await client.update_presence(PresenceStatus.ONLINE)
  255. assert result is PresenceStatus.ONLINE
  256. async def test_update_presence_offline_rejected(client):
  257. async with client:
  258. with pytest.raises(ValueError):
  259. await client.update_presence(PresenceStatus.OFFLINE)
  260. async def test_update_settings(client):
  261. resp = account_pb2.UpdateSettingsResponse()
  262. resp.settings.timezone = "Europe/Rome"
  263. resp.settings.time_format = "TIME_FORMAT_24_HOUR"
  264. _mock_method(client, "account", "update_settings", resp)
  265. async with client:
  266. settings = await client.update_settings(
  267. timezone="Europe/Rome", time_format=TimeFormat.HOUR_24
  268. )
  269. assert settings.timezone == "Europe/Rome"
  270. assert settings.time_format is TimeFormat.HOUR_24
  271. # --- Users -------------------------------------------------------------
  272. async def test_get_user_by_id(client):
  273. resp = member_directory_pb2.GetUserResponse()
  274. resp.user.user.id = "u1"
  275. resp.user.user.login = "alice"
  276. resp.user.user.display_name = "Alice"
  277. resp.user.user.presence_status = "PRESENCE_STATUS_ONLINE"
  278. resp.user.roles.append("everyone")
  279. _mock_method(client, "users", "get_user", resp)
  280. async with client:
  281. member = await client.get_user(user_id="u1")
  282. assert member is not None
  283. assert member.user is not None
  284. assert member.user.login == "alice"
  285. assert "everyone" in member.roles
  286. async def test_get_user_requires_one_target(client):
  287. with pytest.raises(ValueError):
  288. await client.get_user()
  289. with pytest.raises(ValueError):
  290. await client.get_user(user_id="u1", login="alice")
  291. # --- Threads ----------------------------------------------------------
  292. async def test_list_followed_threads(client):
  293. resp = threads_pb2.ListFollowedThreadsResponse()
  294. ft = resp.threads.add()
  295. ft.room.id = "r1"
  296. ft.room.name = "general"
  297. ft.root_message.id = "e1"
  298. ft.root_message.room_id = "r1"
  299. ft.root_message.actor_id = "u1"
  300. ft.root_message.body = "root"
  301. ft.root_message.created_at.FromJsonString("2026-01-01T00:00:00Z")
  302. ft.thread.thread_root_event_id = "e1"
  303. ft.thread.reply_count = 3
  304. ft.thread.last_reply_at.FromJsonString("2026-01-02T00:00:00Z")
  305. ft.thread.viewer_state.is_following = True
  306. ft.thread.viewer_state.has_unread = False
  307. resp.page.total_count = 1
  308. resp.page.has_more = False
  309. _mock_method(client, "threads", "list_followed_threads", resp)
  310. async with client:
  311. page = await client.list_followed_threads()
  312. assert page.page.total_count == 1
  313. assert page.threads[0].thread is not None
  314. assert page.threads[0].thread.reply_count == 3
  315. assert page.threads[0].thread.is_following is True
  316. async def test_follow_and_unfollow_thread(client):
  317. follow_resp = threads_pb2.FollowThreadResponse(following=True)
  318. unfollow_resp = threads_pb2.UnfollowThreadResponse(following=False)
  319. _mock_method(client, "threads", "follow_thread", follow_resp)
  320. _mock_method(client, "threads", "unfollow_thread", unfollow_resp)
  321. async with client:
  322. assert await client.follow_thread("r1", "e1") is True
  323. assert await client.unfollow_thread("r1", "e1") is False
  324. # --- Notifications ----------------------------------------------------
  325. async def test_list_notification_occurrences(client):
  326. resp = notifications_pb2.ListNotificationOccurrencesResponse()
  327. occ = resp.occurrences.add()
  328. occ.id = "n1"
  329. occ.created_at.FromJsonString("2026-01-01T00:00:00Z")
  330. occ.actor.id = "u1"
  331. occ.actor.login = "alice"
  332. occ.actor.display_name = "Alice"
  333. occ.unread = True
  334. sig = occ.signal.direct_mention_received
  335. sig.message.room.id = "r1"
  336. sig.message.room.name = "general"
  337. sig.message.event_id = "e1"
  338. resp.page.total_count = 1
  339. resp.unread_count = 1
  340. _mock_method(client, "notifications", "list_notification_occurrences", resp)
  341. async with client:
  342. page = await client.list_notification_occurrences()
  343. assert page.page.total_count == 1
  344. assert page.unread_count == 1
  345. assert page.occurrences[0].signal is not None
  346. assert page.occurrences[0].signal.kind == "direct_mention_received"
  347. assert page.occurrences[0].signal.message is not None
  348. assert page.occurrences[0].signal.message.event_id == "e1"
  349. assert page.occurrences[0].signal.message.room is not None
  350. assert page.occurrences[0].signal.message.room.name == "general"
  351. async def test_mark_and_delete_notification_occurrence(client):
  352. read_resp = notifications_pb2.MarkNotificationReadResponse()
  353. read_resp.occurrence.id = "n1"
  354. read_resp.occurrence.unread = False
  355. _mock_method(client, "notifications", "mark_notification_read", read_resp)
  356. del_resp = notifications_pb2.DeleteNotificationOccurrenceResponse(deleted=True)
  357. _mock_method(client, "notifications", "delete_notification_occurrence", del_resp)
  358. all_resp = notifications_pb2.DeleteAllNotificationOccurrencesResponse(deleted_count=3)
  359. _mock_method(client, "notifications", "delete_all_notification_occurrences", all_resp)
  360. async with client:
  361. occ = await client.mark_notification_read("n1")
  362. assert occ is not None and occ.unread is False
  363. assert await client.delete_notification_occurrence("n1") is True
  364. assert await client.delete_all_notification_occurrences() == 3
  365. async def test_notification_preference(client):
  366. resp = notification_preferences_pb2.UpdateNotificationPreferenceResponse()
  367. resp.preference.level = "NOTIFICATION_LEVEL_MUTED"
  368. resp.preference.effective_level = "NOTIFICATION_LEVEL_MUTED"
  369. _mock_method(
  370. client,
  371. "notification_prefs",
  372. "update_room_notification_preference",
  373. resp,
  374. )
  375. async with client:
  376. pref = await client.update_room_notification_preference("r1", NotificationLevel.MUTED)
  377. assert pref.level is NotificationLevel.MUTED
  378. assert pref.effective_level is NotificationLevel.MUTED
  379. # --- Push -------------------------------------------------------------
  380. async def test_subscribe_push(client):
  381. resp = push_notifications_pb2.SubscribePushResponse(subscribed=True)
  382. _mock_method(client, "push", "subscribe", resp)
  383. async with client:
  384. assert await client.subscribe_push("https://push", "k", "a") is True
  385. # --- Voice calls ------------------------------------------------------
  386. async def test_active_calls(client):
  387. resp = voice_calls_pb2.ListActiveCallsResponse()
  388. call = resp.calls.add()
  389. call.call_id = "c1"
  390. call.room.id = "r1"
  391. call.room.name = "voice"
  392. p = call.participants.add()
  393. p.user.id = "u1"
  394. p.user.login = "alice"
  395. p.user.display_name = "Alice"
  396. _mock_method(client, "voice_calls", "list_active_calls", resp)
  397. async with client:
  398. calls = await client.list_active_calls()
  399. assert len(calls) == 1
  400. assert calls[0].call_id == "c1"
  401. assert calls[0].room is not None
  402. assert calls[0].room.name == "voice"
  403. # --- Assets -----------------------------------------------------------
  404. async def test_get_asset_with_thumbnail(client):
  405. resp = attachments_pb2.GetAssetResponse()
  406. resp.asset.id = "a1"
  407. resp.asset.filename = "pic.png"
  408. resp.asset.content_type = "image/png"
  409. resp.asset.size = 1024
  410. resp.asset.asset_url.url = "https://cdn/pic.png"
  411. mock = _mock_method(client, "assets", "get_asset", resp)
  412. async with client:
  413. asset = await client.get_asset(
  414. "r1",
  415. "a1",
  416. thumbnail=ImageTransformOptions(128, 128, ImageFitMode.COVER),
  417. )
  418. assert asset is not None
  419. assert asset.filename == "pic.png"
  420. assert asset.asset_url is not None
  421. assert asset.asset_url.url == "https://cdn/pic.png"
  422. # verify the request encoded thumbnail options
  423. req = mock.call_args.args[0]
  424. assert req.thumbnail.width == 128
  425. assert req.thumbnail.height == 128
  426. # --- Roles ------------------------------------------------------------
  427. async def test_list_roles(client):
  428. resp = roles_pb2.ListRolesResponse()
  429. r1 = resp.roles.add()
  430. r1.name = "everyone"
  431. r1.display_name = "Everyone"
  432. r1.is_system = True
  433. r1.position = 0
  434. r2 = resp.roles.add()
  435. r2.name = "admin"
  436. r2.display_name = "Admin"
  437. r2.position = 10
  438. _mock_method(client, "roles", "list_roles", resp)
  439. async with client:
  440. roles = await client.list_roles()
  441. assert [r.name for r in roles] == ["everyone", "admin"]
  442. assert roles[0].is_system is True
  443. # --- Asset uploads ---------------------------------------------------
  444. async def test_asset_upload_end_to_end(client, tmp_path):
  445. import hashlib
  446. payload = b"hello world" * 100
  447. file = tmp_path / "greet.txt"
  448. file.write_bytes(payload)
  449. sha = hashlib.sha256(payload).hexdigest()
  450. create_resp = asset_uploads_pb2.CreateUploadResponse()
  451. create_resp.upload.upload_id = "up1"
  452. create_resp.upload.room_id = "r1"
  453. create_resp.upload.status = "ASSET_UPLOAD_STATUS_OPEN"
  454. create_resp.upload.committed_offset = 0
  455. create_resp.upload.size = len(payload)
  456. create_resp.upload.max_chunk_size = 4096
  457. create_resp.upload.sha256 = sha
  458. create_mock = _mock_method(client, "asset_uploads", "create_upload", create_resp)
  459. chunk_resp = asset_uploads_pb2.UploadChunkResponse()
  460. chunk_resp.upload.upload_id = "up1"
  461. chunk_resp.upload.room_id = "r1"
  462. chunk_resp.upload.status = "ASSET_UPLOAD_STATUS_OPEN"
  463. chunk_resp.upload.committed_offset = len(payload)
  464. chunk_resp.upload.size = len(payload)
  465. chunk_resp.upload.max_chunk_size = 4096
  466. chunk_resp.upload.sha256 = sha
  467. _mock_method(client, "asset_uploads", "upload_chunk", chunk_resp)
  468. complete_resp = asset_uploads_pb2.CompleteUploadResponse()
  469. complete_resp.upload.upload_id = "up1"
  470. complete_resp.upload.room_id = "r1"
  471. complete_resp.upload.status = "ASSET_UPLOAD_STATUS_COMPLETED"
  472. complete_resp.upload.committed_offset = len(payload)
  473. complete_resp.upload.size = len(payload)
  474. complete_resp.upload.sha256 = sha
  475. complete_resp.upload.asset_id = "a1"
  476. complete_resp.asset.id = "a1"
  477. complete_resp.asset.filename = "greet.txt"
  478. complete_resp.asset.content_type = "text/plain"
  479. complete_resp.asset.size = len(payload)
  480. _mock_method(client, "asset_uploads", "complete_upload", complete_resp)
  481. async with client:
  482. asset = await client.upload_attachment("r1", file, content_type="text/plain")
  483. assert asset.id == "a1"
  484. assert asset.filename == "greet.txt"
  485. create_req = create_mock.call_args.args[0]
  486. assert create_req.sha256 == sha
  487. assert create_req.size == len(payload)
  488. # --- External identities --------------------------------------------
  489. # --- Admin (structural smoke tests) ----------------------------------
  490. async def test_admin_list_room_groups(client):
  491. resp = room_layout_pb2.ListRoomGroupsResponse()
  492. g = resp.groups.add()
  493. g.id = "g1"
  494. g.name = "General"
  495. room_item = g.items.add()
  496. room_item.room.id = "r1"
  497. room_item.room.kind = "ROOM_KIND_CHANNEL"
  498. room_item.room.name = "chat"
  499. link_item = g.items.add()
  500. link_item.sidebar_link.id = "sl1"
  501. link_item.sidebar_link.label = "Docs"
  502. link_item.sidebar_link.url = "https://x"
  503. g.can_create_room = True
  504. _mock_method(client, "admin_room_layout", "list_room_groups", resp)
  505. async with client:
  506. groups = await client.admin_list_room_groups()
  507. assert len(groups) == 1
  508. assert groups[0].name == "General"
  509. assert groups[0].rooms[0].id == "r1"
  510. assert groups[0].sidebar_links[0].label == "Docs"
  511. assert groups[0].can_create_room is True
  512. async def test_admin_create_sidebar_link(client):
  513. resp = room_layout_pb2.CreateSidebarLinkResponse()
  514. resp.sidebar_link.id = "sl1"
  515. resp.sidebar_link.label = "Docs"
  516. resp.sidebar_link.url = "https://x"
  517. _mock_method(client, "admin_room_layout", "create_sidebar_link", resp)
  518. async with client:
  519. result = await client.admin_create_sidebar_link("g1", "Docs", "https://x")
  520. assert result == {"id": "sl1", "label": "Docs", "url": "https://x"}
  521. async def test_admin_update_server_config(client):
  522. resp = admin_server_pb2.UpdateServerConfigResponse()
  523. resp.config.server_name = "MyServer"
  524. resp.config.motd = "hi"
  525. resp.public_profile.name = "MyServer"
  526. resp.public_profile.version = "0.4.2"
  527. _mock_method(client, "admin_server", "update_server_config", resp)
  528. async with client:
  529. config, profile = await client.admin_update_server_config(server_name="MyServer", motd="hi")
  530. assert config.server_name == "MyServer"
  531. assert profile.version == "0.4.2"
  532. # --- Neighbors (0.5.0-alpha.3) -----------------------------------------
  533. async def test_list_neighbors_public(client):
  534. resp = discovery_server_pb2.ListNeighborsResponse()
  535. resp.origins.extend(["https://a.example", "https://b.example"])
  536. _mock_method(client, "server_discovery", "list_neighbors", resp)
  537. async with client:
  538. origins = await client.list_neighbors()
  539. assert origins == ["https://a.example", "https://b.example"]
  540. async def test_admin_neighbor_crud(client):
  541. list_resp = admin_server_pb2.ListNeighborsResponse()
  542. n = list_resp.neighbors.add()
  543. n.id = "n1"
  544. n.origin = "https://a.example"
  545. n.revision = "rev1"
  546. _mock_method(client, "admin_server", "list_neighbors", list_resp)
  547. get_resp = admin_server_pb2.GetNeighborResponse()
  548. get_resp.neighbor.id = "n1"
  549. get_resp.neighbor.origin = "https://a.example"
  550. get_resp.neighbor.revision = "rev1"
  551. _mock_method(client, "admin_server", "get_neighbor", get_resp)
  552. create_resp = admin_server_pb2.CreateNeighborResponse()
  553. create_resp.neighbor.id = "n2"
  554. create_resp.neighbor.origin = "https://b.example"
  555. create_resp.neighbor.revision = "rev2"
  556. _mock_method(client, "admin_server", "create_neighbor", create_resp)
  557. update_resp = admin_server_pb2.UpdateNeighborResponse()
  558. update_resp.neighbor.id = "n2"
  559. update_resp.neighbor.origin = "https://b2.example"
  560. update_resp.neighbor.revision = "rev3"
  561. _mock_method(client, "admin_server", "update_neighbor", update_resp)
  562. del_resp = admin_server_pb2.DeleteNeighborResponse()
  563. _mock_method(client, "admin_server", "delete_neighbor", del_resp)
  564. async with client:
  565. listed = await client.admin_list_neighbors()
  566. assert [x.origin for x in listed] == ["https://a.example"]
  567. got = await client.admin_get_neighbor("n1")
  568. assert got is not None and got.revision == "rev1"
  569. created = await client.admin_create_neighbor("https://b.example")
  570. assert created.id == "n2" and created.revision == "rev2"
  571. updated = await client.admin_update_neighbor("n2", "https://b2.example", "rev2")
  572. assert updated.origin == "https://b2.example" and updated.revision == "rev3"
  573. await client.admin_delete_neighbor("n2", "rev3")
  574. async def test_update_settings_share_timezone(client):
  575. resp = account_pb2.UpdateSettingsResponse()
  576. resp.settings.timezone = "Europe/Berlin"
  577. resp.settings.share_timezone = True
  578. _mock_method(client, "account", "update_settings", resp)
  579. async with client:
  580. settings = await client.update_settings(timezone="Europe/Berlin", share_timezone=True)
  581. assert settings.timezone == "Europe/Berlin"
  582. assert settings.share_timezone is True