async def test_call(response_mock):
token = 'asdf1234'
response_mock.post(
'https://slack.com/api/test11',
body=ujson.dumps({
'res': 'hello world!',
}),
headers={'content-type': 'application/json'},
status=200,
)
response_mock.post(
'https://slack.com/api/test12',
body=ujson.dumps({
'res': 'hello world!',
'data': {
'extra': 'wow',
},
}),
headers={'content-type': 'application/json'},
status=200,
)
response_mock.post(
'https://slack.com/api/test21',
body=ujson.dumps({
'error': 'aaa',
}),
headers={'content-type': 'application/json'},
status=404,
)
response_mock.post(
'https://slack.com/api/test22',
body=ujson.dumps({
'error': 'aaa',
}),
headers={'content-type': 'application/json'},
status=404,
)
response_mock.post(
'https://slack.com/api/test3',
body=ujson.dumps({
'res': 'hello world!',
}),
headers={'content-type': 'application/json'},
status=200,
)
config = AttrDict(copy.deepcopy(DEFAULT))
config.DATABASE_URL = 'sqlite:///'
config.TOKEN = 'asdf1234'
config['LOGGING']['loggers']['yui']['handlers'] = ['console']
del config['LOGGING']['handlers']['file']
config.REGISTER_CRONTAB = False
bot = Bot(config)
res = await bot.call('test11')
assert res['res'] == 'hello world!'
res = await bot.call('test12', data={'extra': 'wow'})
assert res['res'] == 'hello world!'
assert res['data']['extra'] == 'wow'
with pytest.raises(APICallError) as e:
await bot.call('test21')
assert str(e.value) == 'fail to call test21 with None'
assert e.value.status_code == 404
assert e.value.result == {'error': 'aaa'}
assert e.value.headers['Content-Type'] == 'application/json'
with pytest.raises(APICallError) as e:
await bot.call('test22', data={'extra': 'wow'})
assert str(e.value) == "fail to call test22 with {'extra': 'wow'}"
assert e.value.status_code == 404
assert e.value.result == {'error': 'aaa'}
assert e.value.headers['Content-Type'] == 'application/json'
res = await bot.call('test3', token=token)
assert res['res'] == 'hello world!'
