1dbb73a1c8
This isolates all the code that depends on our current directory structure, and it de-clutters test_lib.py a bit.
36 lines
1.1 KiB
Python
36 lines
1.1 KiB
Python
import json
|
|
import os
|
|
|
|
from importlib import import_module
|
|
|
|
from typing import Any, Dict
|
|
|
|
'''
|
|
This module helps us find files in the bots directory. Our
|
|
directory structure is currently:
|
|
|
|
zulip_bots/zulip_bots/bots/
|
|
<bot name>/
|
|
<bot name>.py
|
|
fixtures/
|
|
'''
|
|
|
|
def get_bot_message_handler(bot_name):
|
|
# type: (str) -> Any
|
|
# message_handler is of type 'Any', since it can contain any bot's
|
|
# handler class. Eventually, we want bot's handler classes to
|
|
# inherit from a common prototype specifying the handle_message
|
|
# function.
|
|
lib_module = import_module('zulip_bots.bots.{bot}.{bot}'.format(bot=bot_name)) # type: Any
|
|
return lib_module.handler_class()
|
|
|
|
def read_bot_fixture_data(bot_name, test_name):
|
|
# type: (str, str) -> Dict[str, Any]
|
|
base_path = os.path.realpath(os.path.join(os.path.dirname(
|
|
os.path.abspath(__file__)), 'bots', bot_name, 'fixtures'))
|
|
http_data_path = os.path.join(base_path, '{}.json'.format(test_name))
|
|
with open(http_data_path) as f:
|
|
content = f.read()
|
|
http_data = json.loads(content)
|
|
return http_data
|