def setUp(self) -> None:
super().setUp()
basicConfig(level=INFO)
self.db_file.unlink(missing_ok=True)
self.db = Database(f'sqlite:///{self.db_file}')
async def init():
await self.db.connect()
await migrate(self.db)
get_event_loop().run_until_complete(init())
def test_invoke_with_activity_options(self):
class HelloActivities:
@activity_method(task_list="test-tasklist")
def hello(self, arg1, arg2):
pass
stub = HelloActivities()
stub._decision_context = self.decision_context
stub._retry_parameters = None
stub._activity_options = ActivityOptions(
schedule_to_close_timeout_seconds=50,
schedule_to_start_timeout_seconds=100,
start_to_close_timeout_seconds=150,
heartbeat_timeout_seconds=200,
task_list="tasklist-tasklist-tasklist")
async def fn():
await stub.hello(1, "one")
loop = get_event_loop()
self.task = loop.create_task(fn())
run_once(loop)
self.decision_context.schedule_activity_task.assert_called_once()
args, kwargs = self.decision_context.schedule_activity_task.call_args_list[
0]
parameters: ExecuteActivityParameters = kwargs["parameters"]
self.assertEquals(parameters.schedule_to_close_timeout_seconds, 50)
self.assertEquals(parameters.schedule_to_start_timeout_seconds, 100)
self.assertEquals(parameters.start_to_close_timeout_seconds, 150)
self.assertEquals(parameters.heartbeat_timeout_seconds, 200)
self.assertEquals(parameters.task_list, "tasklist-tasklist-tasklist")
def __init__(self, bot: Bot):
self.bot = bot
self.loop = get_event_loop()
self.valid_clips = ["airhorn", "rickroll"]
self.playing_in = []
self.scan.start()
self.logger = self.bot.logger
def test_invoke_stub_with_args(self):
class HelloActivities:
@activity_method(task_list="test-tasklist")
def hello(self, arg1, arg2):
pass
stub = HelloActivities()
stub._decision_context = self.decision_context
async def fn():
await stub.hello(1, "one")
loop = get_event_loop()
self.task = loop.create_task(fn())
run_once(loop)
self.decision_context.schedule_activity_task.assert_called_once()
args, kwargs = self.decision_context.schedule_activity_task.call_args_list[0]
self.assertEqual(b'[1, "one"]', kwargs["parameters"].input)
def setUp(self) -> None:
self.decision_context: DecisionContext = Mock()
self.decision_context.schedule_activity_task = MagicMock(
return_value=get_event_loop().create_future())
self.task = None
def handle():
loop = get_event_loop()
client_socket = socket.socket()
return HandleClient(loop, client_socket)
def tearDown(self) -> None:
super().tearDown()
get_event_loop().run_until_complete(self.db.disconnect())
self.db_file.unlink(missing_ok=True)
def wrapper(*args, **kwargs):
return get_event_loop().run_until_complete(fn(*args, *kwargs))