async def test_start_find_bot_id(self, aiohttp_server, find_bot_id_query): bot = SirBot() bot.load_plugin( SlackPlugin(token="foo", verify="bar", bot_user_id="baz")) bot["plugins"]["slack"].api.query = find_bot_id_query await aiohttp_server(bot) assert bot["plugins"]["slack"].bot_id == "B00000000"
async def test_register_admin_message_no_admin(self, caplog): async def handler(): pass bot = SirBot() bot.load_plugin(SlackPlugin(token='foo', verify='bar', bot_user_id='baz', bot_id='boo')) bot['plugins']['slack'].on_message('hello', handler, admin=True) assert 'Slack admins ids are not set. Admin limited endpoint will not work.' in caplog.text
def bot(): b = SirBot() slack = SlackPlugin(token='token', verify='supersecuretoken', bot_user_id='bot_user_id', bot_id='bot_id') b.load_plugin(slack) return b
async def bot(request, loop): b = SirBot() pg = PgPlugin( dsn="postgres://[email protected]:5432/sirbot", sql_migration_directory="tests/data/sql", version="0.0.2", ) b.load_plugin(pg) yield b
async def test_load_plugin_no_name(self): class MyPlugin: def __init__(self): pass def load(self, test_bot): pass bot = SirBot() with pytest.raises(AttributeError): bot.load_plugin(MyPlugin())
async def bot_signing(): b = SirBot() b.load_plugin( SlackPlugin( token="foo", signing_secret="sharedsigningkey", bot_user_id="baz", bot_id="boo", admins=["aaa", "bbb"], )) return b
async def bot(): b = SirBot() b.load_plugin( SlackPlugin( token="foo", verify="supersecuretoken", bot_user_id="baz", bot_id="boo", admins=["aaa", "bbb"], )) return b
async def test_register_admin_message_no_admin(self, caplog): async def handler(): pass bot = SirBot() bot.load_plugin( SlackPlugin(token="foo", verify="bar", bot_user_id="baz", bot_id="boo")) bot["plugins"]["slack"].on_message("hello", handler, admin=True) assert ( "Slack admins ids are not set. Admin limited endpoint will not work." in caplog.text)
async def bot(loop) -> SirBot: b = SirBot() slack = SlackPlugin( token='token', verify='supersecuretoken', bot_user_id='bot_user_id', bot_id='bot_id', ) airtable = AirtablePlugin() endpoints.slack.create_endpoints(slack) b.load_plugin(slack) b.load_plugin(airtable) return b
async def test_load_plugin(self, test_server): class MyPlugin: __name__ = 'myplugin' def __init__(self): pass def load(self, test_bot): pass bot = SirBot() bot.load_plugin(MyPlugin()) assert 'myplugin' in bot.plugins assert isinstance(bot['plugins']['myplugin'], MyPlugin) await test_server(bot)
async def test_load_plugin(self, aiohttp_server): class MyPlugin: __name__ = "myplugin" def __init__(self): pass def load(self, test_bot): pass bot = SirBot() bot.load_plugin(MyPlugin()) assert "myplugin" in bot.plugins assert isinstance(bot["plugins"]["myplugin"], MyPlugin) await aiohttp_server(bot)
async def test_list_plugin(self, test_client): class MyPlugin: __name__ = 'myplugin' def __init__(self): pass def load(self, test_bot): pass bot = SirBot() bot.load_plugin(MyPlugin()) client = await test_client(bot) rep = await client.get('/sirbot/plugins') data = await rep.json() assert data == {'plugins': ['myplugin']}
async def test_list_plugin(self, aiohttp_client): class MyPlugin: __name__ = "myplugin" def __init__(self): pass def load(self, test_bot): pass bot = SirBot() bot.load_plugin(MyPlugin()) client = await aiohttp_client(bot) rep = await client.get("/sirbot/plugins") data = await rep.json() assert data == {"plugins": ["myplugin"]}
async def bot(loop) -> SirBot: b = SirBot() slack = SlackPlugin( token="token", verify="supersecuretoken", bot_user_id="bot_user_id", bot_id="bot_id", ) airtable = AirtablePlugin() endpoints.slack.create_endpoints(slack) api = APIPlugin() endpoints.api.create_endpoints(api) b.load_plugin(slack) b.load_plugin(airtable) b.load_plugin(api) return b
async def test_bot(self, test_server): bot = SirBot() await test_server(bot)
"../logging.yml")) as log_configfile: logging.config.dictConfig( yaml.load(log_configfile.read(), Loader=yaml.SafeLoader)) except Exception as e: logging.basicConfig(level=logging.DEBUG) logger.exception(e) if "SENTRY_DSN" in os.environ: sentry_sdk.init( dsn=os.environ["SENTRY_DSN"], release=os.environ.get("VERSION", "1.0.0"), environment=os.environ.get("ENVIRONMENT", "production"), integrations=[AioHttpIntegration()], ) bot = SirBot() slack = SlackPlugin(**slack_configs) endpoints.slack.create_endpoints(slack) bot.load_plugin(slack) admin_configs = dict(**slack_configs) admin_token = os.environ.get("APP_ADMIN_OAUTH_TOKEN", "FAKE_ADMIN_TOKEN") if admin_token: admin_configs["token"] = admin_token admin_slack = SlackPlugin(**admin_configs) bot.load_plugin(admin_slack, name="admin_slack") airtable = AirtablePlugin() endpoints.airtable.create_endpoints(airtable) bot.load_plugin(airtable)
async def bot(): b = SirBot() b.load_plugin(APSchedulerPlugin()) return b
async def bot(): b = SirBot() b.load_plugin(GithubPlugin(verify="supersecrettoken")) return b
async def test_bot(self, aiohttp_server): bot = SirBot() await aiohttp_server(bot)
async def test_list_plugin_empty(self, aiohttp_client): bot = SirBot() client = await aiohttp_client(bot) rep = await client.get("/sirbot/plugins") data = await rep.json() assert data == {"plugins": []}
async def bot(): b = SirBot() b.load_plugin(SlackPlugin(token='foo', verify='supersecuretoken', bot_user_id='baz', bot_id='boo', admins=['aaa', 'bbb'])) return b
async def test_start_find_bot_id(self, test_server, find_bot_id_query): bot = SirBot() bot.load_plugin(SlackPlugin(token='foo', verify='bar', bot_user_id='baz')) bot['plugins']['slack'].api.query = find_bot_id_query await test_server(bot) assert bot['plugins']['slack'].bot_id == 'B00000000'
async def test_list_plugin_empty(self, test_client): bot = SirBot() client = await test_client(bot) rep = await client.get('/sirbot/plugins') data = await rep.json() assert data == {'plugins': []}
async def bot(): b = SirBot() b.load_plugin(RTDPlugin()) return b