#1522: create_random_red_agent -> _create_random_red_agent + converting NodeStateInstructionRed into a dataclass
This commit is contained in:
@@ -1,8 +1,11 @@
|
||||
# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.
|
||||
"""Defines node behaviour for Green PoL."""
|
||||
from dataclasses import dataclass
|
||||
|
||||
from primaite.common.enums import NodePOLType
|
||||
|
||||
|
||||
@dataclass()
|
||||
class NodeStateInstructionRed(object):
|
||||
"""The Node State Instruction class."""
|
||||
|
||||
@@ -137,20 +140,3 @@ class NodeStateInstructionRed(object):
|
||||
The source node service state
|
||||
"""
|
||||
return self.source_node_service_state
|
||||
|
||||
def __repr__(self):
|
||||
return (
|
||||
f"{self.__class__.__name__}("
|
||||
f"id={self.id}, "
|
||||
f"start_step={self.start_step}, "
|
||||
f"end_step={self.end_step}, "
|
||||
f"target_node_id={self.target_node_id}, "
|
||||
f"initiator={self.initiator}, "
|
||||
f"pol_type={self.pol_type}, "
|
||||
f"service_name={self.service_name}, "
|
||||
f"state={self.state}, "
|
||||
f"source_node_id={self.source_node_id}, "
|
||||
f"source_node_service={self.source_node_service}, "
|
||||
f"source_node_service_state={self.source_node_service_state}"
|
||||
f")"
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user