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))