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

893 - added new line for assert statements

Browse Source
This commit is contained in:
SunilSamra
2023-06-06 15:54:35 +01:00
parent 2eff3912fb
commit 6cc9516744
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/primaite/environment/primaite_env.py
View File

@@ -423,6 +423,7 @@ class Primaite(Env):
_action: The action space from the agent _action: The action space from the agent
""" """
# At the moment, actions are only affecting nodes # At the moment, actions are only affecting nodes
print("ACTION:", self.action_dict[_action])
if self.action_type == ActionType.NODE: if self.action_type == ActionType.NODE:
self.apply_actions_to_nodes(_action) self.apply_actions_to_nodes(_action)
elif self.action_type == ActionType.ACL: elif self.action_type == ActionType.ACL:

5
tests/test_single_action_space.py
View File

@@ -19,7 +19,7 @@ def run_generic_set_actions(env, config_values):
if step == 5: if step == 5:
# [1, 1, 2, 1, 1, 1] # [1, 1, 2, 1, 1, 1]
# Creates an ACL rule # Creates an ACL rule
# Deny traffic from server_1 to node_1 on port FTP # Allows traffic from server_1 to node_1 on port FTP
action = 7 action = 7
elif step == 7: elif step == 7:
# [1, 1, 2, 0] Node Action # [1, 1, 2, 0] Node Action
@@ -97,4 +97,5 @@ def test_agent_is_executing_actions_from_both_spaces():
# In the scenario, we specified that the agent should create only 1 acl rule # In the scenario, we specified that the agent should create only 1 acl rule
num_of_rules = len(acl_rules_list) num_of_rules = len(acl_rules_list)
# Therefore these statements below MUST be true # Therefore these statements below MUST be true
assert computer_node_hardware_state == HardwareState.OFF and num_of_rules == 1 assert computer_node_hardware_state == HardwareState.OFF
assert num_of_rules == 1