def test_happy_path(): app = Kutana() # Simple echo plugin pl = Plugin("") @pl.on_messages() async def __(message, ctx): await ctx.reply(message.text) app.add_plugin(pl) # Simple debug backend debug = Debug( messages=[("message 1", 1), ("message 2", 2)], on_complete=app.stop, ) app.add_backend(debug) # Run application app.run() # Check replies assert debug.answers[1] == [("message 1", (), {})] assert debug.answers[2] == [("message 2", (), {})]
def test_incorrect_values(): app = Kutana() with pytest.raises(ValueError): app.add_backend(None) with pytest.raises(ValueError): app.set_storage('permanent', None) with pytest.raises(ValueError): app.add_plugin(None)
def test_same_plugins_and_backends(): app = Kutana() plugin = Plugin("") backend = Debug([]) app.add_plugin(plugin) with pytest.raises(RuntimeError): app.add_plugin(plugin) app.add_backend(backend) with pytest.raises(RuntimeError): app.add_backend(backend) assert app.get_backends() == [backend]
def test_add_plugins(): added = [] app = Kutana() assert app.get_plugins() == [] app.add_plugin = lambda pl: added.append(pl) app.add_plugins(["a", "b", "c"]) assert added == ["a", "b", "c"]
def test_happy_path(mock_post): group_change_settings_update = { "type": "group_change_settings", "object": { "changes": { "screen_name": {"old_value": "", "new_value": "sdffff23f23"}, "title": {"old_value": "Спасибо", "new_value": "Спасибо 2"} } } } raw_updates = [ {}, {"type": "present"}, group_change_settings_update, MESSAGES["not_message"], MESSAGES["message"], MESSAGES[".echo"], MESSAGES[".echo chat"], MESSAGES[".echo wa"], ] answers = [] updated_longpoll = [] def acquire_updates(content_type=None): if not raw_updates: return {"updates": [], "ts": "100"} if updated_longpoll == [1]: return {"failed": 3} return {"updates": [raw_updates.pop(0)], "ts": "0"} mock_post.return_value.__aenter__.return_value.json = CoroutineMock( side_effect=acquire_updates ) class _VkontakteLongpoll(VkontakteLongpoll): async def _get_response(self, method, kwargs={}): if method == "groups.setLongPollSettings": return {"response": 1} if method == "groups.getById": return { "response": [ {"id": 1, "name": "group", "screen_name": "grp"}, ], } if method == "groups.getLongPollServer": updated_longpoll.append(1) return { "response": { "server": "s", "key": "k", "ts": "1", }, } if method == "execute": answers.extend(kwargs["code"].split("API.")[1:]) return { "response": [1] * kwargs["code"].count("API."), } print(method, kwargs) app = Kutana() vkontakte = _VkontakteLongpoll(token="token") app.add_backend(vkontakte) echo_plugin = Plugin("echo") @echo_plugin.on_commands(["echo", "ec"]) async def __(message, ctx): assert ctx.resolve_screen_name assert ctx.reply await ctx.reply(message.text, attachments=message.attachments) app.add_plugin(echo_plugin) app.get_loop().call_later( delay=vkontakte.api_request_pause * 4, callback=app.stop, ) app.run() assert vkontakte.group_name == "Спасибо 2" assert vkontakte.group_screen_name == "sdffff23f23" assert len(updated_longpoll) == 2 answers.sort() assert len(answers) == 3 assert '{"message": ".echo chat [michaelkrukov|Михаил]",' in answers[0] assert '{"message": ".echo wa",' in answers[1] assert 'attachment": ""' not in answers[1] assert '{"message": ".echo",' in answers[2]
def test_happy_path(): updates = [ MESSAGES["not_message"], MESSAGES["message"], MESSAGES[".echo"], MESSAGES[".echo chat"], MESSAGES["/echo@bot chat"], MESSAGES["/echo chat"], MESSAGES[".echo@bot chat"], MESSAGES["_image"], ] answers = [] class _Telegram(Telegram): async def _request(self, method, kwargs={}): if method == "getMe": return { "first_name": "te", "last_name": "st", "username": "******" } if method == "getUpdates": if not updates: return [] return [updates.pop(0)] if method == "sendMessage": answers.append(("msg", kwargs["text"])) return 1 if method == "sendPhoto": answers.append(("image", kwargs["photo"])) return 1 print(method, kwargs) app = Kutana() telegram = _Telegram(token="token") app.add_backend(telegram) echo_plugin = Plugin("echo") @echo_plugin.on_commands(["echo", "ec"]) async def _(message, ctx): await ctx.reply(message.text) @echo_plugin.on_attachments(["image"]) async def _(message, ctx): await ctx.reply(message.text, attachments=message.attachments[0]) app.add_plugin(echo_plugin) app.get_loop().call_later( delay=0.5, callback=app.stop, ) app.run() answers.sort() assert len(answers) == 5 assert answers[0][0] == "image" assert answers[1] == ("msg", ".echo") assert answers[2] == ("msg", ".echo chat") assert answers[3] == ("msg", "/echo chat") assert answers[4] == ("msg", "/echo chat")