#2350: add tests to check spaces + acl obs test + nmne space changes

This commit is contained in:
Czar Echavez
2024-03-11 17:47:33 +00:00
parent a228a09917
commit cd6d6325db
9 changed files with 193 additions and 12 deletions

View File

@@ -1,4 +1,5 @@
import pytest
from gymnasium import spaces
from primaite.game.agent.observations.software_observation import ApplicationObservation, ServiceObservation
from primaite.simulator.network.hardware.nodes.host.computer import Computer
@@ -29,6 +30,9 @@ def test_service_observation(simulation):
service_obs = ServiceObservation(where=["network", "nodes", pc.hostname, "services", "NTPServer"])
assert service_obs.space["operating_status"] == spaces.Discrete(7)
assert service_obs.space["health_status"] == spaces.Discrete(5)
observation_state = service_obs.observe(simulation.describe_state())
assert observation_state.get("health_status") == 0