Spaces:
Paused
Paused
| """Taking a channel out of the bot: the bot asks whether the channel's saved posts (its data in My posts) are deleted as | |
| well - Yes removes the channel and the data, No removes only the channel and keeps the data.""" | |
| import asyncio | |
| from app.db import utcnow | |
| from app.sync_engine import sync_channel | |
| from .harness import App, norm | |
| from .test_sync import old_msg, opts, world | |
| def run(coro): | |
| return asyncio.run(coro) | |
| async def make_app(tmp_path, **cfg): | |
| """Two channels: "Anime Channel" (id 1) has three published posts and a draft, "Other Channel" (id 2) has two posts; | |
| channel 1 also has what the sync keeps (where it stopped, a post it must forget) and a timed deletion.""" | |
| app = App(tmp_path, edit_delay=0, **cfg) | |
| await app.start() | |
| await app.db.save_channel(1, 5, "Anime Channel", "animech", 1) | |
| await app.db.save_channel(2, 6, "Other Channel", "otherch", 1) | |
| for cid, n in ((1, 3), (2, 2)): | |
| for i in range(1, n + 1): | |
| await app.db.create_post( | |
| channel_id=cid, message_id=i, status="sent", source="adopted", text=f"post {cid}/{i}", entities=[], buttons=[] | |
| ) | |
| await app.db.create_post(channel_id=1, status="draft", source="bot", text="a draft", entities=[], buttons=[]) | |
| for cid in (1, 2): | |
| await app.db.save_sync(cid, last_top=3, last_pts=3) | |
| await app.db.start_history(cid, 3) | |
| await app.db.ignore_post(cid, 9) | |
| app.ctx.sync_ignore.add((cid, 9)) | |
| await app.db.schedule_deletes( | |
| [dict(channel_id=1, message_id=2, delete_at=utcnow(), group_id="g", created_by=1)] | |
| ) | |
| return app | |
| async def tap_channel(app, label_part="1"): | |
| await app.text("/channels") | |
| await app.press(app.out.callback_exact(label_part)) | |
| # ============================================================================================ the question | |
| def test_the_owner_is_asked_whether_the_saved_posts_go_too(tmp_path): | |
| async def go(): | |
| app = await make_app(tmp_path) | |
| await tap_channel(app) | |
| text = norm(app.out.last_text) | |
| assert "Take Anime Channel out of the bot" in text | |
| assert "Do you also want to delete its saved posts" in text and "4 post(s)" in text # three published + a draft | |
| assert "Nothing is deleted in the channel itself" in text | |
| assert app.out.last_buttons() == [["🗑 Yes, delete them too", "📦 No, keep them"], ["🔙 Back"]] | |
| # the question alone changes nothing | |
| assert [c.id for c in await app.db.list_channels()] == [1, 2] and await app.db.count_channel_posts(1) == 4 | |
| await app.press(app.out.callback_exact("🔙 Back")) | |
| assert "Your channels" in norm(app.out.last_text) | |
| await app.db.close() | |
| run(go()) | |
| def test_a_channel_without_saved_posts_has_nothing_to_ask(tmp_path): | |
| async def go(): | |
| app = await make_app(tmp_path) | |
| await app.db.save_channel(3, 7, "Empty Channel", "emptych", 1) | |
| await app.text("/channels") | |
| await app.press(app.out.callback_exact("2")) # Anime, Empty, Other: the second one | |
| text = norm(app.out.last_text) | |
| assert "Take Empty Channel out of the bot?" in text and "no saved posts" in text | |
| assert app.out.last_buttons() == [["✅ Yes, remove it", "🔙 Back"]] | |
| await app.press(app.out.callback_exact("✅ Yes, remove it")) | |
| assert [c.id for c in await app.db.list_channels()] == [1, 2] | |
| await app.db.close() | |
| run(go()) | |
| # ================================================================================================ yes | |
| def test_yes_removes_the_channel_and_deletes_its_saved_posts(tmp_path): | |
| async def go(): | |
| app = await make_app(tmp_path) | |
| await tap_channel(app) | |
| await app.press(app.out.callback_data("Yes, delete them too")) | |
| text = norm(app.out.last_text) | |
| assert "Anime Channel was removed from the bot" in text and "4 saved post(s) were deleted from My posts" in text | |
| assert "Nothing was deleted in the channel" in text | |
| assert [c.id for c in await app.db.list_channels()] == [2] | |
| row = await app.db.get_channel(1) | |
| assert row is not None and not row.active # switched off, not erased: a repost or shift that names it still finds it | |
| assert await app.db.count_channel_posts(1) == 0 and await app.db.sent_posts(1) == [] | |
| _, total = await app.db.list_posts(1, "draft", 0, 10) | |
| assert total == 0 | |
| # what the sync kept about the channel goes too, so it starts from scratch when it is added again | |
| assert await app.db.get_sync(1) is None and await app.db.get_history(1) is None | |
| assert await app.db.ignored_ids(1) == set() and not any(k[0] == 1 for k in app.ctx.sync_ignore) | |
| # ... and nothing of the other channel is touched | |
| assert await app.db.count_channel_posts(2) == 2 | |
| assert await app.db.get_sync(2) is not None and await app.db.get_history(2) is not None | |
| assert await app.db.ignored_ids(2) == {9} and (2, 9) in app.ctx.sync_ignore | |
| # a timed deletion is a promise about a message that is still in the channel: it stays | |
| assert [r.message_id for r in await app.db.pending_deletes()] == [2] | |
| await app.db.close() | |
| run(go()) | |
| def test_nothing_is_deleted_in_telegram_when_the_data_goes(tmp_path): | |
| async def go(): | |
| app = await make_app(tmp_path) | |
| before = len(app.tg.requests) | |
| await tap_channel(app) | |
| await app.press(app.out.callback_data("Yes, delete them too")) | |
| assert app.tg.deleted == [] and len(app.tg.requests) == before | |
| await app.db.close() | |
| run(go()) | |
| def test_the_posts_are_not_deleted_while_a_look_at_the_channels_is_running(tmp_path): | |
| async def go(): | |
| app = await make_app(tmp_path) | |
| lock = app.ctx.syncer.exclusive() | |
| await lock.acquire() # a look at the channels is in progress: it could save posts again | |
| task = asyncio.create_task(app.press("chy:1:1")) | |
| await asyncio.sleep(0.2) | |
| assert not task.done() and await app.db.count_channel_posts(1) == 4 | |
| lock.release() | |
| await task | |
| assert await app.db.count_channel_posts(1) == 0 | |
| await app.db.close() | |
| run(go()) | |
| # =================================================================================================== no | |
| def test_no_removes_only_the_channel_and_keeps_its_saved_posts(tmp_path): | |
| async def go(): | |
| app = await make_app(tmp_path) | |
| await tap_channel(app) | |
| await app.press(app.out.callback_data("No, keep them")) | |
| text = norm(app.out.last_text) | |
| assert "Anime Channel was removed from the bot" in text and "saved posts are kept" in text | |
| assert [c.id for c in await app.db.list_channels()] == [2] | |
| assert await app.db.count_channel_posts(1) == 4 and await app.db.get_sync(1) is not None | |
| assert await app.db.ignored_ids(1) == {9} and (1, 9) in app.ctx.sync_ignore | |
| # added again: everything is there again | |
| await app.db.save_channel(1, 5, "Anime Channel", "animech", 1) | |
| assert [c.id for c in await app.db.list_channels()] == [1, 2] and await app.db.count_channel_posts(1) == 4 | |
| await app.db.close() | |
| run(go()) | |
| def test_a_button_from_the_old_question_still_removes_only_the_channel(tmp_path): | |
| async def go(): | |
| app = await make_app(tmp_path) | |
| await app.press("chy:1") # what the earlier version of the bot put on its button | |
| assert [c.id for c in await app.db.list_channels()] == [2] | |
| assert await app.db.count_channel_posts(1) == 4 | |
| assert "saved posts are kept" in norm(app.out.last_text) | |
| await app.db.close() | |
| run(go()) | |
| # ============================================================================================ who may do it | |
| def test_only_an_owner_can_remove_a_channel_or_delete_its_data(tmp_path): | |
| async def go(): | |
| app = await make_app(tmp_path, owner=7, admins=frozenset({1})) | |
| app.uid = 1 # an admin, not an owner | |
| for data in ("chd:1", "chy:1:1", "chy:1:0", "chy:1"): | |
| await app.press(data) | |
| assert [c.id for c in await app.db.list_channels()] == [1, 2] and await app.db.count_channel_posts(1) == 4 | |
| await app.db.close() | |
| run(go()) | |
| # ========================================================================================= coming back | |
| def test_a_channel_that_comes_back_after_its_data_was_deleted_is_read_in_again(tmp_path): | |
| async def go(): | |
| chan, db, ch = await world(tmp_path, {i: old_msg(i, f"older post {i}") for i in range(1, 6)}, save=False) | |
| rep = await sync_channel(chan, db, ch, opts()) | |
| assert rep.first_look and rep.imported == 5 | |
| assert await db.delete_channel_data(1) == 5 | |
| assert await db.sent_posts(1) == [] and await db.get_sync(1) is None and await db.get_history(1) is None | |
| rep = await sync_channel(chan, db, ch, opts()) # as if it had never been looked at | |
| assert rep.first_look and rep.imported == 5 and rep.history_done | |
| assert len(await db.sent_posts(1)) == 5 | |
| await db.close() | |
| run(go()) | |