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 make_kutana_no_run(backend_source=None): app = Kutana() debug = Debug(messages=[]) if backend_source: debug.get_identity = lambda: backend_source app.add_backend(debug) async def _handle_update(update): ctx = await Context.create( app=app, config={ "prefixes": (".",), "mention_prefix": ("",), "ignore_initial_spaces": True }, update=update, backend=debug ) return await app._handle_update(update, ctx) def handle_update(update): return app.get_loop().run_until_complete(_handle_update(update)) return app, debug, handle_update
def test_get_backend(): app = Kutana() app.add_backend(Debug([], name="backend1")) app.add_backend(Debug([], name="backend2")) assert app.get_backend("backend1").name == "backend1" assert app.get_backend("backend2").name == "backend2" assert app.get_backend("backend3") is None
def test_not_active_backend(): app = Kutana() app.add_backend(Debug([], name="backend1", active=False)) async def test(): with patch('asyncio.ensure_future') as ensure_future: await app._on_start(None) ensure_future.assert_not_called asyncio.get_event_loop().run_until_complete(test())
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 make_kutana(messages): app = Kutana() debug = Debug( messages=messages, on_complete=app.stop, ) app.add_backend(debug) return app, debug
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 main(): app = Kutana() # set_logger_level(logging.DEBUG) backend = Vkontakte(SHUF_SETTINGS['TOKEN']) app.add_backend(backend) app.config['settings'] = SHUF_SETTINGS app.config['prefixes'] = ('еш ', 'есб ', 'esb ', 'ешаф ', 'eshuf ' ) # ('шаф ', 'sb ', 'шб ', 'shuf ', 'shufbot ') app.config['inform_time'] = time(20, 50) # 12:00 app.config['votekick_time'] = 5 * 60 init_db(app) app.add_plugins(load_plugins(os.path.join(os.curdir, 'bot', 'plugins'))) return app
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_add_backend(): app = Kutana() with pytest.raises(ValueError): app.add_backend(None)
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")
import json from kutana import Kutana, load_plugins from kutana.backends import Vkontakte # Import configuration with open("config.json") as fh: config = json.load(fh) # Create application app = Kutana() # Add manager to application app.add_backend(Vkontakte(token=config["vk_token"])) # Load and register plugins app.add_plugins(load_plugins("plugins/")) if __name__ == "__main__": # Run application app.run()
from ai_dungeon import config from ai_dungeon.storage.redis import RedisStorage from kutana import Kutana, load_plugins from kutana.backends import Vkontakte, Telegram import logging logging.basicConfig(level=logging.DEBUG) # Create application if config.REDIS_URL is not None: app = Kutana(storage=RedisStorage(config.REDIS_URL)) else: app = Kutana() # Add manager to application if config.TELEGRAM_TOKEN: app.add_backend(Telegram(token=config.TELEGRAM_TOKEN)) if config.VK_TOKEN: app.add_backend(Vkontakte(token=config.VK_TOKEN)) # Load and register plugins app.add_plugins(load_plugins("plugins/")) if __name__ == "__main__": # Run application app.run()
import os from pathlib import Path from kutana import Kutana, load_plugins from kutana.backends import Telegram import dotenv env = dotenv.dotenv_values() if Path(".env").exists() else os.environ # Create application app = Kutana() # Add manager to application app.add_backend(Telegram(env.get("TELEGRAM_TOKEN"), proxy=env.get("PROXY"))) # Load and register plugins app.add_plugins(load_plugins("plugins/")) if __name__ == "__main__": # Run application app.run()
import json from kutana import Kutana, load_plugins from kutana.backends import Vkontakte, Telegram # Import configuration with open("configuration.json") as fh: config = json.load(fh) # Create application app = Kutana() # Set config app.config.update({ "vk_chat_id": config["vk_chat_id"], "tg_chat_id": config["tg_chat_id"] }) # Add manager to application app.add_backend(Telegram(token=config["tg_token"])) app.add_backend(Vkontakte(token=config["vk_token"])) # Load and register plugins app.add_plugins(load_plugins("plugins/")) if __name__ == "__main__": # Run application app.run()
import json from kutana import Kutana, load_plugins from kutana.backends import Vkontakte, VkontakteCallback, Telegram # Import configuration with open("config.json") as fh: config = json.load(fh) # Create application app = Kutana() # Add backends to application if "vk" in config: if "address" in config["vk"]: app.add_backend( VkontakteCallback(token=config["vk"]["token"], address=config["vk"]["address"])) else: app.add_backend(Vkontakte(token=config["vk"]["token"])) if "tg" in config: app.add_backend(Telegram(token=config["tg"]["token"])) # Load and register plugins app.add_plugins(load_plugins("plugins/")) if __name__ == "__main__": # Run application app.run()