fix: initialize private attributes in MatrixEntry and Tactic to prevent AttributeError#243
Open
Alisha-chaudhary wants to merge 1 commit into
Conversation
…nt AttributeError
|
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.



While reviewing the navlayers exporter code, I noticed that
MatrixEntryandTacticonly initialize some private attributes when the corresponding constructor arguments are notNone.As a result, creating these objects with omitted or default arguments can leave attributes undefined. Accessing the related properties later raises an
AttributeErrorinstead of returningNone.Reproduction
from mitreattack.navlayers.exporters.matrix_gen import MatrixEntry
entry = MatrixEntry()
print(entry.id)
The same issue can occur in
Tacticfor its private attributes when constructor arguments are omitted.Cause
Several private attributes are only assigned inside conditional blocks that check whether the provided argument is not
None. If an argument is omitted, the attribute is never created, causing property getters to fail when accessed.Fix
Initialize the affected private attributes to
Noneat the start of__init__before the existing conditional setter logic runs.MatrixEntry
self.__id = None
self.__name = None
This preserves the existing behavior for valid inputs while ensuring the attributes always exist and can be accessed safely.
Tests
Added tests in
tests/test_matrix_gen.pyto verify that:MatrixEntrycan be instantiated with default arguments without raisingAttributeErrorTacticcan be instantiated with default arguments without raisingAttributeError