Add extracted tools: CitrineOS, OpenOCPP, ShapeShifter
- 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
This commit is contained in:
@@ -0,0 +1,72 @@
|
||||
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
|
||||
Reference in New Issue
Block a user