56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
import pytest
|
|
|
|
from primaite.simulator.core import Action
|
|
from primaite.simulator.network.hardware.nodes.computer import Computer
|
|
from primaite.simulator.network.hardware.nodes.server import Server
|
|
from primaite.simulator.network.hardware.nodes.switch import Switch
|
|
from primaite.simulator.sim_container import Simulation
|
|
from primaite.simulator.system.services.database_service import DatabaseService
|
|
|
|
|
|
def test_passing_actions_down(monkeypatch) -> None:
|
|
"""Check that an action is passed down correctly to the child component."""
|
|
|
|
sim = Simulation()
|
|
|
|
pc1 = Computer(hostname="PC-1", ip_address="10.10.1.1", subnet_mask="255.255.255.0")
|
|
pc2 = Computer(hostname="PC-2", ip_address="10.10.1.2", subnet_mask="255.255.255.0")
|
|
srv = Server(hostname="WEBSERVER", ip_address="10.10.1.100", subnet_mask="255.255.255.0")
|
|
s1 = Switch(hostname="switch1")
|
|
|
|
for n in [pc1, pc2, srv, s1]:
|
|
sim.network.add_node(n)
|
|
|
|
database_service = DatabaseService(file_system=srv.file_system)
|
|
srv.install_service(database_service)
|
|
|
|
downloads_folder = pc1.file_system.create_folder("downloads")
|
|
pc1.file_system.create_file("bermuda_triangle.png", folder_name="downloads")
|
|
|
|
sim.network.connect(pc1.ethernet_port[1], s1.switch_ports[1])
|
|
sim.network.connect(pc2.ethernet_port[1], s1.switch_ports[2])
|
|
sim.network.connect(s1.switch_ports[3], srv.ethernet_port[1])
|
|
|
|
# call this method to make sure no errors occur.
|
|
sim._action_manager.get_action_tree()
|
|
|
|
# patch the action to do something which we can check the result of.
|
|
action_invoked = False
|
|
|
|
def succeed():
|
|
nonlocal action_invoked
|
|
action_invoked = True
|
|
|
|
monkeypatch.setitem(
|
|
downloads_folder._action_manager.actions, "repair", Action(func=lambda request, context: succeed())
|
|
)
|
|
|
|
assert not action_invoked
|
|
|
|
# call the patched method
|
|
sim.apply_action(
|
|
["network", "node", pc1.uuid, "file_system", "folder", pc1.file_system.get_folder("downloads").uuid, "repair"]
|
|
)
|
|
|
|
assert action_invoked
|