grpc: updated start_session to have a definition option, to allow saving xml files and getting service configurations, without fully starting a session

This commit is contained in:
Blake Harnden 2021-04-27 10:49:52 -07:00
parent f891974e3a
commit e0fe86bcb2
7 changed files with 64 additions and 69 deletions

View file

@ -47,7 +47,8 @@ from core.xml.corexml import CoreXmlWriter
class TestGrpcw:
def test_start_session(self, grpc_server: CoreGrpcServer):
@pytest.mark.parametrize("definition", [False, True])
def test_start_session(self, grpc_server: CoreGrpcServer, definition):
# given
client = CoreGrpcClient()
with client.context_connect():
@ -164,10 +165,15 @@ class TestGrpcw:
# when
with patch.object(CoreXmlWriter, "write"):
with client.context_connect():
client.start_session(session)
client.start_session(session, definition=definition)
# then
real_session = grpc_server.coreemu.sessions[session.id]
if definition:
state = EventTypes.DEFINITION_STATE
else:
state = EventTypes.RUNTIME_STATE
assert real_session.state == state
assert node1.id in real_session.nodes
assert node2.id in real_session.nodes
assert wlan_node.id in real_session.nodes