#126: fix MakeNote.key_to_log not being updated. (#129)

ScreenPyHQ · Feb 20, 2024 · ed809bf · ed809bf
1 parent c75cee2
commit ed809bf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/screenpy/actions/make_note.py b/screenpy/actions/make_note.py
@@ -37,6 +37,7 @@ class MakeNote:
     """
 
     key: str | None
+    key_to_log: str | None
     question: T_Q
 
     @classmethod
@@ -56,6 +57,7 @@ def of_the(cls, question: T_Q) -> Self:
     def as_(self, key: str) -> Self:
         """Set the key to use to recall this noted value."""
         self.key = key
+        self.key_to_log = represent_prop(key)
         return self
 
     def describe(self) -> str:

diff --git a/tests/test_actions.py b/tests/test_actions.py
@@ -384,6 +384,7 @@ def test_key_value_set(self) -> None:
 
         assert mn.question == test_question
         assert mn.key == test_key
+        assert mn.key_to_log == f"'{test_key}'"
 
     def test_answers_question(self, Tester: Actor) -> None:
         mock_question = FakeQuestion()