def test_custom_log_handler(context, capsys, mocker: MockerFixture):
@scheduled(handler=logging.StreamHandler())
def app(event, api, cache):
Logger.info('Info message!')
event = RawScheduledEvent(
asset_id=0,
interval=int(),
schedule=int(),
schedule_start=int(),
app_connection=1,
app_stream=int(),
company=int(),
)
mocker.patch.object(RawScheduledEvent, 'set_schedule_as_completed')
app(
[[event.dict(
by_alias=True,
exclude_unset=True,
)]],
context,
)
captured = capsys.readouterr()
assert captured.out.endswith('Info message!\n')
assert captured.err == 'Info message!\n'
def test_scheduled_logging(context, capsys, mocker: MockerFixture):
@scheduled
def app(event, api, cache):
Logger.warning('Hello, World!')
event = RawScheduledEvent(
asset_id=0,
interval=int(),
schedule=int(),
schedule_start=int(),
app_connection=1,
app_stream=int(),
company=int(),
)
mocker.patch.object(RawScheduledEvent, 'set_schedule_as_completed')
with freezegun.freeze_time(datetime.datetime(2021, 1, 2, 3, 4, 5, 678910)):
app(
[[event.dict(
by_alias=True,
exclude_unset=True,
)]],
context,
)
assert (
capsys.readouterr().out ==
f'2021-01-02T03:04:05.678Z {context.aws_request_id} WARNING '
f'ASSET={event.asset_id} AC={event.app_connection_id} | Hello, World!\n'
)