Merge pull request #777 from tlsfuzzer/descriptive-ExpectAlert

ExpectAlert - add __repr__
tlsfuzzer · Sep 29, 2021 · dbd56c1 · dbd56c1
2 parents 09d51ea + 34b0bc2
commit dbd56c1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/tests/test_tlsfuzzer_expect.py b/tests/test_tlsfuzzer_expect.py
@@ -3104,6 +3104,12 @@ def test_process_with_multiple_values_not_matching_anything(self):
                          "not match received "
                          "\"255\"")
 
+    def test___str__(self):
+        exp = ExpectAlert(AlertLevel.warning,
+                          AlertDescription.illegal_parameter)
+
+        self.assertEqual(str(exp), "ExpectAlert(level=1, description=47)")
+
 
 class TestExpectSSL2Alert(unittest.TestCase):
     def test___init__(self):

diff --git a/tlsfuzzer/expect.py b/tlsfuzzer/expect.py
@@ -1845,6 +1845,10 @@ def process(self, state, msg):
         if problem_desc:
             raise AssertionError(problem_desc)
 
+    def __repr__(self):
+        """Return human readable representation of object."""
+        return self._repr(["level", "description"])
+
 
 class ExpectSSL2Alert(ExpectHandshake):
     """Processing of SSLv2 Handshake protocol alert messages"""