test: Data Classes str()

Add tests for DictWrapper str() functionality, including Data Classes
with:
- no properties
- single property
- blacklisted property
- recursive properties (Data Classes inside Data Classes)

Author: neilramsay

tests/functional/test_data_classes.py

@@ -126,6 +126,97 @@ class DataClassSample(DictWrapper):
     assert event_source.items() == data.items()
 
 
+def test_dict_wrapper_str_no_property():
+    class DataClassSample(DictWrapper):
+        attribute = None
+
+        def function(self):
+            pass
+
+    event_source = DataClassSample({})
+    assert event_source._properties() == ["raw_event"]
+    assert event_source._str_helper() == {"raw_event": "[SENSITIVE]"}
+    assert event_source.__str__() == "{'raw_event': '[SENSITIVE]'}"
+    assert str(event_source) == "{'raw_event': '[SENSITIVE]'}"
+
+
+def test_dict_wrapper_str_single_property():
+    class DataClassSample(DictWrapper):
+        attribute = None
+
+        def function(self):
+            pass
+
+        @property
+        def data_property(self):
+            return "value"
+
+    event_source = DataClassSample({})
+    assert event_source._properties() == ["data_property", "raw_event"]
+    assert event_source._str_helper() == {"data_property": "value", "raw_event": "[SENSITIVE]"}
+    assert event_source.__str__() == "{'data_property': 'value', 'raw_event': '[SENSITIVE]'}"
+    assert str(event_source) == "{'data_property': 'value', 'raw_event': '[SENSITIVE]'}"
+
+
+def test_dict_wrapper_str_recursive_property():
+    class DataClassTerminal(DictWrapper):
+        attribute = None
+
+        def function(self):
+            pass
+
+        @property
+        def terminal_property(self):
+            return "end-recursion"
+
+    class DataClassRecursive(DictWrapper):
+        attribute = None
+
+        def function(self):
+            pass
+
+        @property
+        def data_property(self):
+            return DataClassTerminal({})
+
+    event_source = DataClassRecursive({})
+    assert event_source._properties() == ["data_property", "raw_event"]
+    assert event_source._str_helper() == {
+        "data_property": {"raw_event": "[SENSITIVE]", "terminal_property": "end-recursion"},
+        "raw_event": "[SENSITIVE]",
+    }
+    assert (
+        event_source.__str__()
+        == "{'data_property': {'raw_event': '[SENSITIVE]', 'terminal_property': 'end-recursion'},"
+        + " 'raw_event': '[SENSITIVE]'}"
+    )
+    assert (
+        str(event_source)
+        == "{'data_property': {'raw_event': '[SENSITIVE]', 'terminal_property': 'end-recursion'},"
+        + " 'raw_event': '[SENSITIVE]'}"
+    )
+
+
+def test_dict_wrapper_str_sensitive_data_property():
+    class DataClassSample(DictWrapper):
+        attribute = None
+
+        def function(self):
+            pass
+
+        _str_sensitive_data = ["data_property"]
+
+        @property
+        def data_property(self):
+            return "value"
+
+    event_source = DataClassSample({})
+    assert event_source._properties() == ["data_property", "raw_event"]
+    assert event_source._str_helper() == {"data_property": "[SENSITIVE]", "raw_event": "[SENSITIVE]"}
+    assert event_source.__str__() == "{'data_property': '[SENSITIVE]', 'raw_event': '[SENSITIVE]'}"
+    assert str(event_source) == "{'data_property': '[SENSITIVE]', 'raw_event': '[SENSITIVE]'}"
+
+
 def test_cloud_watch_dashboard_event():
     event = CloudWatchDashboardCustomWidgetEvent(load_event("cloudWatchDashboardEvent.json"))
     assert event.describe is False