platypush/tests/test_procedure.py

83 lines
2 KiB
Python
Raw Normal View History

import os
import pytest
import tempfile
import time
from platypush.message.event.custom import CustomEvent
from .utils import register_user, send_request
2021-03-06 17:03:50 +01:00
tmp_files = []
@pytest.fixture(scope='module', autouse=True)
def user(*_):
register_user()
2021-03-06 17:03:50 +01:00
@pytest.fixture(scope='module', autouse=True)
def tmp_file(*_):
2021-03-06 17:03:50 +01:00
n_files = 2
for _ in range(n_files):
tmp_file = tempfile.NamedTemporaryFile(prefix='platypush-test-procedure-', suffix='.txt', delete=False)
tmp_files.append(tmp_file.name)
yield
2021-03-06 17:03:50 +01:00
for f in tmp_files:
if os.path.isfile(f):
os.unlink(f)
2021-03-06 17:03:50 +01:00
def check_file_content(expected_content: str, tmp_file: str, timeout: int = 10):
error = None
start_time = time.time()
while time.time() - start_time < timeout:
try:
assert os.path.isfile(tmp_file), 'The expected output file was not created'
with open(tmp_file, 'r') as f:
content = f.read()
assert content == expected_content, 'The output file did not contain the expected text'
error = None
break
except Exception as e:
error = e
if error:
raise error
2021-03-06 17:03:50 +01:00
def test_procedure_call(base_url):
"""
Test the result of a procedure invoked directly over HTTP.
"""
output_text = 'Procedure test'
send_request(
action='procedure.write_file',
args={
2021-03-06 17:03:50 +01:00
'file': tmp_files[0],
'content': output_text,
})
2021-03-06 17:03:50 +01:00
check_file_content(expected_content=output_text, tmp_file=tmp_files[0])
2021-03-06 17:03:50 +01:00
def test_procedure_from_event(app, base_url):
"""
Test the result of a procedure triggered by an event.
"""
output_text = 'Procedure from event test'
2021-03-06 17:03:50 +01:00
event_type = 'test_procedure'
# noinspection PyUnresolvedReferences
2021-03-06 17:03:50 +01:00
app.bus.post(CustomEvent(subtype=event_type, file=tmp_files[1], content=output_text))
check_file_content(expected_content=output_text, tmp_file=tmp_files[1])
if __name__ == '__main__':
pytest.main()
# vim:sw=4:ts=4:et: