26 lines
1.0 KiB
Python
26 lines
1.0 KiB
Python
# © Crown-owned copyright 2025, Defence Science and Technology Laboratory UK
|
|
from gymnasium import spaces
|
|
|
|
from primaite.game.agent.observations.file_system_observations import FileObservation
|
|
from primaite.simulator.network.hardware.nodes.host.computer import Computer
|
|
from primaite.simulator.sim_container import Simulation
|
|
|
|
|
|
def test_file_observation():
|
|
sim = Simulation()
|
|
pc: Computer = Computer.from_config(
|
|
config={"type": "computer", "hostname": "beep", "ip_address": "123.123.123.123", "subnet_mask": "255.255.255.0"}
|
|
)
|
|
sim.network.add_node(pc)
|
|
f = pc.file_system.create_file(file_name="dog.png")
|
|
|
|
state = sim.describe_state()
|
|
|
|
dog_file_obs = FileObservation(
|
|
where=["network", "nodes", pc.config.hostname, "file_system", "folders", "root", "files", "dog.png"],
|
|
include_num_access=False,
|
|
file_system_requires_scan=False,
|
|
)
|
|
assert dog_file_obs.observe(state) == {"health_status": 1}
|
|
assert dog_file_obs.space == spaces.Dict({"health_status": spaces.Discrete(6)})
|