oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#2799 - Update observation tests

Browse Source
This commit is contained in:
Marek Wolan
2024-08-07 13:43:11 +01:00
parent fe599f7745
commit b193b46b7b
3 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/integration_tests/game_layer/observations/test_file_system_observations.py
View File

@@ -26,6 +26,7 @@ def test_file_observation(simulation):
dog_file_obs = FileObservation( dog_file_obs = FileObservation(
where=["network", "nodes", pc.hostname, "file_system", "folders", "root", "files", "dog.png"], where=["network", "nodes", pc.hostname, "file_system", "folders", "root", "files", "dog.png"],
include_num_access=False, include_num_access=False,
file_system_requires_scan=True,
) )
assert dog_file_obs.space["health_status"] == spaces.Discrete(6) assert dog_file_obs.space["health_status"] == spaces.Discrete(6)
@@ -53,6 +54,7 @@ def test_folder_observation(simulation):
root_folder_obs = FolderObservation( root_folder_obs = FolderObservation(
where=["network", "nodes", pc.hostname, "file_system", "folders", "test_folder"], where=["network", "nodes", pc.hostname, "file_system", "folders", "test_folder"],
include_num_access=False, include_num_access=False,
file_system_requires_scan=True,
num_files=1, num_files=1,
files=[], files=[],
) )

1
tests/integration_tests/game_layer/observations/test_node_observations.py
View File

@@ -38,6 +38,7 @@ def test_host_observation(simulation):
applications=[], applications=[],
folders=[], folders=[],
network_interfaces=[], network_interfaces=[],
file_system_requires_scan=True,
) )
assert host_obs.space["operating_status"] == spaces.Discrete(5) assert host_obs.space["operating_status"] == spaces.Discrete(5)

1
tests/integration_tests/game_layer/test_observations.py
View File

@@ -17,6 +17,7 @@ def test_file_observation():
dog_file_obs = FileObservation( dog_file_obs = FileObservation(
where=["network", "nodes", pc.hostname, "file_system", "folders", "root", "files", "dog.png"], where=["network", "nodes", pc.hostname, "file_system", "folders", "root", "files", "dog.png"],
include_num_access=False, include_num_access=False,
file_system_requires_scan=True,
) )
assert dog_file_obs.observe(state) == {"health_status": 1} assert dog_file_obs.observe(state) == {"health_status": 1}
assert dog_file_obs.space == spaces.Dict({"health_status": spaces.Discrete(6)}) assert dog_file_obs.space == spaces.Dict({"health_status": spaces.Discrete(6)})