d398a6ced2
- CitrineOS core extracted (CSMS OCPP 2.0.1) - OpenOCPP extracted (firmware OCPP 1.6J/2.0.1) - ShapeShifter library installed (pip install -e) - ShapeShifter specification extracted - EVerest extracted TODO updated with progress
73 lines
2.6 KiB
Python
73 lines
2.6 KiB
Python
from concurrent.futures import ThreadPoolExecutor
|
|
|
|
import pytest
|
|
|
|
from shapeshifter_uftp.service.base_service import snake_case
|
|
from shapeshifter_uftp.uftp import routing_map
|
|
|
|
from .helpers.messages import messages
|
|
from .helpers.services import DummyAgrService, DummyCroService, DummyDsoService
|
|
|
|
|
|
# These fixtures allow us to only start up the services once and use
|
|
# them for all the parametrized test cases, which speeds up testing.
|
|
@pytest.fixture(scope='module')
|
|
def agr_service():
|
|
with DummyAgrService() as service:
|
|
yield service
|
|
|
|
@pytest.fixture(scope='module')
|
|
def cro_service():
|
|
with DummyCroService() as service:
|
|
yield service
|
|
|
|
@pytest.fixture(scope='module')
|
|
def dso_service():
|
|
with DummyDsoService() as service:
|
|
yield service
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
'message',
|
|
messages,
|
|
ids=[message.__class__.__name__ for message in messages]
|
|
)
|
|
def test_communications(message, agr_service, cro_service, dso_service):
|
|
service_map = {
|
|
"AGR": agr_service,
|
|
"CRO": cro_service,
|
|
"DSO": dso_service,
|
|
}
|
|
|
|
sender_role, recipient_role = routing_map[type(message)]
|
|
sender = service_map[sender_role]
|
|
recipient = service_map[recipient_role]
|
|
client_method = f"{recipient.sender_role.lower()}_client"
|
|
recipient.reset_futures(snake_case(message.__class__.__name__))
|
|
with ThreadPoolExecutor() as executor:
|
|
with getattr(sender, client_method)(recipient.sender_domain, version="3.1.0") as client:
|
|
sending_method = f"send_{snake_case(message.__class__.__name__)}"
|
|
main_future = executor.submit(getattr(client, sending_method), message)
|
|
|
|
assert recipient.request_futures[f"process_{snake_case(message.__class__.__name__)}"].result() == message
|
|
assert main_future.result() is None
|
|
|
|
|
|
def test_test_messages(agr_service, cro_service, dso_service):
|
|
service_map = {
|
|
"AGR": agr_service,
|
|
"CRO": cro_service,
|
|
"DSO": dso_service,
|
|
}
|
|
|
|
agr_service.dso_client(dso_service.sender_domain).send_test_message()
|
|
assert dso_service.request_futures["process_test_message"].result() is not None
|
|
assert agr_service.request_futures["process_test_message_response"].result() is not None
|
|
|
|
agr_service.cro_client(cro_service.sender_domain).send_test_message()
|
|
assert cro_service.request_futures["process_test_message"].result() is not None
|
|
|
|
dso_service.agr_client(agr_service.sender_domain).send_test_message()
|
|
assert agr_service.request_futures["process_test_message"].result() is not None
|
|
assert dso_service.request_futures["process_test_message_response"].result() is not None
|