def test_virtual_server_connection_recv_immedaite_msg_without_reply() -> None:
"""Test that VirtualServerConnection.recv_immedaite_msg_without_reply works."""
node = MockNode()
server = VirtualServerConnection(node=node)
msg = construct_dummy_message(SignedImmediateSyftMessageWithoutReply)
assert not server.recv_immediate_msg_without_reply(msg=msg)
def test_virtual_server_connection_recv_eventual_msg_without_reply() -> None:
"""
Test that VirtualServerConnection.recv_eventual_msg_with_reply raises NotImplementedError.
"""
node = MockNode()
server = VirtualServerConnection(node=node)
msg = construct_dummy_message(SignedEventualSyftMessageWithoutReply)
assert server.recv_eventual_msg_without_reply(msg=msg) is None
def test_virtual_client_connection_init() -> None:
"""
Test that VirtualClientConnection uses the VirtualServerConnection
passed into its constructor.
"""
node = MockNode()
server = VirtualServerConnection(node=node)
client = VirtualClientConnection(server=server)
assert client.server == server
def test_virtual_server_connection_init() -> None:
"""
Test that VirtualServerConnection uses the AbstractNode
passed into its constructor.
"""
node = MockNode()
server = VirtualServerConnection(node=node)
assert server.node == node
def test_virtual_client_connection_send_immediate_msg_without_reply() -> None:
"""
Test that VirtualClientConnection.send_immediate_msg_without_reply works.
"""
node = MockNode()
server = VirtualServerConnection(node=node)
client = VirtualClientConnection(server=server)
msg = construct_dummy_message(SignedImmediateSyftMessageWithoutReply)
assert client.send_immediate_msg_without_reply(msg=msg) is None
def test_solo_route_send_eventual_msg_without_reply() -> None:
"""Test SoloRoute.send_eventual_msg_without_reply method works."""
node = MockNode()
destination = SpecificLocation()
server = VirtualServerConnection(node=node)
connection = VirtualClientConnection(server=server)
h_solo = SoloRoute(destination=destination, connection=connection)
msg = construct_dummy_message(SignedEventualSyftMessageWithoutReply)
assert not h_solo.send_eventual_msg_without_reply(msg)
def test_broadcast_route_init() -> None:
"""
Test that BroadcastRoute will use the Location and
ClientConnection passed into its constructor.
"""
destination = SpecificLocation()
virtual_server = VirtualServerConnection(node=Node())
virtual_client = VirtualClientConnection(server=virtual_server)
b_route = BroadcastRoute(destination=destination, connection=virtual_client)
assert b_route.schema.destination is destination
assert b_route.connection is virtual_client
def test_solo_route_init() -> None:
"""
Test that SoloRoute will use the Location and
ClientConnection/BidirectionalConnection passed into its constructor.
"""
# Test SoloRoute with VirtualClientConnection (ClientConnection) in constructor
destination = SpecificLocation()
virtual_server = VirtualServerConnection(node=Node())
virtual_client = VirtualClientConnection(server=virtual_server)
h_solo = SoloRoute(destination=destination, connection=virtual_client)
assert h_solo.schema.destination is destination
assert h_solo.connection is virtual_client
def test_solo_route_send_immediate_msg_with_reply() -> None:
"""Test SoloRoute.send_immediate_msg_with_reply method works."""
node = MockNode()
destination = SpecificLocation()
server = VirtualServerConnection(node=node)
connection = VirtualClientConnection(server=server)
h_solo = SoloRoute(destination=destination, connection=connection)
msg = construct_dummy_message(SignedImmediateSyftMessageWithReply)
ret = h_solo.send_immediate_msg_with_reply(msg)
assert isinstance(
ret,
SignedImmediateSyftMessageWithoutReply,
)